o b@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. cCdSNselfr r =/usr/lib/python3/dist-packages/twisted/test/test_factories.pystopConnectingzFakeConnector.stopConnectingcCrrr r r r r connectrzFakeConnector.connectN)__name__ __module__ __qualname____doc__r rr r r r rs rc@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..NoConnectConnectorcStd)Nz%Shouldn't be called, we're connected. RuntimeErrorr r r r r )z`ReconnectingFactoryTests.test_stopTryingWhenConnected..NoConnectConnector.stopConnectingcSr)NzShouldn't be reconnecting.rr r r r r,rzYReconnectingFactoryTests.test_stopTryingWhenConnected..NoConnectConnector.connectN)rrrr rr r r r NoConnectConnector(s rN)rrprotocol 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 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 r r r rHs zcReconnectingFactoryTests.test_stopTryingDoesNotReconnect..FactoryAwareFakeConnector.connectN)rrrr%r rr r#r r FactoryAwareFakeConnector>s  r&N) rrrclock connectorrrr%getDelayedCalls)r r&r r#r 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 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'r0r1r r r 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)rr"r addCleanuprr+r-r,r3r(_callIDr.retriesdelay initialDelay assertTruer)r factory serialized unserializedr r r $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.lencalls)r r'r;r r r test_parametrizedClocks z/ReconnectingFactoryTests.test_parametrizedClockN) rrrrr!r*r2r4r>rBr r r r rs#  r) rr+twisted.internet.protocolrrtwisted.internet.taskrtwisted.trial.unittestrrrr r r r s