| Index: third_party/twisted_8_1/twisted/test/test_factories.py
|
| diff --git a/third_party/twisted_8_1/twisted/test/test_factories.py b/third_party/twisted_8_1/twisted/test/test_factories.py
|
| deleted file mode 100644
|
| index 63cf210669c281388784405cc8229313fca7a1c5..0000000000000000000000000000000000000000
|
| --- a/third_party/twisted_8_1/twisted/test/test_factories.py
|
| +++ /dev/null
|
| @@ -1,122 +0,0 @@
|
| -# Copyright (c) 2001-2008 Twisted Matrix Laboratories.
|
| -# See LICENSE for details.
|
| -
|
| -"""
|
| -Test code for basic Factory classes.
|
| -"""
|
| -
|
| -import pickle
|
| -
|
| -from twisted.trial import unittest
|
| -
|
| -from twisted.internet import reactor, defer
|
| -from twisted.internet.protocol import Factory, ReconnectingClientFactory
|
| -from twisted.protocols.basic import Int16StringReceiver
|
| -
|
| -class In(Int16StringReceiver):
|
| - def __init__(self):
|
| - self.msgs = {}
|
| -
|
| - def connectionMade(self):
|
| - self.factory.connections += 1
|
| -
|
| - def stringReceived(self, msg):
|
| - n, msg = pickle.loads(msg)
|
| - self.msgs[n] = msg
|
| - self.sendString(pickle.dumps(n))
|
| -
|
| - def connectionLost(self, reason):
|
| - self.factory.allMessages.append(self.msgs)
|
| - if len(self.factory.allMessages) >= self.factory.goal:
|
| - self.factory.d.callback(None)
|
| -
|
| -class Out(Int16StringReceiver):
|
| - msgs = dict([(x, 'X' * x) for x in range(10)])
|
| -
|
| - def __init__(self):
|
| - self.msgs = Out.msgs.copy()
|
| -
|
| - def connectionMade(self):
|
| - for i in self.msgs.keys():
|
| - self.sendString(pickle.dumps( (i, self.msgs[i])))
|
| -
|
| - def stringReceived(self, msg):
|
| - n = pickle.loads(msg)
|
| - del self.msgs[n]
|
| - if not self.msgs:
|
| - self.transport.loseConnection()
|
| - self.factory.howManyTimes -= 1
|
| - if self.factory.howManyTimes <= 0:
|
| - self.factory.stopTrying()
|
| -
|
| -
|
| -
|
| -class ReconnectingFactoryTestCase(unittest.TestCase):
|
| - """
|
| - Tests for L{ReconnectingClientFactory}.
|
| - """
|
| - def testStopTrying(self):
|
| - f = Factory()
|
| - f.protocol = In
|
| - f.connections = 0
|
| - f.allMessages = []
|
| - f.goal = 2
|
| - f.d = defer.Deferred()
|
| -
|
| - c = ReconnectingClientFactory()
|
| - c.initialDelay = c.delay = 0.2
|
| - c.protocol = Out
|
| - c.howManyTimes = 2
|
| -
|
| - port = reactor.listenTCP(0, f)
|
| - self.addCleanup(port.stopListening)
|
| - PORT = port.getHost().port
|
| - reactor.connectTCP('127.0.0.1', PORT, c)
|
| -
|
| - f.d.addCallback(self._testStopTrying_1, f, c)
|
| - return f.d
|
| - testStopTrying.timeout = 10
|
| -
|
| -
|
| - def _testStopTrying_1(self, res, f, c):
|
| - self.assertEquals(len(f.allMessages), 2,
|
| - "not enough messages -- %s" % f.allMessages)
|
| - self.assertEquals(f.connections, 2,
|
| - "Number of successful connections incorrect %d" %
|
| - f.connections)
|
| - self.assertEquals(f.allMessages, [Out.msgs] * 2)
|
| - self.failIf(c.continueTrying, "stopTrying never called or ineffective")
|
| -
|
| -
|
| - def test_serializeUnused(self):
|
| - """
|
| - A L{ReconnectingClientFactory} which hasn't been used for anything
|
| - can be pickled and unpickled and end up with the same state.
|
| - """
|
| - original = ReconnectingClientFactory()
|
| - reconstituted = pickle.loads(pickle.dumps(original))
|
| - self.assertEqual(original.__dict__, reconstituted.__dict__)
|
| -
|
| -
|
| - def test_deserializationResetsParameters(self):
|
| - """
|
| - A L{ReconnectingClientFactory} which is unpickled does not have an
|
| - L{IConnector} and has its reconnectioning timing parameters reset to
|
| - their initial values.
|
| - """
|
| - class FakeConnector(object):
|
| - def stopConnecting(self):
|
| - pass
|
| -
|
| - factory = ReconnectingClientFactory()
|
| - factory.clientConnectionFailed(FakeConnector(), None)
|
| - try:
|
| - serialized = pickle.dumps(factory)
|
| - unserialized = pickle.loads(serialized)
|
| - self.assertEqual(unserialized.connector, None)
|
| - self.assertEqual(unserialized._callID, None)
|
| - self.assertEqual(unserialized.retries, 0)
|
| - self.assertEqual(unserialized.delay, factory.initialDelay)
|
| - self.assertEqual(unserialized.continueTrying, True)
|
| - finally:
|
| - factory.stopTrying()
|
|
|