U  W[@shdZddlmZmZddlZddlmZddlmZddl m Z m Z Gddde Z Gd d d eZdS) z& Test code for basic Factory classes. )divisionabsolute_importN)TestCase)Clock)ReconnectingClientFactoryProtocolc@s eZdZdZddZddZdS) FakeConnectorzP A fake connector class, to be used to mock connections failed or lost. cCsdSNselfr r =/usr/lib/python3/dist-packages/twisted/test/test_factories.pystopConnectingszFakeConnector.stopConnectingcCsdSr r r r r r connectszFakeConnector.connectN)__name__ __module__ __qualname____doc__rrr r r r rsrc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)ReconnectingFactoryTestsz1 Tests for L{ReconnectingClientFactory}. cCsLGdddt}t}t|_|d|||d||jdS)z If a L{ReconnectingClientFactory} has C{stopTrying} called while it is connected, it does not subsequently attempt to reconnect if the connection is later lost. c@seZdZddZddZdS)zQReconnectingFactoryTests.test_stopTryingWhenConnected..NoConnectConnectorcSs tddS)Nz%Shouldn't be called, we're connected. RuntimeErrorr r r r r,sz`ReconnectingFactoryTests.test_stopTryingWhenConnected..NoConnectConnector.stopConnectingcSs tddS)NzShouldn't be reconnecting.rr r r r r.szYReconnectingFactoryTests.test_stopTryingWhenConnected..NoConnectConnector.connectN)rrrrrr r r r NoConnectConnector+srN) objectrrZprotocolZ buildProtocol stopTryingclientConnectionLost assertFalsecontinueTrying)r rcr r r test_stopTryingWhenConnected%s z5ReconnectingFactoryTests.test_stopTryingWhenConnectedcsTGfdddt}tt_|_|jj|jdS)z Calling stopTrying on a L{ReconnectingClientFactory} doesn't attempt a retry on any active connector. cs$eZdZdZfddZddZdS)z[ReconnectingFactoryTests.test_stopTryingDoesNotReconnect..FactoryAwareFakeConnectorFcs|ddS)z Behave as though an ongoing connection attempt has now failed, and notify the factory of this. N)clientConnectionFailedr fr r rCszjReconnectingFactoryTests.test_stopTryingDoesNotReconnect..FactoryAwareFakeConnector.stopConnectingcSs d|_dS)z| Record an attempt to reconnect, since this is what we are trying to avoid. TN)attemptedRetryr r r r rJszcReconnectingFactoryTests.test_stopTryingDoesNotReconnect..FactoryAwareFakeConnector.connectN)rrrr"rrr r r r FactoryAwareFakeConnector@s r#N) rrrclock connectorrrr"ZgetDelayedCalls)r r#r r r test_stopTryingDoesNotReconnect;sz8ReconnectingFactoryTests.test_stopTryingDoesNotReconnectcCs*t}tt|}||j|jdS)z A L{ReconnectingClientFactory} which hasn't been used for anything can be pickled and unpickled and end up with the same state. N)rpickleloadsdumps assertEqual__dict__)r original reconstitutedr r r test_serializeUnused^sz-ReconnectingFactoryTests.test_serializeUnusedcCs2t}t}||_tt|}||jdS)z The clock attribute of L{ReconnectingClientFactory} is not serialized, and the restored value sets it to the default value, the reactor. N)rrr$r'r(r) assertIsNone)r r$r,r-r r r test_serializeWithClockhs z0ReconnectingFactoryTests.test_serializeWithClockcCszt}|td||jt|}t|}||j ||j | |j d| |j |j||jdS)z A L{ReconnectingClientFactory} which is unpickled does not have an L{IConnector} and has its reconnecting timing parameters reset to their initial values. Nr)rrrZ addCleanuprr'r)r(r/r%Z_callIDr*ZretriesZdelayZ initialDelayZ assertTruer)r factoryZ serializedZ unserializedr r r $test_deserializationResetsParametersts     z=ReconnectingFactoryTests.test_deserializationResetsParameterscCs6t}t}||_|td|t|jddS)z The clock used by L{ReconnectingClientFactory} can be parametrized, so that one can cleanly test reconnections. N)rrr$rrr*lenZcalls)r r$r1r r r test_parametrizedClocks z/ReconnectingFactoryTests.test_parametrizedClockN) rrrrrr&r.r0r2r5r r r r r s#  r)rZ __future__rrr'Ztwisted.trial.unittestrZtwisted.internet.taskrZtwisted.internet.protocolrrrrrr r r r s