class TestNetworkOperator(unittest.TestCase): def setUp(self): self.pager = Pager() self.transport = proto_helpers.FakeDatagramTransport() self.pager.transport = self.transport def test_location_announcement(self): self.pager.startProtocol() self.pager.location_announcement() self.assertEqual(len(self.transport.written), 2) packet_serialized, addr = self.transport.written[0] packet = deserialize(packet_serialized) self.assertEqual(packet[0], constants.Header.LOCATIONANNOUNCEMENT)
def setUp(self): self.pager = Pager() self.transport = proto_helpers.FakeDatagramTransport() self.pager.transport = self.transport