class ProxyPacketHandlingTests(unittest.TestCase): def setUp(self): self.proxy = Proxy() self.proxy.hosts["host"] = TrivialObject() self.proxy.hosts["host"].secret = "supersecret" self.packet = TrivialObject() self.packet.code = AccessAccept self.packet.source = ("host", "port") def testHandleProxyPacketUnknownHost(self): self.packet.source = ("stranger", "port") try: self.proxy._HandleProxyPacket(self.packet) except ServerPacketError as e: self.failUnless("unknown host" in str(e)) else: self.fail() def testHandleProxyPacketSetsSecret(self): self.proxy._HandleProxyPacket(self.packet) self.assertEqual(self.packet.secret, "supersecret") def testHandleProxyPacketHandlesWrongPacket(self): self.packet.code = AccessRequest try: self.proxy._HandleProxyPacket(self.packet) except ServerPacketError as e: self.failUnless("non-response" in str(e)) else: self.fail()
class ProxyPacketHandlingTests(unittest.TestCase): def setUp(self): self.proxy = Proxy() self.proxy.hosts['host'] = TrivialObject() self.proxy.hosts['host'].secret = 'supersecret' self.packet = TrivialObject() self.packet.code = AccessAccept self.packet.source = ('host', 'port') def testHandleProxyPacketUnknownHost(self): self.packet.source = ('stranger', 'port') try: self.proxy._HandleProxyPacket(self.packet) except ServerPacketError as e: self.failUnless('unknown host' in str(e)) else: self.fail() def testHandleProxyPacketSetsSecret(self): self.proxy._HandleProxyPacket(self.packet) self.assertEqual(self.packet.secret, 'supersecret') def testHandleProxyPacketHandlesWrongPacket(self): self.packet.code = AccessRequest try: self.proxy._HandleProxyPacket(self.packet) except ServerPacketError as e: self.failUnless('non-response' in str(e)) else: self.fail()