U j@sVdZddlZddlmZmZddlmZddlmZGdddZ Gdd d eZ dS) z& Test code for basic Factory classes. N)ProtocolReconnectingClientFactory)Clock)TestCasec@s eZdZdZddZddZdS) FakeConnectorzP A fake connector class, to be used to mock connections failed or lost. cCsdSNselfrr?/tmp/pip-unpacked-wheel-j5h_c8_q/twisted/test/test_factories.pystopConnectingszFakeConnector.stopConnectingcCsdSrrr rrr connectszFakeConnector.connectN)__name__ __module__ __qualname____doc__r r rrrr rsrc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)ReconnectingFactoryTestsz1 Tests for L{ReconnectingClientFactory}. cCsJGddd}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 rrr r )sz`ReconnectingFactoryTests.test_stopTryingWhenConnected..NoConnectConnector.stopConnectingcSs tddS)NzShouldn't be reconnecting.rr rrr r ,szYReconnectingFactoryTests.test_stopTryingWhenConnected..NoConnectConnector.connectN)rrrr r rrrr NoConnectConnector(srN)rrprotocolZ buildProtocol stopTryingclientConnectionLost assertFalsecontinueTrying)r rcrrr 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 frr r AszjReconnectingFactoryTests.test_stopTryingDoesNotReconnect..FactoryAwareFakeConnector.stopConnectingcSs d|_dS)z| Record an attempt to reconnect, since this is what we are trying to avoid. TN)attemptedRetryr rrr r HszcReconnectingFactoryTests.test_stopTryingDoesNotReconnect..FactoryAwareFakeConnector.connectN)rrrr r r rrrr FactoryAwareFakeConnector>s r!N) rrrclock connectorrrr ZgetDelayedCalls)r r!rrr test_stopTryingDoesNotReconnect8sz8ReconnectingFactoryTests.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 reconstitutedrrr 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+rrr test_serializeWithClockds 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(retriesdelayZ initialDelay assertTruer)r factoryZ serializedZ unserializedrrr $test_deserializationResetsParametersos     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"r2rrr test_parametrizedClocks z/ReconnectingFactoryTests.test_parametrizedClockN) rrrrrr$r,r.r3r6rrrr rs#  r) rr%Ztwisted.internet.protocolrrZtwisted.internet.taskrZtwisted.trial.unittestrrrrrrr s