Пример #1
0
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)