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()
