def prepare_test(self, config, capabilitites=None):
"""
Cannot be done by setUp as we need special config
"""
self.multicast = Multicast(config, self.overlay_bridge, 1234, 1, self.peer_db,
capabilities=capabilitites)
self.test_server = TestUDPServer(self.multicast.getSocket(),
self.multicast)
self.test_server.start()
class MulticastTest(unittest.TestCase):
"""
Test multicast class
"""
def __init__(self, param):
unittest.TestCase.__init__(self, param)
#TestAsServer.__init__(self, param)
self.test_server = None
self.overlay_bridge = FakeOverlayBridge()
self.peer_db = FakePeerDBHandler()
def prepare_test(self, config, capabilitites=None):
"""
Cannot be done by setUp as we need special config
"""
self.multicast = Multicast(config, self.overlay_bridge, 1234, 1, self.peer_db,
capabilities=capabilitites)
self.test_server = TestUDPServer(self.multicast.getSocket(),
self.multicast)
self.test_server.start()
def tearDown(self):
if self.test_server is not None:
self.test_server.stop()
self.multicast = None
def testIPv4(self):
# Dummy config
config = {'permid':'123',
'hostname':'myhostname',
'port':'1234',
'multicast_ipv4_address':'224.0.1.43',
'multicast_ipv6_address':'ff02::4124:1261:ffef',
'multicast_port':'6124',
'multicast_enabled':True,
'multicast_ipv4_enabled':True,
'multicast_ipv6_enabled':False,
'multicast_announce':True}
self.prepare_test(config)
failed = True
seen = 0
for (permid, addr, capabilities) in self.multicast.discoverNodes():
if permid == '123':
failed = False
if failed:
raise Exception("Didn't discover myself using IPv4")
def testIPv6(self):
# Dummy config
config = {'permid':'123',
'multicast_ipv4_address':'224.0.1.43',
'multicast_ipv6_address':'ff02::4124:1261:ffef',
'multicast_port':'6124',
'multicast_enabled':True,
'multicast_ipv4_enabled':False,
'multicast_ipv6_enabled':True,
'multicast_announce':True}
self.prepare_test(config)
failed = True
for (permid, addr, capabilities) in self.multicast.discoverNodes():
if permid == '123':
failed = False
if failed:
raise Exception("Didn't discover myself using IPv6")
def testBoth(self):
# Dummy config
config = {'permid':'123',
'multicast_ipv4_address':'224.0.1.43',
'multicast_ipv6_address':'ff02::4124:1261:ffef',
'multicast_port':'6124',
'multicast_enabled':True,
'multicast_ipv4_enabled':True,
'multicast_ipv6_enabled':True,
'multicast_announce':True}
self.prepare_test(config)
seen = 0
for (permid, addr, capabilities) in self.multicast.discoverNodes():
if permid == '123':
seen += 1
if seen < 2:
raise Exception("Didn't discover myself enough using both (saw me %d times, expected 2)"%seen)
def testAllDisabled(self):
# Dummy config
config = {'permid':'123',
'multicast_ipv4_address':'224.0.1.43',
'multicast_ipv6_address':'ff02::4124:1261:ffef',
'multicast_port':'6124',
'multicast_ipv4_enabled':False,
'multicast_ipv6_enabled':False,
'multicast_announce':True}
self.prepare_test(config)
try:
if len(self.multicast.discoverNodes()) > 0:
raise Exception("Discovered nodes even though multicast is not allowed")
except:
# Expected
pass
def testAnnounce(self):
# Dummy config
config = {'permid':'123',
'multicast_ipv4_address':'224.0.1.43',
'multicast_ipv6_address':'ff02::4124:1261:ffef',
'multicast_port':'6124',
'multicast_enabled':True,
'multicast_ipv4_enabled':True,
'multicast_ipv6_enabled':True,
'multicast_announce':True}
self.prepare_test(config)
# Handle the announce
self.announces = []
self.multicast.addAnnounceHandler(self.handleAnnounce)
self.multicast.sendAnnounce(['elem1','elem2'])
# Wait for asynchronous handling
time.sleep(2.0)
for announce in self.announces:
if announce == ['123', 'elem1', 'elem2']:
return # Got it
raise Exception("Failed to get announce")
def handleAnnounce(self, permid, addr, list):
"""
Handle announce callback function
"""
self.announces.append([permid] + list)
def testCapabilities(self):
"""
Test capabilities thingy
"""
myCapabilities = ["Something", "something else", "something totally different"]
# Dummy config
config = {'permid':'testCapabilities',
'multicast_ipv4_address':'224.0.1.43',
'multicast_ipv6_address':'ff02::4124:1261:ffef',
'multicast_port':'6124',
'multicast_enabled':True,
'multicast_ipv4_enabled':False,
'multicast_ipv6_enabled':True,
'multicast_announce':True}
self.prepare_test(config, myCapabilities)
failed = True
for (permid, addr, capabilities) in self.multicast.discoverNodes():
if permid == config['permid']:
failed = False
if capabilities != myCapabilities:
raise Exception("Got bad capabilities, got %s, expected %s"%(str(capabilities), str(myCapabilities)))
if failed:
raise Exception("Didn't discover myself using IPv6")
