class TestCoordClass(unittest.TestCase): def setUp(self): """ initial setup""" self.net = Network() self.nc1 = NodeClass('simple_node_1', 0.1, 'coordbased', \ Sources(), self.net) self.nc2 = NodeClass('simple_node_2', 0.1, 'coordbased', \ Sources(), self.net) self.cc = CoordClass('coord', 0.1, 'coordbased', Sources(), self.net) self.li = [self.nc1, self.nc2, self.cc] self.nc1.node.protocol.initProtocol(self.li) self.nc2.node.protocol.initProtocol(self.li) self.cc.node.protocol.initProtocol(self.li) self.net.NodeList = self.li def testRun(self): """ tests the run method. """ self.cc.run() pass def testIncomingMsgs(self): """ tests the incomingMsgs method. """ # wrong test initialize() activate(self.cc, self.cc.incomingMsgs()) simulate(until=10) self.assertTrue(self.cc.busy_cpu) def test_sim(self): """ tests a real simple small simulation. """ initialize() for node in self.li: activate(node, node.run()) activate(node, node.incomingMsgs()) self.net.run() simulate(until=10) self.assertTrue(self.cc.node.protocol.NodeList)