24 lines
431 B
Ruby
24 lines
431 B
Ruby
|
#!/usr/bin/ruby
|
||
|
|
|
||
|
|
$:.unshift(File.join(File.dirname(__FILE__), '..', '..'))
|
||
|
|
|
||
|
|
require 'test/unit'
|
||
|
|
require 'rex/exploitation/egghunter'
|
||
|
|
|
||
|
|
class Rex::Exploitation::Egghunter::UnitTest < Test::Unit::TestCase
|
||
|
|
|
||
|
|
Klass = Rex::Exploitation::Egghunter
|
||
|
|
|
||
|
|
def test_egghunter
|
||
|
|
r = Klass.new('bogus')
|
||
|
|
assert_nil(r.generate)
|
||
|
|
|
||
|
|
r = Klass.new('win')
|
||
|
|
assert_nil(r.generate)
|
||
|
|
|
||
|
|
r = Klass.new('win', 'x86')
|
||
|
|
assert_not_nil(r.generate)
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|