42 lines
1.1 KiB
Ruby
42 lines
1.1 KiB
Ruby
|
#!/usr/bin/ruby
|
||
|
|
|
||
|
|
$:.unshift(File.join(File.dirname(__FILE__), '..', '..'))
|
||
|
|
|
||
|
|
require 'test/unit'
|
||
|
|
require 'rex/socket/switch_board'
|
||
|
|
|
||
|
|
class Rex::Socket::SwitchBoard::UnitTest < Test::Unit::TestCase
|
||
|
|
|
||
|
|
Klass = Rex::Socket::SwitchBoard
|
||
|
|
|
||
|
|
def test_add
|
||
|
|
Klass.flush_routes
|
||
|
|
assert_equal(true, Klass.add_route('0.0.0.0', 0, 'foo'))
|
||
|
|
assert_equal(false, Klass.add_route('0.0.0.0', 0, 'foo'))
|
||
|
|
assert_equal(1, Klass.routes.length)
|
||
|
|
|
||
|
|
assert_equal('0.0.0.0', Klass.routes[0].subnet)
|
||
|
|
assert_equal('0.0.0.0', Klass.routes[0].netmask)
|
||
|
|
assert_equal(0, Klass.routes[0].bitmask)
|
||
|
|
assert_equal('foo', Klass.routes[0].comm)
|
||
|
|
end
|
||
|
|
|
||
|
|
def test_remove
|
||
|
|
Klass.flush_routes
|
||
|
|
assert_equal(true, Klass.add_route('0.0.0.0', 0, 'foo'))
|
||
|
|
assert_equal(true, Klass.remove_route('0.0.0.0', 0, 'foo'))
|
||
|
|
assert_equal(false, Klass.remove_route('0.0.0.0', 0, 'foo'))
|
||
|
|
assert_equal(0, Klass.routes.length)
|
||
|
|
end
|
||
|
|
|
||
|
|
def test_best_comm
|
||
|
|
Klass.flush_routes
|
||
|
|
Klass.add_route('0.0.0.0', 0, 'default')
|
||
|
|
Klass.add_route('1.2.3.0', 24, 'spec')
|
||
|
|
|
||
|
|
assert_equal('default', Klass.best_comm('4.5.6.7'))
|
||
|
|
assert_equal('spec', Klass.best_comm('1.2.3.7'))
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|