def setUp(self): self.mac = Workstation(name="mac") self.sun = Workstation(name="sun", next_node=self.mac) self.link = Link(name="link", next_node=self.sun) self.printer = Printer(name="printer", next_node=self.link) self.fileserver = Fileserver(name="fileserver", next_node=self.printer) self.pc = Workstation(name="pc", next_node=self.fileserver) self.mac.next_node = self.pc
class LanTest(unittest.TestCase): def setUp(self): self.mac = Workstation(name="mac") self.sun = Workstation(name="sun", next_node=self.mac) self.link = Link(name="link", next_node=self.sun) self.printer = Printer(name="printer", next_node=self.link) self.fileserver = Fileserver(name="fileserver", next_node=self.printer) self.pc = Workstation(name="pc", next_node=self.fileserver) self.mac.next_node = self.pc def test_good_receiver_packet(self): p = Packet(content="x", receiver="sun") self.mac.originate(p) self.assertEquals(self.sun.last_packet, p) def test_good_path_packet(self): p = Packet(content="x", receiver="sun") self.mac.originate(p) self.assertEquals(self.pc.last_packet, p) def test_loop_in_lan(self): # samo to, że ta funkcja działa świadczy o # tym, że pętle w sieci są obsługiwane p = Packet(content="x", receiver="nobody") self.mac.originate(p) self.assertEquals(p._hopcounter, Packet.MAX_HOP)
def test_originate_packet(self): w1 = Workstation(name="w1") p = Packet(content="X", receiver="w2") w1.originate(p) self.assertEquals(p.sender, w1.name)