class TestControlProtocolUsage(unittest.TestCase):
def setUp(self):
self.cp = ControlProtocol()
self.cp.transport = DummyTransport()
def test_trivial(self):
pass
def test_bad_data(self):
l = [
"""{"hurp": "derp"}""",
]
for line in l:
self.cp.lineReceived(line)
self.assertTrue(self.cp.transport.buf.startswith("FAILED"),
"%s didn't cause an error" % line)
self.cp.transport.buf = ""
def test_privileged_port(self):
self.cp.factory = DummyControlFactory()
self.cp.lineReceived("""{
"sport": 1,
"dport": 11048,
"daddr": "localhost",
"password": "******"
}""")
self.assertTrue(self.cp.transport.buf.startswith("FAILED"))
def test_forwarding(self):
# Set up our test. We want to make sure that no actual ports are
# opened, so let's patch out listenTCP().
self.patch(reactor, "listenTCP",
lambda port, factory, **kwargs: Deferred())
self.cp.factory = DummyControlFactory()
self.cp.lineReceived("""{
"sport": 55555,
"dport": 11048,
"daddr": "localhost",
"password": "******"
}""")
self.assertTrue(self.cp.transport.buf.startswith("55555"))
# Clean up one single delayed call: The timeout.
delayed = reactor.getDelayedCalls()
self.assertEqual(len(delayed), 1)
delayed[0].cancel()
class TestControlProtocolUsage(unittest.TestCase):
def setUp(self):
self.cp = ControlProtocol()
self.cp.transport = DummyTransport()
def test_trivial(self):
pass
def test_bad_data(self):
l = [
"""{"hurp": "derp"}""",
]
for line in l:
self.cp.lineReceived(line)
self.assertTrue(self.cp.transport.buf.startswith("FAILED"),
"%s didn't cause an error" % line)
self.cp.transport.buf = ""
def test_privileged_port(self):
self.cp.factory = DummyControlFactory()
self.cp.lineReceived("""{
"sport": 1,
"dport": 11048,
"daddr": "localhost",
"password": "******"
}""")
self.assertTrue(self.cp.transport.buf.startswith("FAILED"))
def test_forwarding(self):
# Set up our test. We want to make sure that no actual ports are
# opened, so let's patch out listenTCP().
self.patch(reactor, "listenTCP",
lambda port, factory, **kwargs: Deferred())
self.cp.factory = DummyControlFactory()
self.cp.lineReceived("""{
"sport": 55555,
"dport": 11048,
"daddr": "localhost",
"password": "******"
}""")
self.assertTrue(self.cp.transport.buf.startswith("55555"))
# Clean up one single delayed call: The timeout.
delayed = reactor.getDelayedCalls()
self.assertEqual(len(delayed), 1)
delayed[0].cancel()
def setUp(self):
self.cp = ControlProtocol()
self.cp.transport = DummyTransport()
def setUp(self):
self.cp = ControlProtocol()
self.cp.transport = DummyTransport()
def setUp(self):
self.cp = ControlProtocol()
