def setUp(self):
self.old_random = random.Random
random.Random = MonkeyRandom
# Create Switch, add controller, and connect to it
self.switch = create_switch(1, 2)
self.switch.use_delayed_commands()
self.switch.randomize_flow_mods()
self.configs = self.ControllerConfigs()
self.switch.add_controller_info(self.configs)
self.switch.connect(create_mock_connection)
self.conn = self.switch.get_connection(self.configs.cid)
def setUp(self):
self.old_random = random.Random
random.Random = MonkeyRandom
# Create Switch, add controller, and connect to it
self.switch = create_switch(1,2)
self.switch.use_delayed_commands()
self.switch.randomize_flow_mods()
self.configs = self.ControllerConfigs()
self.switch.add_controller_info(self.configs)
self.switch.connect(create_mock_connection)
self.conn = self.switch.get_connection(self.configs.cid)
def test_manual(self):
# Store the actual random.Random class
old_random = random.Random
random.Random = MonkeyRandom
# Create Switch, add controller, and connect to it
switch = create_switch(1, 2)
switch.use_delayed_commands()
switch.randomize_flow_mods()
configs = self.ControllerConfigs()
switch.add_controller_info(configs)
switch.connect(create_mock_connection)
conn = switch.get_connection(configs.cid)
# Create packets to send
flow_mod1 = ofp_flow_mod(match=ofp_match(in_port=1, nw_src="1.1.1.1"), action=ofp_action_output(port=1))
flow_mod2 = ofp_flow_mod(match=ofp_match(in_port=2, nw_src="2.2.2.2"), action=ofp_action_output(port=1))
flow_mod3 = ofp_flow_mod(match=ofp_match(in_port=3, nw_src="3.3.3.3"), action=ofp_action_output(port=1))
flow_mod4 = ofp_flow_mod(match=ofp_match(in_port=4, nw_src="4.4.4.4"), action=ofp_action_output(port=1))
other_packet = ofp_packet_out()
# Read packets into switch
conn.read(flow_mod1)
conn.read(flow_mod2)
conn.read(other_packet)
conn.read(flow_mod3)
conn.read(flow_mod4)
# Tell switch to process commands the way Fuzzer would
self.assertTrue(switch.has_pending_commands())
switch.process_delayed_command()
self.assertTrue(switch.has_pending_commands())
switch.process_delayed_command()
self.assertTrue(switch.has_pending_commands())
switch.process_delayed_command()
self.assertTrue(switch.has_pending_commands())
switch.process_delayed_command()
# Make sure flow mods were processed according to the generated weights
# IMPORTANT: Works off assumption switch's table stores entries in order they were processed
self.assertEqual(switch.table.table[0].match.in_port, flow_mod3.match.in_port)
self.assertEqual(switch.table.table[1].match.in_port, flow_mod2.match.in_port)
self.assertEqual(switch.table.table[2].match.in_port, flow_mod4.match.in_port)
self.assertEqual(switch.table.table[3].match.in_port, flow_mod1.match.in_port)
# Make sure other_packet wasn't put into queue
self.assertRaises(Queue.Empty, switch.process_delayed_command)
# Make sure nothing else got in the table.
self.assertRaises(IndexError, switch.table.table.__getitem__, 4)
# Restore legitimate random.Random class
random.Random = old_random
