def setUp(self):
self.node_a = Node(node_id="1")
self.node_b = Node(node_id="2")
self.node_c = Node(node_id="3")
self.few_nodes = {}
self.many_nodes = {}
few_nodes_list = Node.generate_nodes(3, Node, node_parameters={}, protocol_manager=None)
for node in few_nodes_list:
self.few_nodes[node.id] = node
many_nodes_list = Node.generate_nodes(300, Node, node_parameters={}, protocol_manager=None)
for node in many_nodes_list:
self.many_nodes[node.id] = node
self.plane_1 = Plane(10, 20)
self.plane_2 = Plane(30, 30)
def __init__(self, simulation_parameters):
"""
Constructor.
simulation_parameters - The parameters to be used to run this simulation
"""
self.type_of_nodes = simulation_parameters.config_p['type_of_nodes']
self._max_cycles = simulation_parameters.config_p['max_cycles']
self.number_messages_out = 0
self.number_messages_in = 0
self.number_asynchronous_events = 0
self.number_simulation_events = 0
self.all_nodes = {}
self.time = 0
self.protocol_manager = \
simulation_parameters.config_p['protocol_manager'](protocol_arguments=simulation_parameters.protocol_p)
self.event_broker = EventBroker()
self.message_broker = MessageBroker()
node_parameters = simulation_parameters.node_p
type_of_nodes = simulation_parameters.config_p['type_of_nodes']
events_to_launch = simulation_parameters.config_p['events']
# If the simulator is generating the plane
if not 'scenario_file' in simulation_parameters.plane_p:
generated_nodes = Node.generate_nodes(number_to_generate=simulation_parameters.plane_p['number_of_nodes'], \
type_of_nodes=type_of_nodes, node_parameters=node_parameters, \
protocol_manager=self.protocol_manager)
for node in generated_nodes:
self.all_nodes[node.id] = node
self.plane = Plane.generate_plane(all_nodes=self.all_nodes, plane_parameters=simulation_parameters.plane_p)
# If the simulator is loading the plane from a file
else:
(self.all_nodes, self.plane) = Plane.load_plane(plane_parameters=simulation_parameters.plane_p, \
node_parameters=simulation_parameters.node_p, \
type_of_nodes=self.type_of_nodes, \
protocol_manager=self.protocol_manager)
#Now putt the events to launch in the event broker
for event in events_to_launch:
self.event_broker.add_asynchronous_event(event)
def test_node_generation(self):
generated_nodes = Node.generate_nodes(20, Node, protocol_manager=None)
self.assertEqual(len(generated_nodes), 20)
for n in generated_nodes:
self.assert_(isinstance(n, Node))
