Python NetworkOperator Examples

Example #1

File: test_network_operator.py Project: t2d/pyanotel

 def setUp(self):
     self.factory = NetworkOperator(Pager())
     self.protocol = NetworkOperatorProtocol()
     self.factory.protocol = self.protocol
     self.pseudonym_provider = SimpleProtocol()
     self.pseudonym_provider.transport = proto_helpers.StringTransport()
     self.factory.pseudonym_provider = self.pseudonym_provider

Example #2

File: test_network_operator.py Project: t2d/pyanotel

class TestCidsHandling(unittest.TestCase):
    def setUp(self):
        self.factory = NetworkOperator(Pager())
        self.protocol = NetworkOperatorProtocol()
        self.factory.protocol = self.protocol
        self.pseudonym_provider = SimpleProtocol()
        self.pseudonym_provider.transport = proto_helpers.StringTransport()
        self.factory.pseudonym_provider = self.pseudonym_provider

    def tearDown(self):
        self.factory.cleanup()

    def test_store_cid(self):
        cid = pick_random()
        self.factory.store_cid(cid)
        self.assertEqual(self.factory.cids, [cid])
        self.factory.connect_call(cid, self.protocol)
        self.assertEqual(self.factory.cids, [])