0x1949 Team - FAZEMRX - MANAGER
Edit File: test_amp.cpython-310.pyc
o �b�� � @ s@ d Z ddlZddlZddlmZmZ ddlmZ ddlm Z ddl m Z m Z ddl mZmZmZmZmZmZ ddlmZ dd lmZ dd lmZ dd lmZ dd l m Z dd l m Z zddl m Z! W n e"ym dZ Y n w e!j#stdZ ne!Z e du r}dZ$ndZ$ej%�&e�s�dZ'ndZ'ej(j)Z*G dd� dej+�Z,G dd� dej-�Z.G dd� de/�Z0G dd� de/�Z1G dd� de/�Z2G dd � d ej3�Z4G d d � d ej5�Z6G d d � d e6�Z7G d!d"� d"ej5�Z8G d#d$� d$ej5�Z9G d%d&� d&ej5�Z:G d'd(� d(ej5�Z;G d)d*� d*ej5�Z<G d+d,� d,ej5�Z=G d-d.� d.ej5�Z>G d/d0� d0ej5�Z?G d1d2� d2ej@�ZAG d3d4� d4ejB�ZCd5ZDG d6d7� d7e/�ZEG d8d9� d9ej-�ZFG d:d;� d;eF�ZGG d<d=� d=eG�ZHG d>d?� d?eG�ZIG d@dA� dAeG�ZJe.e.fdBdC�ZKG dDdE� dEej+�ZLG dFdG� dGej-�ZMG dHdI� dIe �ZNG dJdK� dKe �ZOG dLdM� dM�ZPG dNdO� dO�ZQG dPdQ� dQe �ZRG dRdS� dSej5�ZSG dTdU� dUejT�ZUG dVdW� dWeU�ZVG dXdY� dYeV�ZWG dZd[� d[ej-�ZXG d\d]� d]e �ZYd^ZZd_Z[G d`da� dae �Z\G dbdc� dce �Z]G ddde� de�Z^G dfdg� dg�Z_G dhdi� die_�Z`G djdk� dke_�ZaG dldm� dme_�ZbG dndo� doeF�Zcee$dp�ee'dq�G drds� dse ���ZdG dtdu� due �ZeG dvdw� dwe/�ZfG dxdy� dye/�ZgG dzd{� d{ej5�ZhG d|d}� d}eh�ZiG d~d� deh�ZjG d�d�� d�ej-�ZkG d�d�� d�ej-�ZlG d�d�� d�ej-�ZmG d�d�� d�e �Znd�d�� ZoG d�d�� d��Zpd�d�� Zqd�d�� Zre du�r�er� Zsee$dp�ee'dq�G d�d�� d�epe ���ZtG d�d�� d�eG�Zuee$dp�ee'dq�G d�d�� d�epe ���Zvee$dp�ee'dq�G d�d�� d�epe ���ZwG d�d�� d�ej3�ZxG d�d�� d�ej5�ZyG d�d�� d�ej5�ZzG d�d�� d�ej-�Z{G d�d�� d�e|�Z}G d�d�� d�ey�Z~G d�d�� d�e �ZG d�d�� d��Z�G d�d�� d�e e��Z�G d�d�� d�e e��Z�G d�d�� d�e e��Z�G d�d�� d�e e��Z�G d�d�� d�e e��Z�G d�d�� d�e �Z�G d�d�� d�e �Z�G d�d�� d�e e��Z�G d�d�� d�e �Z�e ej��G d�d�� d���Z�e ej�e�� e ej�e�� G d�d�� d�e �Z�G d�d�� d�e �Z�G d�d�� d�e �Z�G d�dÄ d�e �Z�dS )�z% Tests for L{twisted.protocols.amp}. � N)�Dict�Type)�skipIf)� implementer)� verifyClass� verifyObject)�address�defer�error� interfaces�protocol�reactor)�amp)�filepath)�Failure)�iosim)�StringTransport)�TestCase��sslTFc @ sB e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd e fd d �Z dS )� TestProtoz� A trivial protocol for use in testing where a L{Protocol} is expected. @ivar instanceId: the id of this instance @ivar onConnLost: deferred that will fired when the connection is lost @ivar dataToSend: data to send on the protocol r c C s: t |t�s J t|���|| _|| _tj| _tjd t_d S )N� )� isinstance�bytes�repr� onConnLost� dataToSendr � instanceCount� instanceId)�selfr r � r �7/usr/lib/python3/dist-packages/twisted/test/test_amp.py�__init__= s zTestProto.__init__c C s g | _ | j�| j� d S �N)�data� transport�writer �r r r r! �connectionMadeD s zTestProto.connectionMadec C � | j �|� d S r# )r$ �append)r r r r r! � dataReceivedH � zTestProto.dataReceivedc C s | j �| j� d S r# )r �callbackr$ �r �reasonr r r! �connectionLostK � zTestProto.connectionLost�returnc C s d| j f S )z� Custom repr for testing to avoid coupling amp tests with repr from L{Protocol} Returns a string which contains a unique identifier that can be looked up using the instanceId property:: <TestProto #3> z<TestProto #%d>)r r' r r r! �__repr__N s zTestProto.__repr__N) �__name__� __module__� __qualname__�__doc__r r" r( r+ r0 �strr3 r r r r! r 2 s r c @ � e Zd Zdd� Zdd� ZdS )�SimpleSymmetricProtocolc C s | j d|d�S �N� hello��hello)�callRemoteString�r �textr r r! � sendHello\ � z!SimpleSymmetricProtocol.sendHelloc C s t j|d d�S r; �r �Box�r �boxr r r! � amp_HELLO_ r, z!SimpleSymmetricProtocol.amp_HELLON)r4 r5 r6 rB rH r r r r! r: [ � r: c @ � e Zd ZdZdS )�UnfriendlyGreeting�!Greeting was insufficiently kind.N�r4 r5 r6 r7 r r r r! rK c � rK c @ rJ )� DeathThreatrL NrM r r r r! rO g rN rO c @ rJ )�UnknownProtocolz&Asked to switch to the wrong protocol.NrM r r r r! rP k rN rP c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TransportPeerc C � dS )N� r )r �d�name�protor r r! �retriever � zTransportPeer.retrievec C s |j �� S r# )r% �getPeer)r � notAStringrV r r r! �fromStringProtou � z TransportPeer.fromStringProtoc C � d S r# r )r rU �strings�objectsrV r r r! �toBoxx rX zTransportPeer.toBoxN)r4 r5 r6 rW r[ r` r r r r! rQ o s rQ c @ s� e Zd ZU dZde�� fdejdd�fdejdd�fdedd�fdejdd�fdejdd�fd ejdd�fgZ de�� fdejdd�fgZ e d iZ e ee ef ed < ed iZe ee ef ed < dS )�Hellor< � optionalT��optional� print� from� mixedCase� dash-arg� underscore_arg� UNFRIENDLY�errorss DEAD� fatalErrorsN)r4 r5 r6 � commandNamer �String�Boolean�UnicoderQ � arguments�responserK rk r r � Exceptionr �__annotations__rO rl r r r r! ra | s � ra c @ s e Zd ZejZdZdS )� NoAnswerHelloFN)r4 r5 r6 ra rm �requiresAnswerr r r r! ru � s ru c @ sx e Zd ZdZde�� fdejdd�fdejdd�fdedd�fdejdd�fgZ de�� fdejdd�fgZ e diZ d S ) � FutureHellor< rb Trc re rf s bonusrj N) r4 r5 r6 rm r rn ro rp rQ rq rr rK rk r r r r! rw � s � rw c @ rJ )�WTFz+ An example of an invalid command. NrM r r r r! rx � rN rx c @ s e Zd ZdZdZdS )� BrokenReturnz[An example of a perfectly good command, but the handler is going to return None... s broken_returnN)r4 r5 r6 r7 rm r r r r! ry � � ry c @ s e Zd Zde�� fgZejZdS )�Goodbyes goodbyeN)r4 r5 r6 r rn rr �QuitBox� responseTyper r r r! r{ � s r{ c @ rJ )� WaitForevers wait_foreverN)r4 r5 r6 rm r r r r! r~ � rN r~ c @ s8 e Zd ZdZde�� fgZde�de�� fg�fgZdS )�GetLists getlists length� body� xN) r4 r5 r6 rm r �Integerrq �AmpListrr r r r r! r � s r c @ sF e Zd ZdZde�� fdejde�� fgdd�fgZde�� fgZdS ) � DontRejectMes dontrejectmes magicWords list� nameTrc s responseN) r4 r5 r6 rm r rp r� rq rr r r r r! r� � s �r� c @ s e Zd Zde�� fgZdS )� SecuredPings pingedN)r4 r5 r6 r ro rr r r r r! r� � s r� c @ s& e Zd ZdZde�� fgZediZdS )�TestSwitchProtos Switch-Protor� s UNKNOWNN) r4 r5 r6 rm r rn rq rP rk r r r r! r� � s � r� c @ s( e Zd Zdd� Zdd� ZdZdd� ZdS )�SingleUseFactoryc C s || _ | | j _d S r# )rV �factory�r rV r r r! r" � s zSingleUseFactory.__init__c C s | j d }| _ |S r# �rV )r �addr�pr r r! � buildProtocol� � z SingleUseFactory.buildProtocolNc C � || _ d S r# )� reasonFailed�r � connectorr/ r r r! �clientConnectionFailed� s z'SingleUseFactory.clientConnectionFailed)r4 r5 r6 r" r� r� r� r r r r! r� � s r� s gwebol nargoc @ s e Zd ZdS )�ThingIDontUnderstandErrorN)r4 r5 r6 r r r r! r� � s r� c @ s* e Zd ZdZdd� Zdd� Ze�e� dS )�FactoryNotifierNc C s8 | j d ur| | j _t| j d�r| j j�d � d S d S d S )N�onMade)r� �theProto�hasattrr� r- r' r r r! r( � s �z FactoryNotifier.connectionMadec C s( ddl m} |�| j�std��ddiS )Nr )� ISSLTransportz+only send secure pings over secure channels�pingedT)�twisted.internet.interfacesr� � providedByr% rO )r r� r r r! �emitpong� s zFactoryNotifier.emitpong)r4 r5 r6 r� r( r� r� � responderr r r r! r� � s r� c @ s� e Zd ZdZddd�Zdd� Zdd� ZdZ dd d �Ze � e� d d � Z e � e � dd d�Z e� e � dd� Ze� e� dd� Ze� e� ddd�Zdd� Ze� e� dd� Ze� e� dS ) � SimpleSymmetricCommandProtocolNc C s t j�| � || _d S r# )r �AMPr" r )r r r r r! r" s z'SimpleSymmetricCommandProtocol.__init__c C s | j t|d�S )Nr= �� callRemotera r@ r r r! rB rC z(SimpleSymmetricCommandProtocol.sendHelloc C s | j t||d�S )N�r> �Printr� )r rA � translationr r r! �sendUnicodeHello r, z/SimpleSymmetricCommandProtocol.sendUnicodeHelloFc C sn || j �� ks J �|tkrt� �|�d�rtd��|dkr!td��t|d�}|d ur2|�t|d�� d| _ |S )Ns fuck�Don't be a dick.� die� aieeeeeeeeer= )r� T) r% rY �THING_I_DONT_UNDERSTANDr� � startswithrK rO �dict�update�greeted) r r> �Fromrd r� � mixedCase�dash_arg�underscore_arg�resultr r r! �cmdHello s z'SimpleSymmetricCommandProtocol.cmdHelloc C s dt dd�g| iS )N�bodyr ��x�r� )r �lengthr r r! � cmdGetlist0 � z)SimpleSymmetricCommandProtocol.cmdGetlistc C s( |d u rd}nd|d d }t |d�S )N� list omittedz %s acceptedr rU )rr r� )r � magicWord�listrr r r r! �okiwont5 s z&SimpleSymmetricCommandProtocol.okiwontc C s t �� | _| jS r# )r �Deferred�waitingr' r r r! � waitforit> s z(SimpleSymmetricCommandProtocol.waitforitc C s t dd�S )N� everyone)�goodbyer� r' r r r! �saybyeD r\ z%SimpleSymmetricCommandProtocol.saybyec s< |rd}nd}t | jt�� | jtt� �|d��� fdd��S )Ns no-proto� test-proto)rU c � � S r# r ��ign�r� r r! �<lambda>Q � zESimpleSymmetricCommandProtocol.switchToTestProtocol.<locals>.<lambda>)r r �SWITCH_CLIENT_DATAr� r� r� � addCallback)r �failrU r r� r! �switchToTestProtocolI s ��z3SimpleSymmetricCommandProtocol.switchToTestProtocolc C s |dkr t | jt�S t|���Nr� )r r �SWITCH_SERVER_DATArP �r rU r r r! �switchitS s z'SimpleSymmetricCommandProtocol.switchitc C r] r# r r' r r r! � donothingZ rX z(SimpleSymmetricCommandProtocol.donothingr# �NNNNN)F)r4 r5 r6 � maybeLaterr" rB r� r� r� ra r� r� r r� r� r� r~ r� r{ r� r� r� r� ry r r r r! r� s4 � r� c @ s e Zd Zdd� Ze�e� dS )� DeferredSymmetricCommandProtocolc C s* |dkrt | jt�| _t�� | _| jS d S r� )r r r� �maybeLaterProtor r� r� r� r r r! r� a s �z)DeferredSymmetricCommandProtocol.switchitN)r4 r5 r6 r� r� r� r r r r! r� ` s r� c @ �* e Zd Z ddd�Ze�e� dS )�BadNoAnswerCommandProtocolNc C rR )zQ This responder does nothing and forgets to return a dictionary. Nr �r r> r� rd r� r� r� r� r r r! � badResponderk r� z'BadNoAnswerCommandProtocol.badResponderr� )r4 r5 r6 r� ru r� r r r r! r� j s �r� c @ r� )�NoAnswerCommandProtocolNc C s t |d d�S )Ns -noanswerr= r� r� r r r! �goodNoAnswerResponder} s z-NoAnswerCommandProtocol.goodNoAnswerResponderr� )r4 r5 r6 r� ru r� r r r r! r� | s � r� c O s t j| |g|�R i |��S )z)Returns a 3-tuple: (client, server, pump))r �connectedServerAndClient)� ServerClass� ClientClass�a�kwr r r! r� � � r� c @ � e Zd ZdZdd� ZdS )�TotallyDumbProtocolrS c C s | j |7 _ d S r# )�buf�r r$ r r r! r+ � r1 z TotallyDumbProtocol.dataReceivedN)r4 r5 r6 r� r+ r r r r! r� � � r� c @ r9 )� LiteralAmpc C � g | _ d S r# )�boxesr' r r r! r" � r\ zLiteralAmp.__init__c C r) r# �r� r* rF r r r! �ampBoxReceived� � zLiteralAmp.ampBoxReceivedN)r4 r5 r6 r" r� r r r r! r� � rI r� c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) � AmpBoxTestszm Test a few essential properties of AMP boxes, mostly with respect to serialization correctness. c C s$ t jdd�}| �t|�� �t� dS )z8 Make sure that strs serialize to strs. s value��keyN)r �AmpBox� assertEqual�type� serializer �r r� r r r! �test_serializeStr� s z AmpBoxTests.test_serializeStrc C s& t jdi ddi��}| �t|j� dS )zX Verify that TypeError is raised when trying to serialize Unicode keys. r� �valueNr �r r� � assertRaises� TypeErrorr� r� r r r! � test_serializeUnicodeKeyRaises� s z*AmpBoxTests.test_serializeUnicodeKeyRaisesc C s t jdd�}| �t|j� dS )zb Verify that TypeError is raised when trying to serialize Unicode values. r� r� Nr� r� r r r! � test_serializeUnicodeValueRaises� s z,AmpBoxTests.test_serializeUnicodeValueRaisesN)r4 r5 r6 r7 r� r r r r r r! r� � s r� c @ �, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � ParsingTestsc C s� t �� }| �|�d�� | �|�d�� | �t|jd� | �t|jd� | �t|jd� | �|�d�d� | �|�d�d� dS ) ze Verify that the Boolean parser parses 'True' and 'False', but nothing else. s Trues Falses ninjas trues TRUETFN) r ro � assertTrue� fromString� assertFalser� r� r� �toString)r �br r r! �test_booleanValues� s z ParsingTests.test_booleanValuesc C sF t �| �� �}t�� }|�|�}|�|�}| �||� | �||� dS )zL Verify the 'Path' argument can parse and emit a file path. N) r �FilePath�mktempr �Pathr r � assertIsNotr� )r �fpr� �s�vr r r! �test_pathValueRoundTrip� s z$ParsingTests.test_pathValueRoundTripc C s t �� }| �t j|jt �� �S )zk Test that empty boxes raise an error; they aren't supposed to be sent on purpose. )r r� r� � NoEmptyBoxesr� rE r� r r r! �test_sillyEmptyThing� s z!ParsingTests.test_sillyEmptyThingc C s� t ttd�\}}}d}d}d}d}d}d} d} |g|| g||g||g||||g|||g|||| g| ||gg} | D ] } t�� } | �t| �� | �|� |�� | �|j d | � q8d S ) zq Verify that various kinds of data make it through the encode/parse round-trip unharmed. )r� r� )� simples test)s ceqs : )s crtests test )s lftests hello )s newlines test one two)s newline2s test one two)r� s blah testtest���N) r� r� r rE r� r� �_sendTo�flushr� r� )r �cr r� �SIMPLE�CE�CR�LF�NEWLINE�NEWLINE2�BODYTEST�testData�test�jbr r r! �test_ParsingRoundTrip� s4 � � �z"ParsingTests.test_ParsingRoundTripN)r4 r5 r6 r r r r# r r r r! r � s r c @ � e Zd ZdZdd� Zdd� ZdS )� FakeLocatorzZ This is a fake implementation of the interface implied by L{CommandLocator}. c C s i | _ dS )zN Remember the given keyword arguments as a set of responders. N��commandsr' r r r! r" � zFakeLocator.__init__c C s | j | S )zz Look up and return a function passed as a keyword argument of the given name to the constructor. r& )r rm r r r! �locateResponder � zFakeLocator.locateResponderN)r4 r5 r6 r7 r" r) r r r r! r% s r% c @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � FakeSenderz\ This is a fake implementation of the 'box sender' interface implied by L{AMP}. c C s g | _ g | _d| _dS )zn Create a fake sender and initialize the list of received boxes and unhandled errors. r N)� sentBoxes�unhandledErrors�expectedErrorsr' r r r! r" s zFakeSender.__init__c C � | j d7 _ dS )zB Expect one error, so that the test doesn't fail. r N)r/ r' r r r! � expectError) s zFakeSender.expectErrorc C � | j �|� dS )z6 Accept a box, but don't do anything. N)r- r* rF r r r! �sendBox/ � zFakeSender.sendBoxc C s4 | j d8 _ | j dk r|�� dS | j�|� dS )zW Deal with failures by instantly re-raising them for easier debugging. r r N)r/ �raiseExceptionr. r* )r �failurer r r! �unhandledError5 s zFakeSender.unhandledErrorN)r4 r5 r6 r7 r" r1 r3 r7 r r r r! r, s r, c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dS )�CommandDispatchTestsa� The AMP CommandDispatcher class dispatches converts AMP boxes into commands and responses using Command.responder decorator. Note: Originally, AMP's factoring was such that many tests for this functionality are now implemented as full round-trip tests in L{AMPTests}. Future tests should be written at this level instead, to ensure API compatibility and to provide more granular, readable units of test coverage. c C s0 t � | _t� | _t�| j�| _| j�| j� dS )z- Create a dispatcher to use. N)r% �locatorr, �senderr � BoxDispatcher� dispatcher�startReceivingBoxesr' r r r! �setUpL s zCommandDispatchTests.setUpc sJ g � � fdd�}t jdddd�}|| jjd< | j�|� | �� |g� dS )z� L{CommandDispatcher.ampBoxReceived} should locate the appropriate command in its responder lookup, based on the '_ask' key. c s � � | � t�ddi�S )Nr> r� )r* r rE )rG ��receivedr r! �thunk\ � z4CommandDispatchTests.test_receivedAsk.<locals>.thunkr> �test-command-id�world��_command�_askr> N)r rE r9 r' r< r� r� �r rA �inputr r? r! �test_receivedAskU s z%CommandDispatchTests.test_receivedAskc C sP t d�}| j�� | j�t|�� | �t| jj�d� | �| jjd j |� dS )zy L{CommandDispatcher} should relay its unhandled errors in responding to boxes to its boxSender. zsomething went wrong, oh nor r N) � RuntimeErrorr: r1 r< r7 r r� �lenr. r� )r �errr r r! �test_sendUnhandledErrore s z,CommandDispatchTests.test_sendUnhandledErrorc st t d�� � fdd�}|| jjd< tjdddd�}| j�� | j�|� | � t | jj �d� | � | jj d j � � d S ) zp Errors during serialization ought to be relayed to the sender's unhandledError method. z something undefined went wrongc s G � fdd�dt j�}|� S )Nc s e Zd Z� fdd�ZdS )zWCommandDispatchTests.test_unhandledSerializationError.<locals>.thunk.<locals>.BrokenBoxc s � �r# r r� �rM r r! r y rX z_CommandDispatchTests.test_unhandledSerializationError.<locals>.thunk.<locals>.BrokenBox._sendToN)r4 r5 r6 r r rO r r! � BrokenBoxx s rP rD )r� rP rO r r! rA w s zDCommandDispatchTests.test_unhandledSerializationError.<locals>.thunkr> rC rD rE r r N) rK r9 r' r rE r: r1 r< r� r� rL r. r� rH r rO r! � test_unhandledSerializationErrorp s z5CommandDispatchTests.test_unhandledSerializationErrorc C s~ | j jtdd�}| �| jjtjdddd�g� g }|�|j � | �|g � | j � t�dddd��� | �|t dd d �g� d S ) a[ L{CommandDispatcher.callRemote} should emit a properly formatted '_ask' box to its boxSender and record an outstanding L{Deferred}. When a corresponding '_answer' packet is received, the L{Deferred} should be fired, and the results translated via the given L{Command}'s response de-serialization. � worldr= r< � 1rE � yay� ignored�r< re s _answer�ignoredr� N) r< r� ra r� r: r- r r� r� r* r� r� )r �D�answersr r r! �test_callRemote� s � �z$CommandDispatchTests.test_callRemotec C sL g }|� |j� | �dt| jj�� | �| jjd jt� | �dg|� dS )z� Verify that C{callResult} completes with a L{None} result and that an unhandled error has been logged. r r N) �addBothr* r� rL r: r. �assertIsInstancer� �ZeroDivisionError)r � callResult� finalResultr r r! � _localCallbackErrorLoggingTest� s z3CommandDispatchTests._localCallbackErrorLoggingTestc C �P | j �� | jjtdd�}|�dd� � | j�t�dddd��� | � |� d S ) z� If the last callback on the L{Deferred} returned by C{callRemote} (added by application code calling C{callRemote}) fails, the failure is passed to the sender's C{unhandledError} method. rR r= c S � dd S �Nr r r �r� r r r! r� � � zVCommandDispatchTests.test_callRemoteSuccessLocalCallbackErrorLogging.<locals>.<lambda>rT rU rS rV N) r: r1 r< r� ra r� r� r r� r` �r r^ r r r! �/test_callRemoteSuccessLocalCallbackErrorLogging� s �zDCommandDispatchTests.test_callRemoteSuccessLocalCallbackErrorLoggingc C ra ) z� Like L{test_callRemoteSuccessLocalCallbackErrorLogging}, but for the case where the L{Deferred} returned by C{callRemote} fails. rR r= c S rb rc r rd r r r! r� � re zTCommandDispatchTests.test_callRemoteErrorLocalCallbackErrorLogging.<locals>.<lambda>rS s bugss stuff)� _error� _error_code� _error_descriptionN) r: r1 r< r� ra � addErrbackr� r r� r` rf r r r! �-test_callRemoteErrorLocalCallbackErrorLogging� s ��� zBCommandDispatchTests.test_callRemoteErrorLocalCallbackErrorLoggingN) r4 r5 r6 r7 r> rJ rN rQ rZ r` rg rl r r r r! r8 @ s r8 c @ s: e Zd ZdZdZde�� fde�� fgZde�� fgZ dS )�SimpleGreetingzN A very simple greeting command that uses a few basic argument types. r s greetings cookies cookieplusN) r4 r5 r6 r7 rm r rp r� rq rr r r r r! rm � s rm c @ s* e Zd ZdZdd� Zdd� Ze�e�ZdS )� TestLocatorzI A locator which implements a responder to the 'simple' command. c C r� r# )� greetingsr' r r r! r" � r\ zTestLocator.__init__c C � | j �||f� t|d d�S �N� �� cookieplus�ro r* r� �r �greeting�cookier r r! �greetingResponder� � z TestLocator.greetingResponderN)r4 r5 r6 r7 r" ry rm r� r r r r! rn � s rn c @ s" e Zd ZdZdd� Ze�e�ZdS )�OverridingLocatorzJ A locator which overrides the responder to the 'simple' command. c C rp )zV Return a different cookieplus than L{TestLocator.greetingResponder}. � rs ru rv r r r! ry � s z#OverridingLocator.greetingResponderN)r4 r5 r6 r7 ry rm r� r r r r! r{ � s r{ c @ rJ )�InheritingLocatorzN This locator should inherit the responder from L{OverridingLocator}. NrM r r r r! r} � rN r} c s6 e Zd Zdd� Z� fdd�Zdd� Ze�e�Z� ZS )�OverrideLocatorAMPc C s* t j�| � t� | _d| ji| _g | _d S �Ns custom)r r� r" �object�customResponder� expectationsro r' r r r! r" � s zOverrideLocatorAMP.__init__c s$ || j v r | j | }|S t� �|�S )zB Override the deprecated lookupFunction function. )r� �super�lookupFunction)r rU r� �� __class__r r! r� s z!OverrideLocatorAMP.lookupFunctionc C rp rq ru rv r r r! ry rz z$OverrideLocatorAMP.greetingResponder) r4 r5 r6 r" r� ry rm r� � __classcell__r r r� r! r~ � s r~ c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�CommandLocatorTestsz� The CommandLocator should enable users to specify responders to commands as functions that take structured objects, annotated with metadata. c s: |� }|� d�}|tjddd��}� �fdd�}|�|�S )z� Check that a locator of type C{locatorClass} finds a responder for command named I{simple} and that the found responder answers with the C{expected} result to a C{SimpleGreeting<"ni hao", 5>} command. r � ni hao� 5�rw rx c s �� | tjd� f d�� d S )Ns %drs �r� r r� ��values��expectedr r r! �done& s z6CommandLocatorTests._checkSimpleGreeting.<locals>.done)r) r rE r� )r � locatorClassr� r9 �responderCallabler� r� r r� r! �_checkSimpleGreeting s z(CommandLocatorTests._checkSimpleGreetingc C � | � td�S )z� A method on a L{CommandLocator} subclass decorated with a L{Command} subclass's L{responder} decorator should be returned from locateResponder, wrapped in logic to serialize and deserialize its arguments. � )r� rn r' r r r! �test_responderDecorator+ s z+CommandLocatorTests.test_responderDecoratorc C r� )z� L{CommandLocator} subclasses can override a responder inherited from a base class by using the L{Command.responder} decorator to register a new responder method. � )r� r{ r' r r r! �test_responderOverriding4 s z,CommandLocatorTests.test_responderOverridingc C r� )z� Responder lookup follows the same rules as normal method lookup rules, particularly with respect to inheritance. r� )r� r} r' r r r! �test_responderInheritance< � z-CommandLocatorTests.test_responderInheritancec sl t � � ��tdt� fdd��}��� j|� ��tdt� fdd��}|tjddd��}�fdd �}|�|�S ) a0 Subclasses which override locateResponder under its old name, lookupFunction, should have the override invoked instead. (This tests an AMP subclass, because in the version of the code that could invoke this deprecated code path, there was no L{CommandLocator}.) z-Override locateResponder, not lookupFunction.c � � � d�S r �r) r �r9 r r! r� O � zKCommandLocatorTests.test_lookupFunctionDeprecatedOverride.<locals>.<lambda>c r� �Nr r� r r� r r! r� W r� r� r� r� c � � � | tjdd�� d S �N� 8rs r� r� r' r r! r� [ � zGCommandLocatorTests.test_lookupFunctionDeprecatedOverride.<locals>.done) r~ � assertWarns�PendingDeprecationWarning�__file__r� r� r rE r� )r �customResponderObject�normalResponderObjectr� r� r �r9 r r! �%test_lookupFunctionDeprecatedOverrideC s" � � z9CommandLocatorTests.test_lookupFunctionDeprecatedOverridec sF t � � ��tdt� fdd��}|tjddd��}�fdd�}|�|�S ) z� Invoking locateResponder under its old name, lookupFunction, should emit a deprecation warning, but do the same thing. z)Call locateResponder, not lookupFunction.c r� r� )r� r r� r r! r� j r� zICommandLocatorTests.test_lookupFunctionDeprecatedInvoke.<locals>.<lambda>r� r� r� c r� r� r� r� r' r r! r� n r� zECommandLocatorTests.test_lookupFunctionDeprecatedInvoke.<locals>.done)rn r� r� r� r rE r� )r r� r� r� r r� r! �#test_lookupFunctionDeprecatedInvoke` s � z7CommandLocatorTests.test_lookupFunctionDeprecatedInvokeN) r4 r5 r6 r7 r� r� r� r� r� r� r r r r! r� s r� s Success!s No, really. Success.c @ s� e Zd ZdZdd� Zdd� Zdd� ZdZd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z dS )5�BinaryProtocolTestsz� Tests for L{amp.BinaryBoxProtocol}. @ivar _boxSender: After C{startReceivingBoxes} is called, the L{IBoxSender} which was passed to it. c C s g | _ g | _dS )zz Keep track of all boxes received by this test in its capacity as an L{IBoxReceiver} implementor. N)r� r$ r' r r r! r> � � zBinaryProtocolTests.setUpc C � || _ dS )zm Implement L{IBoxReceiver.startReceivingBoxes} to just remember the value passed in. N)� _boxSender�r r: r r r! r= � r* z'BinaryProtocolTests.startReceivingBoxesc C r2 )z5 A box was received by the protocol. Nr� rF r r r! r� � r4 z"BinaryProtocolTests.ampBoxReceivedNc C r� )zD Record the reason that we stopped receiving boxes. N)� stopReasonr. r r r! �stopReceivingBoxes� r( z&BinaryProtocolTests.stopReceivingBoxesc C rR )Nzno peerr r' r r r! rY � rX zBinaryProtocolTests.getPeerc C rR )Nzno hostr r' r r r! �getHost� rX zBinaryProtocolTests.getHostc C s | � |t� | j�|� d S r# )r\ r r$ r* r� r r r! r& � s zBinaryProtocolTests.writec C s& t �| �}|�d� | �| j|� dS )z� When L{amp.BinaryBoxProtocol} is connected to a transport, it calls C{startReceivingBoxes} on its L{IBoxReceiver} with itself as the L{IBoxSender} parameter. N)r �BinaryBoxProtocol�makeConnection�assertIsr� �r r r r r! �test_startReceivingBoxes� s z,BinaryProtocolTests.test_startReceivingBoxesc C s> G dd� d�}t � }t�|� �}|�|� | �|�� d� dS )z� The L{IBoxReceiver} which is started when L{amp.BinaryBoxProtocol} is connected to a transport can call C{sendBox} on the L{IBoxSender} passed to it before C{startReceivingBoxes} returns and have that box sent. c @ � e Zd Zdd� ZdS )z[BinaryProtocolTests.test_sendBoxInStartReceivingBoxes.<locals>.SynchronouslySendingReceiverc S s |� t�ddi�� d S )N� foo� bar)r3 r rE r� r r r! r= � r� zoBinaryProtocolTests.test_sendBoxInStartReceivingBoxes.<locals>.SynchronouslySendingReceiver.startReceivingBoxesN)r4 r5 r6 r= r r r r! � SynchronouslySendingReceiver� � r� s foo bar N)r r r� r� r� r� )r r� r% r r r r! �!test_sendBoxInStartReceivingBoxes� s z5BinaryProtocolTests.test_sendBoxInStartReceivingBoxesc C sD t �| �}|�d� |�d� |�d� | �| jt jdd�g� dS )z� When a binary box protocol receives: * a key * a value * an empty string it should emit a box and send it to its boxReceiver. r< rR rS r= N)r r� �stringReceivedr� r� r� r� r r r! �test_receiveBoxStateMachine� s z/BinaryProtocolTests.test_receiveBoxStateMachinec C s4 t � }t�| �}|�|� |�d� | �|j� dS )z� L{amp.BinaryBoxProtocol} drops its connection if the length prefix for the first a key it receives is larger than 255. � N)r r r� r� r+ r � disconnecting�r r% r r r r! �$test_firstBoxFirstKeyExcessiveLength� s z8BinaryProtocolTests.test_firstBoxFirstKeyExcessiveLengthc C �J t � }t�| �}|�|� |�d� | �|j� |�d� | �|j� dS )z� L{amp.BinaryBoxProtocol} drops its connection if the length prefix for a subsequent key in the first box it receives is larger than 255. s k vr� N�r r r� r� r+ r r� r r� r r r! �)test_firstBoxSubsequentKeyExcessiveLength� � z=BinaryProtocolTests.test_firstBoxSubsequentKeyExcessiveLengthc C r� )z� L{amp.BinaryBoxProtocol} drops its connection if the length prefix for the first key in a subsequent box it receives is larger than 255. s k v r� Nr� r� r r r! �)test_subsequentBoxFirstKeyExcessiveLength� r� z=BinaryProtocolTests.test_subsequentBoxFirstKeyExcessiveLengthc C s� t �| �}|�t� � |�d� |�tt�d��� | j � t j � | � | j j j� | �| j j j� | �| j j j � | �| j j j� dS )z� If L{amp.BinaryBoxProtocol} disconnects because it received a key length prefix which was too large, the L{IBoxReceiver}'s C{stopReceivingBoxes} method is called with a L{TooLong} failure. r� zsimulated connection doneN)r r� r� r r+ r0 r r �ConnectionDoner� �trap�TooLongr r� �isKeyr �isLocal� assertIsNone�keyNamer� r r r! �test_excessiveKeyFailure� s �z,BinaryProtocolTests.test_excessiveKeyFailurec C sR t � }t�| �}|�|� |�ttd��� | �dt| � t��� | � |j � dS )z~ L{amp.BinaryBoxProtocol.unhandledError} logs the failure passed to it and disconnects its transport. � Fake errorr N) r r r� r� r7 r rK r� rL �flushLoggedErrorsr r� r� r r r! � test_unhandledErrorWithTransport s z4BinaryProtocolTests.test_unhandledErrorWithTransportc C sT t �| �}|�t� � |�ttd��� |�ttd��� | � dt | � t��� dS )z� L{amp.BinaryBoxProtocol.unhandledError} completes without error when there is no associated transport. � Simulatedr� r N) r r� r� r r0 r rs r7 rK r� rL r� r� r r r! �#test_unhandledErrorWithoutTransport s z7BinaryProtocolTests.test_unhandledErrorWithoutTransportc C sD t �| �}|�t �ddd���� � | �| jt �ddd��g� dS )z� When a binary box protocol receives the serialized form of an AMP box, it should emit a similar box to its boxReceiver. � valueTests anotherValue)� testKeys anotherKeyN)r r� r+ rE r� r� r� r� r r r! �test_receiveBoxData s ���z'BinaryProtocolTests.test_receiveBoxDatac C sh d}d| }t � }t�| �}|�|� |�t�d|i��� � | �| jt�d|i�g� | � |j � dS )z� An L{amp.BinaryBoxProtocol} can receive serialized AMP boxes with values of up to (2 ** 16 - 1) bytes. i�� r� �kN) r r r� r� r+ rE r� r� r� r r� )r r� r� r% r r r r! �test_receiveLongerBoxData/ s z-BinaryProtocolTests.test_receiveLongerBoxDatac C sT t �| �}|�| � t �ddd��}|�| � |�|� | �d�| j�|�� � dS )z� When a binary box protocol sends a box, it should emit the serialized bytes of that box to its transport. r� r< )r� s someDatarS N) r r� r� rE r3 r� �joinr$ r� )r r� �aBoxr r r! � test_sendBox= s z BinaryProtocolTests.test_sendBoxc C s: t �| �}|�| � tt� �}|�|� | �| j|� dS )z� When a binary box protocol loses its connection, it should notify its box receiver that it has stopped receiving boxes. N)r r� r� r rK r0 r� r� )r r� �connectionFailurer r r! �#test_connectionLostStopSendingBoxesI s z7BinaryProtocolTests.test_connectionLostStopSendingBoxesc s� t dd��| �G � ��fdd�d�}t�|� �� t�ddd��}� �| � |�� d }� �|� | ��j| � | � d � �j �d� | � | j dg� � �d � | � d � �j �d � | � tj � j|� dS ) z� L{BinaryBoxProtocol} has the capacity to switch to a different protocol on a box boundary. When a protocol is in the process of switching, it cannot receive traffic. N� outgoing datac s( e Zd ZdZdd� Z� ��fdd�ZdS )z@BinaryProtocolTests.test_protocolSwitch.<locals>.SwitchyReceiverFc S r] r# r r� r r r! r= ` rX zTBinaryProtocolTests.test_protocolSwitch.<locals>.SwitchyReceiver.startReceivingBoxesc s* �� | jd� d| _� �� � ��� d S )Nz Should only receive one box!T)r �switched�_lockForSwitch� _switchTorF �r� � otherProtor! r r! r� c s zOBinaryProtocolTests.test_protocolSwitch.<locals>.SwitchyReceiver.ampBoxReceivedN)r4 r5 r6 r� r= r� r r� r r! �SwitchyReceiver] s r� s lots� data)s includes ofs Hello, world!rS s more datas Hello, world!more data)r r r� rE r� r� r+ r� r% r� r� r$ r� �ProtocolSwitchedr3 )r r� � anyOldBox�moreThanOneBoxr r� r! �test_protocolSwitchT s z'BinaryProtocolTests.test_protocolSwitchc C s: t �| �}|�| � tdd�}|�|� | �|jg � dS )z� After switching to a different protocol, if no extra bytes beyond the switch box were delivered, an empty string is not passed to the switched protocol's C{dataReceived} method. NrS )r r� r� r r� r� r$ )r r� r� r r r! � test_protocolSwitchEmptyBufferx s z2BinaryProtocolTests.test_protocolSwitchEmptyBufferc C s� t �| �}|�| � t �ddi�}|�� | �t j|j|� |�� |�|� | � d� | j �|� � � |�� t dd�}|�|� | �t j|j� dS )a8 In order to make sure the protocol never gets any invalid data sent into the middle of a box, it must be locked for switching before it is switched. It can only be unlocked if the switch failed, and attempting to send a box while it is locked should raise an exception. s somer� rS Nr� )r r� r� rE r� r� r� r3 �_unlockFromSwitchr� r� r$ r� r r� )r r� � sampleBoxr� r r r! � test_protocolSwitchInvalidStates� s z4BinaryProtocolTests.test_protocolSwitchInvalidStatesc C sd G dd� dt j�}|� }t�| �}|�| � |�� |�|� tt� �}|� |� | � |j |� dS )zo When the protocol is switched, it should notify its nested protocol of disconnection. c @ � e Zd ZdZdd� ZdS )zDBinaryProtocolTests.test_protocolSwitchLoseConnection.<locals>.LoserNc S � || _ d S r# �r/ r. r r r! r0 � r\ zSBinaryProtocolTests.test_protocolSwitchLoseConnection.<locals>.Loser.connectionLost)r4 r5 r6 r/ r0 r r r r! �Loser� r� r� N) r �Protocolr r� r� r� r� r rK r0 r� r/ )r r� �connectionLoserr� r� r r r! �!test_protocolSwitchLoseConnection� s z5BinaryProtocolTests.test_protocolSwitchLoseConnectionc C sj G dd� d�}t �| �}t�� }|� }|�| � |�� |�||� tt� �}|� |� | � |j |� dS )z~ When the protocol is switched, it should notify its nested client protocol factory of disconnection. c @ r� )zPBinaryProtocolTests.test_protocolSwitchLoseClientConnection.<locals>.ClientLoserNc S r� r# r� r� r r r! �clientConnectionLost� r\ zeBinaryProtocolTests.test_protocolSwitchLoseClientConnection.<locals>.ClientLoser.clientConnectionLost)r4 r5 r6 r/ r� r r r r! � ClientLoser� r� r� N) r r� r r� r� r� r� r rK r0 r� r/ )r r� r� r� � clientLoserr� r r r! �'test_protocolSwitchLoseClientConnection� s z;BinaryProtocolTests.test_protocolSwitchLoseClientConnection) r4 r5 r6 r7 r> r= r� r� r� rY r� r& r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r! r� x s8 $ r� c @ sR e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zeed�dd� �Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z!d<d=� Z"e#d>d>fd?d@�Z$dAdB� Z%e#fdCdD�Z&dEdF� Z'dGdH� Z(dIdJ� Z)dKdL� Z*dMdN� Z+dOS )P�AMPTestsc C sv t jt jft jt jft jt jft jt jft jt jft jt jft jt jffD ]\}}| � |� |�|� d|� d�� q%dS )z~ The classes in the amp module ought to implement the interfaces that are declared for their benefit. z does not implements(�)N) r � IBoxSenderr� � IBoxReceiverr; �IResponderLocator�CommandLocator�SimpleStringLocatorr� r � implementedBy)r � interface�implementationr r r! �test_interfaceDeclarations� s � ��z#AMPTests.test_interfaceDeclarationsc C �F t � \}}}g }d}|�|��|j� |�� | �|d d |� dS )z� Verify that a simple command can be sent and its response received with the simple low-level string-based API. rR r r< N�r� rB r� r* r r� �r r r r� �L�HELLOr r r! �test_helloWorld� � zAMPTests.test_helloWorldc C r )z} Verify that mixed-case, underscored and dashed arguments are mapped to their python names properly. rR r r< Nr r r r r! �test_wireFormatRoundTrip� r z!AMPTests.test_wireFormatRoundTripc C sf t ttd�\}}}g }d}d}|�||��|j� |�� | �|d d |� | �|d d |� dS )zK Verify that unicode arguments can be encoded and decoded. �r� r� rR u worሴldr r> r� N)r� r� r� r� r* r r� )r r r r� r r � HELLO_UNICODEr r r! �test_helloWorldUnicode� s �z AMPTests.test_helloWorldUnicodec C s( t � \}}}|jddd�}| �|� dS )zm L{BoxDispatcher.callRemoteString} returns L{None} if C{requiresAnswer} is C{False}. � WTFF)rv N)r� r? r� )r r r r� �retr r r! �(test_callRemoteStringRequiresAnswerFalse s z1AMPTests.test_callRemoteStringRequiresAnswerFalsec C s~ t � \}}}g }dd� }|�d��|��|j� |�� | �|�� d� d}|�|��|j� |�� | �|d d |� dS ) z� Verify that unknown commands using low-level APIs will be rejected with an error, but will NOT terminate the connection. c S � | � tj� dS �z: You can't propagate the error... �OK�r� r �UnhandledCommand��er r r! � clearAndAdd � z4AMPTests.test_unknownCommandLow.<locals>.clearAndAddr r rR r r< N) r� r? rk r� r* r r� �poprB �r r r r� r r r r r r! �test_unknownCommandLow � z AMPTests.test_unknownCommandLowc C s~ t � \}}}g }dd� }|�t��|��|j� |�� | �|�� d� d}|� |��|j� |�� | �|d d |� dS )z� Verify that unknown commands using high-level APIs will be rejected with an error, but will NOT terminate the connection. c S r r r r r r r! r ( r z5AMPTests.test_unknownCommandHigh.<locals>.clearAndAddr rR r r< N) r� r� rx rk r� r* r r� r rB r r r r! �test_unknownCommandHigh r" z AMPTests.test_unknownCommandHighc C sL g }t � �tjtjd���|j� |d �tj � | � dt |d j �� dS )a It can be very confusing if you write some code which responds to a command, but gets the return value wrong. Most commonly you end up returning None instead of a dictionary. Verify that if that happens, the framework logs a useful error. )rF r �NoneN) r� �dispatchCommandr r� ry rm rk r* r� �BadLocalReturn� failUnlessInr r� )r r r r r! �test_brokenReturnValue7 s � z AMPTests.test_brokenReturnValuec C sR t ttd�\}}}g }d}|jt|dd��|j� |�� | �|d d |� dS )z� Verify that unknown arguments are ignored, and not passed to a Python function which can't accept them. r rR s I'm not in the book!)r> �bonusr r> N)r� r� r� rw r� r* r r� r r r r! �test_unknownArgumentF s �� z AMPTests.test_unknownArgumentc C sb | � ttt�d���t� | � ttt�� ��t� | � ttt�� ��t� | �dtt�� �� dS )zS Verify that the various Box objects repr properly, for debugging. r� r� N) r� r� r r � _SwitchBoxr8 r| r� �assertInr' r r r! �test_simpleReprsX s zAMPTests.test_simpleReprsc C s: t dd�}t�� }||_| �t|�d|jt|�f � dS )zQ Verify that L{AMP} objects output their innerProtocol when set. Nr� z#<AMP inner <TestProto #%d> at 0x%x>)r r r� � innerProtocolr� r r �id)r r� r� r r r! �test_innerProtocolInRepra s �z!AMPTests.test_innerProtocolInReprc C s* t �� }| �t|�dt|�d�d�� dS )zh Verify that L{AMP} objects do not output 'inner' when no innerProtocol is set. z <AMP at 0xr� �>N)r r� r� r r/ r� r r r! �test_innerProtocolNotInReprn s "z$AMPTests.test_innerProtocolNotInRepr�SSL not availablec C s | � ttt�� ��t� dS )z; L{amp._TLSBox.__repr__} returns a string. N)r� r� r r �_TLSBoxr8 r' r r r! �test_simpleSSLReprv s zAMPTests.test_simpleSSLReprc C s� t � \}}}d}| jtj|jdfi |di��}| �|j� | �|j� | �|j � | � |j |� d�� | � tt|��t|�� | � dt|�� dS )zl Verify that a key that is too long will immediately raise a synchronous exception. � HHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHs Hellos hi�asciir� N)r� r� r r� r? r r� r� r� r� r� r� �encoder, r8 rL r �r r r r� r� �tlr r r! �test_keyTooLong} s zAMPTests.test_keyTooLongc C s� t � \}}}dd }| �tj|j|�}|�� | �|j� | �|j � | � |j d� | � |j |� | �tt|��t|�� | �dt|�� | �dt|�� dS )zs Verify that attempting to send value longer than 64k will immediately raise an exception. � Hi r< r� r> N)r� r� r r� rB r r r� r r� r� r� �failUnlessIdenticalr� r, r8 rL r r9 r r r! �test_valueTooLong� s zAMPTests.test_valueTooLongc C sL t ttd�\}}}g }d}|�|��|j� |�� | �|d d |� dS )z� Verify that a simple command can be sent and its response received with the high-level value parsing API. r rR r r> N)r� r� rB r� r* r r� r r r r! �test_helloWorldCommand� s �z AMPTests.test_helloWorldCommandc C s\ g }t ttd�\}}}d}|�|��|j� |�� |d �t� | �t |d j �d� dS )z� Verify that if a known error type is raised and handled, it will be properly relayed to the other end of the connection and translated into an exception, and no error will be logged. r � fuck your r� N) r� r� rB rk r* r r� rK r� r8 r� �r r r r r� r r r r! �test_helloErrorHandling� s �z AMPTests.test_helloErrorHandlingc C sp g }t ttd�\}}}d}|�|��|j� |�� |�� �t� |�|��|j� |�� |�� �t j � dS )a Verify that if a known, fatal error type is raised and handled, it will be properly relayed to the other end of the connection and translated into an exception, no error will be logged, and the connection will be terminated. r r� N) r� r� rB rk r* r r r� rO r r� rA r r r! � test_helloFatalErrorHandling� s �z%AMPTests.test_helloFatalErrorHandlingc C s� g }t ttd�\}}}t}|�|��|j� |�� |�� }|�t j � |�|��|j� |�� }|�t j � | � | �t�� dS )z� Verify that if an unknown error type is raised, it will be relayed to the other end of the connection and translated into an exception, it will be logged, and then the connection will be dropped. r N)r� r� r� rB rk r* r r r� r �UnknownRemoteErrorr r� r r� r� )r r r r r� r �ure�clr r r! �test_helloNoErrorHandling� s � z"AMPTests.test_helloNoErrorHandlingc C sp t ttd�\}}}g }|�t��|j� |�� | �|g � |j� � |�� |� � � t j � |j�i � |jS )z� Verify that a command that does not get answered until after the connection terminates will not cause any errors. r )r� r� r� r~ rk r* r r� r% �loseConnectionr r� r r� r� r- �r r r r� r r r r! �test_lateAnswer� s � zAMPTests.test_lateAnswerc C s< t ttd�\}}}d}|jt|d� |�� | �|j� dS )zG Verify that a command that requires no answer is run. r rR r= N)r� r� r� ru r r r� )r r r r� r r r r! �test_requiresNoAnswer� s �z AMPTests.test_requiresNoAnswerc C s� g }t ttd�\}}}d}|jt|d� |�� | �| �tj�� d}|jt |d�� |j � |�� |� � � tj� | �|j� dS )z] Verify that commands sent after a failed no-answer request do not complete. r r@ r= rR N)r� r� r� ru r r r� r �RemoteAmpErrorra rk r* r r� r r� r r� rA r r r! �test_requiresNoAnswerFail s �z"AMPTests.test_requiresNoAnswerFailc C sB t ttd�\}}}|j�� |�� |jtdd�}| �|d� dS )zv No-answer commands sent after the connection has been torn down do not return a L{Deferred}. r rU r= N)r� r� r% rH r r� ru r� )r r r r� r� r r r! � test_requiresNoAnswerAfterFail s � z'AMPTests.test_requiresNoAnswerAfterFailc C sH t ttd�\}}}|jtdd� |�� | �tj�}| � t |�d� dS )a+ Verify that responders of requiresAnswer=False commands have to return a dictionary anyway. (requiresAnswer is a hint from the _client_ - the server may be called upon to answer commands in any case, if the client wants to know when they complete.) r r< r= r N) r� r� r� r� ru r r� r r& r� rL )r r r r� �ler r r! � test_noAnswerResponderBadAnswer* s � z(AMPTests.test_noAnswerResponderBadAnswerc C s^ t ttd�\}}}g }|jtdd��|j� |�� | �t |�d� | �|t ddd�g� dS )z� Verify that responders with requiresAnswer=False will actually respond if the client sets requiresAnswer=True. In other words, verify that requiresAnswer is a hint honored only by the client. r s Hello!r= r s Hello!-noanswerNr� ) r� r� r� r� ra r� r* r r� rL r� rI r r r! �$test_noAnswerResponderAskedForAnswer< s ��z-AMPTests.test_noAnswerResponderAskedForAnswerc C s\ t ttd�\}}}g }|jtdd��|j� |�� |�� �d�}| � |ddigd � dS )zN Test encoding of an argument that uses the AmpList encoding. r � )r� r� r� r N) r� r� r� r r� r* r r �getr� )r r r r� r r� r r r! �test_ampListCommandN s �z AMPTests.test_ampListCommandc C sR t ttd�\}}}g }|jtdd��|j� |�� |�� �d�}| � |d� dS )z� Sending a command with an omitted AmpList argument that is designated as optional does not raise an InvalidSignature error. r �please)r� rr r� N� r� r� r� r� r� r* r r rS r� �r r r r� r rr r r r! �test_optionalAmpListOmitted\ s �z$AMPTests.test_optionalAmpListOmittedc C sZ t ttd�\}}}g }|jtdddigd��|j� |�� |�� �d�}| � |d� dS ) zV Sanity check that optional AmpList arguments are processed normally. r rU rU �foo)r� r� rr z foo acceptedNrV rW r r r! �test_optionalAmpListPresentk s � � z$AMPTests.test_optionalAmpListPresentc C s | � tjt� dS )z~ Verify that if we pass an invalid argument list (omitting an argument), an exception will be raised. N)r� r �InvalidSignaturera r' r r r! �test_failEarlyOnArgSending{ s z#AMPTests.test_failEarlyOnArgSendingc sv t �� }t|��t �� }t|��t�fdd��fdd�d�\� }}� �fdd�}� �� �|� |�� ���j� dS )z� As a debugging aid, a protocol system should raise a L{ProtocolSwitched} exception when asked to switch a protocol that is already switched. c r� r# r r �� serverProtor r! r� � r� z4AMPTests.test_doubleProtocolSwitch.<locals>.<lambda>c r� r# r r �� clientProtor r! r� � r� r c s �� tj� j� d�_d S �NT)r� r r� r� � testSucceededrd )r r r r! r� � s z4AMPTests.test_doubleProtocolSwitch.<locals>.switchedN) r r� r� r� r� r� r r rb )r �serverDeferred�clientDeferredr r� r� r )r r` r r^ r! �test_doubleProtocolSwitch� s �z"AMPTests.test_doubleProtocolSwitchFc s" d�_ t�� �|���t�� �|���t�fdd��fdd�d�\}}}|r1g }|�t��|j� |�� }|r@�� t j |j d� �fdd�� � ��fd d �} |� | � |�� �jd urg�j��j� |�� |r�|rv|j�t �d d �� n|j�i � |�� |j�� |�� ���j � d S )z� Verify that it is possible to switch to another protocol mid-connection and send data to it successfully. Fc r� r# r r r] r r! r� � r� z.AMPTests.test_protocolSwitch.<locals>.<lambda>c r� r# r r r_ r r! r� � r� r rR c sR | \\}}\}}� � |� � � |� � �d�|�t� � �d�|�t� d� _d S )NrS T)r r� r� r� r� rb )�info� serverSuccess� serverData� clientSuccess� clientDatar' r r! � cbConnsLost� s z1AMPTests.test_protocolSwitch.<locals>.cbConnsLostc s t ���g��� �S r# )r � DeferredListr� r� )rk rd rc r r! �cbSwitch� s �z.AMPTests.test_protocolSwitch.<locals>.cbSwitchNs SPURIOUSz,Here's some traffic in the form of an error.)rb r r� r� r� r~ rk r* r� r� r r� rB r� r r� r- r� r� �errbackrL r% rH r ) r �switcher�spuriousTraffic� spuriousErrorr r r� �wfdr�switchDeferredrm r )rk rd r` r rc r^ r! r� � sB � �� z AMPTests.test_protocolSwitchc C s | j td�S )z� Verify that protocol-switching even works if the value returned from the command that does the switch is deferred. )ro )r� r� r' r r r! �test_protocolSwitchDeferred� r� z$AMPTests.test_protocolSwitchDeferredc s� d| _ t�� }||��t�� }||�� t�fdd�� fdd�d�\}}}g }|jdd��|j� |�� |�� � t � | � | j � |� d�� |j� |�� | �|�� d d� d S ) z� Verify that if we try to switch protocols and it fails, the connection stays up and we can go back to speaking AMP. Fc r� r# r r r] r r! r� � r� z2AMPTests.test_protocolSwitchFail.<locals>.<lambda>c r� r# r r r_ r r! r� � r� r T)r� rR r> N)rb r r� r� r� rk r* r r r� rP r rB r� r� )r ro rc rd r r r� r r )r` r^ r! �test_protocolSwitchFail� s � z AMPTests.test_protocolSwitchFailc C s | j dd�S )zs Verify that attempts to send traffic after a switch will not corrupt the nested protocol. T)rp �r� r' r r r! �test_trafficAfterSwitch� r� z AMPTests.test_trafficAfterSwitchc C s | j ddd�S )zh Returning an error after a protocol switch should record the underlying error. T)rp rq rv r' r r r! �test_errorAfterSwitch� � z AMPTests.test_errorAfterSwitchc C s� t ttd�\}}}g }d}d}|�|��|j� |�� | �|�� d |� |�t ��|j� |�� | �|�� d |� |�|�� |j� |�� � t j � dS )zt Verify that commands with a responseType of QuitBox will in fact terminate the connection. r rR r� r> r� N)r� r� rB r� r* r r� r r� r{ rk r� r r� )r r r r� r r �GOODBYEr r r! �test_quitBoxQuits s �zAMPTests.test_quitBoxQuitsc C s� t � \}}}g }|j|_|jtddddd� |�� | �t|�d� dtjfdd d d fD ]\}}| �|d � |�|� q,|d � d � | �|d i � dS )z� Verify that the command dictionaries for a callRemoteN look correct after being serialized and parsed. � hello test� mixed case arg testr� � y�r> r� r� r� r � _command)r< r| )rg r} )rh r� )ri r~ r � _askN) r� r* r� r� ra r r� rL rm r )r r r r� r r� r r r r! �test_basicLiteralEmit s* � �z AMPTests.test_basicLiteralEmitc s� g � G � fdd�dt j�}t|d�\}}}|jtddddd��� j� |�� | �t � �d � | �� d d t dddd|j � � d d d �f� | �� d t d dd�� d S )z� Verify that a call similar to basicLiteralEmit's is handled properly with high-level quoting and passing to Python methods, and that argument names are correctly handled. c s" e Zd Z� fdd�Ze�e� dS )z:AMPTests.test_basicStructuredEmit.<locals>.StructuredHelloc s � � ||f� tdd�S )N� aaar= )r* r� )r r� r� �r r r! �h> s z<AMPTests.test_basicStructuredEmit.<locals>.StructuredHello.hN)r4 r5 r6 r� ra r� r r� r r! �StructuredHello= s r� )r� r| r} r� r~ r � r r N)r> r� r� r� r� r� rd r r� )r� r> ) r r� r� r� ra r� r* r r� rL r� r% rY )r r� r r r� r r� r! �test_basicStructuredEmit5 s8 � ��� z!AMPTests.test_basicStructuredEmitN),r4 r5 r6 r r r r r r! r# r( r* r- r0 r2 r �skipSSLr5 r; r> r? rB rC rG rJ rK rM rN rP rQ rT rX rZ r\ re r� r� rt ru rw rx r{ r� r� r r r r! r� � sV �= r� c @ r� )�!PretendRemoteCertificateAuthorityc C rR ra r r' r r r! �checkIsPretendRemoted rX z6PretendRemoteCertificateAuthority.checkIsPretendRemoteN)r4 r5 r6 r� r r r r! r� c r� r� c @ r$ )� IOSimCertr c G s | S r# r )r r� r r r! �optionsk rX zIOSimCert.optionsc C s || u sJ �| j d7 _ dS )a This isn't a real certificate, and wouldn't work on a real socket, but iosim specifies a different API so that we don't have to do any crypto math to demonstrate that the right functions get called in the right places. r T�� verifyCount�r � otherCertr r r! � iosimVerifyn s zIOSimCert.iosimVerifyN)r4 r5 r6 r� r� r� r r r r! r� h s r� c @ r� )�OKCertc C s |� � sJ �| S r# )r� �r r� r r r! r� { r� zOKCert.optionsN)r4 r5 r6 r� r r r r! r� z r� r� c @ r� )� GrumpyCertc C r0 )Nr Fr� r� r r r! r� � r� zGrumpyCert.iosimVerifyN)r4 r5 r6 r� r r r r! r� � r� r� c @ r9 )� DroppyCertc C r� r# )�toDrop)r r� r r r! r" � r\ zDroppyCert.__init__c C s | j d7 _ | j�� dS )Nr T)r� r� rH r� r r r! r� � s zDroppyCert.iosimVerifyN)r4 r5 r6 r" r� r r r r! r� � rI r� c @ s, e Zd ZdZdd� Zdd� Zej�e� dS )�SecurableProtoNc C s t � gS r# )r� r' r r r! � verifyFactory� s z SecurableProto.verifyFactoryc C s | � � }| �� }t||d�S )N��tls_localCertificate�tls_verifyAuthorities)� certFactoryr� r� )r �cert�verifyr r r! � getTLSVars� s zSecurableProto.getTLSVars) r4 r5 r6 r� r� r� r �StartTLSr� r r r r! r� � s r� r3 z2This test case requires SSL support in the reactorc @ r ) �TLSTestsc s� t ttd�\}}}t� � � fdd�|_|jtj� t� gd� g }|�t�� |j � |� � | � � j d� g }|�t�� |j � |� � | � |d ddi� d S ) z� Verify that starting TLS and succeeding at handshaking sends all the notifications to all the right places. r c r� r# r r ��okcr r! r� � r� z+TLSTests.test_startingTLS.<locals>.<lambda>r� r� r r� TN)r� r� r� r� r� r r� r� r� r� r* r r� r� )r �cli�svrr� r r r� r! �test_startingTLS� s$ ��zTLSTests.test_startingTLSc sl t ttd�\}}}t� � � fdd�|_|jtj� t� gd� |�� d|_ | j tj |jtj� t� gd� dS )zz Verify that the protocol will complain if we attempt to renegotiate TLS, which we don't support. r c r� r# r r r� r r! r� � r� z1TLSTests.test_startTooManyTimes.<locals>.<lambda>r� TN) r� r� r� r� r� r r� r� r �noPeerCertificater� � OnlyOneTLS�r r� r� r� r r� r! �test_startTooManyTimes� s& �� �z TLSTests.test_startTooManyTimesc sp t � � tttd�\}}}� fdd�|_|jtj� d� |�� | �� j d� |�t �}|�� | � |t j � dS )z� Verify that starting TLS and failing on both sides at handshaking sends notifications to all the right places and terminates the connection. r c r� r# r r ��badCertr r! r� � r� z1TLSTests.test_negotiationFailed.<locals>.<lambda>�r� r� N)r� r� r� r� r� r r� r r� r� r� � assertFailurer �NativeOpenSSLError�r r� r� r� rT r r� r! �test_negotiationFailed� s � z TLSTests.test_negotiationFailedc st t ttd�\}}}t|j�� � fdd�|_|jtj� d� |�� | � � j d� |�t �}|�� | � |t j� dS )z� Verify that starting TLS and failing by way of a lost connection notices that it is probably an SSL problem. r c r� r# r r �� droppyCertr r! r� � r� z:TLSTests.test_negotiationFailedByClosing.<locals>.<lambda>r� r� N)r� r� r� r% r� r� r r� r r� r� r� r� r �PeerVerifyErrorr� r r� r! � test_negotiationFailedByClosing� s � z(TLSTests.test_negotiationFailedByClosingN)r4 r5 r6 r� r� r� r� r r r r! r� � s r� c @ r+ ) �TLSNotAvailableTestszP Tests what happened when ssl is not available in current installation. c C s t j| _dt _dS )z% Disable ssl in amp. N)r r r' r r r! r> s zTLSNotAvailableTests.setUpc C s | j t_ dS )z% Restore ssl module. N)r r r' r r r! �tearDown s z TLSNotAvailableTests.tearDownc sD t ttd�\}}}t� � � fdd�|_| �|jtj� t� gd�t �S )zg Check that callRemote raises an exception when called with a L{amp.StartTLS}. r c r� r# r r r� r r! r� & r� z;TLSNotAvailableTests.test_callRemoteError.<locals>.<lambda>r� ) r� r� r� r� r� r� r r� r� rK r� r r� r! �test_callRemoteError s ���z)TLSNotAvailableTests.test_callRemoteErrorc sn t � }t� � � fdd�|_t�� }d|d< d|d< g }|j|_|�t� � |� |� | � |dddd �g� d S ) z| When a client with SSL enabled talks to a server without SSL, it should return a meaningful error. c r� r# r r r� r r! r� 8 r� z@TLSNotAvailableTests.test_messageReceivedError.<locals>.<lambda>s StartTLSr� rS r� s TLS_ERRORs TLS not available)ri rh rj N) r� r� r� r rE r* r3 r� r r� r� )r r� rG r� r r� r! �test_messageReceivedError1 s$ ���z.TLSNotAvailableTests.test_messageReceivedErrorN)r4 r5 r6 r7 r> r� r� r� r r r r! r� s r� c @ rJ )�InheritedErrorz2 This error is used to check inheritance. NrM r r r r! r� L rN r� c @ rJ )�OtherInheritedErrorz< This is a distinct error for checking inheritance. NrM r r r r! r� R rN r� c @ s. e Zd ZU dZediZeee e f e d< dS )� BaseCommandz: This provides a command that will be subclassed. s INHERITED_ERRORrk N) r4 r5 r6 r7 r� rk r r rs r rt r r r r! r� X s r� c @ rJ )�InheritedCommandz` This is a command which subclasses another command but does not override anything. NrM r r r r! r� ` rN r� c @ s< e Zd ZU dZde�� fgZediZe e e e f e d< dS )�AddErrorsCommandz] This is a command which subclasses another command but adds errors to the list. s others OTHER_INHERITED_ERRORrk N)r4 r5 r6 r7 r ro rq r� rk r r rs r rt r r r r! r� g s �r� c @ �" e Zd ZdZdd� Ze�e� dS )�NormalCommandProtocolz� This is a protocol which responds to L{BaseCommand}, and is used to test that inheritance does not interfere with the normal handling of errors. c C � t � �r# �r� r' r r r! �respy � zNormalCommandProtocol.respN)r4 r5 r6 r7 r� r� r� r r r r! r� s s r� c @ r� )�InheritedCommandProtocolz� This is a protocol which responds to L{InheritedCommand}, and is used to test that inherited commands inherit their bases' errors if they do not respond to any of their own. c C r� r# r� r' r r r! r� � r� z InheritedCommandProtocol.respN)r4 r5 r6 r7 r� r� r� r r r r! r� s r� c @ r� )�AddedCommandProtocolz� This is a protocol which responds to L{AddErrorsCommand}, and is used to test that inherited commands can add their own new types of errors, but still respond in the same way to their parents types of errors. c C s |rt � �t� �r# )r� r� )r �otherr r r! r� � s zAddedCommandProtocol.respN)r4 r5 r6 r7 r� r� r� r r r r! r� � s r� c @ �8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �CommandInheritanceTestszM These tests verify that commands inherit error conditions properly. c K s< t ||d�\}}}|j|fi |��}| �||�} |�� | S )z~ Check that the appropriate kind of error is raised when a given command is sent to a given protocol. r )r� r� �failUnlessFailurer ) r rM rV �cmdr� r r r� rT �d2r r r! � errorCheck� s z"CommandInheritanceTests.errorCheckc C � | � ttt�S )zx Verify that errors specified in a superclass are respected normally even if it has subclasses. )r� r� r� r� r' r r r! �test_basicErrorPropagation� ry z2CommandInheritanceTests.test_basicErrorPropagationc C r� )zp Verify that errors specified in a superclass command are propagated to its subclasses. )r� r� r� r� r' r r r! � test_inheritedErrorPropagation� s �z6CommandInheritanceTests.test_inheritedErrorPropagationc C � | j tttdd�S )z� Verify that new errors specified in a subclass of an existing command are honored even if the superclass defines some errors. T�r� )r� r� r� r� r' r r r! �test_inheritedErrorAddition� � �z3CommandInheritanceTests.test_inheritedErrorAdditionc C r� )z� Verify that errors specified in a command's superclass are respected even if that command defines new errors itself. Fr� )r� r� r� r� r' r r r! � test_additionWithOriginalError� r� z6CommandInheritanceTests.test_additionWithOriginalErrorN) r4 r5 r6 r7 r� r� r� r� r� r r r r! r� � s r� c C s" | � tjtj� |`|�| � d S r# )r� r �ConnectionLostr� r0 )rM rV r r r! � _loseAndPass� s r� c @ r$ )� LiveFireBasez4 Utility for connected reactor-using tests. c s� ddl m} t�� � _� j� j_t�� � _� j� j_t � � � j_ t � � � j_ |� d� j�� _ � �� j j� |�d� j �� j� j�� _� �� jj� � fdd�}t �� jj � jj g�}|�|�S )zB Create an amp server and connect a client to it. r )r z 127.0.0.1c s � j j� _� jj� _d S r# )� clientFactoryr� r� � serverFactoryr� )�rlstr' r r! � getProtos� rB z%LiveFireBase.setUp.<locals>.getProtos)�twisted.internetr r � ServerFactoryr� r^ � ClientFactoryr� r` r r� r� � listenTCP� serverPort� addCleanup� stopListening� connectTCPr� �port� clientConn� disconnectrl r� )r r r� �dlr r' r! r> � s � zLiveFireBase.setUpc C s` g }| j | jfD ] }|jdur%t�� �t|�}|j|_|j� � |� |� qt� |��dd� �S )zn Cleanup client and server connections, and check the error got at C{connectionLost}. Nc S s | j jS r# )r� � subFailure)�firstr r r! r� re z'LiveFireBase.tearDown.<locals>.<lambda>) r� r� r% r r� rk r� rn r0 rH r* � gatherResults)r r �connrT r r r! r� � s �zLiveFireBase.tearDownN)r4 r5 r6 r7 r> r� r r r r! r� � s r� c C s& dd l }|j�| d � |j�� d S )Nr � )�sys�stdoutr& r )r� r� r r r! �show s r� c C sN ddl m} | jdd�}| j�� }|�|�}|�||dd� d�}|�|�}|S )Nr r �shared)�CNc S rR ra r )�dnr r r! r� r� z tempSelfSigned.<locals>.<lambda>i�� )r� r �DN�KeyPair�generate�certificateRequest�signCertificateRequest�newCertificate)r �sharedDNr� �cr�sscrdr� r r r! �tempSelfSigned s r� c @ � e Zd ZeZeZdd� ZdS )�LiveFireTLSTestsc sN t � � fdd��j_� fdd��j_� �fdd�}�jjtj� � gd��|�S )aH Using real, live TLS, actually negotiate a connection. This also looks at the 'peerCertificate' attribute's correctness, since that's actually loaded using OpenSSL calls, but the main purpose is to make sure that we didn't miss anything obvious in iosim about TLS negotiations. c s � gS r# r r �r� r r! r� . s z9LiveFireTLSTests.test_liveFireCustomTLS.<locals>.<lambda>c r� r# r r r r r! r� / r� c s( �� � � �� fdd�}�j�t��|�S )Nc sT � � �� jj�� � � � �� jj�� � � � �� jj�� � � � �� jj�� � d S r# )r� r� �hostCertificate�digest�peerCertificater� )�rslt2r� r r! r� 5 s zHLiveFireTLSTests.test_liveFireCustomTLS.<locals>.secured.<locals>.pinged)r r� r� r� r� )�rsltr� �r� r r� r! �secured2 s z8LiveFireTLSTests.test_liveFireCustomTLS.<locals>.securedr� ) �tempcertr� r� r� r� r� r r� r� �r r r r r! �test_liveFireCustomTLS" s ��z'LiveFireTLSTests.test_liveFireCustomTLSN)r4 r5 r6 r� r` r^ r r r r r! r � r c @ s$ e Zd ZdZdd� Zej�e� dS )�SlightlySmartTLSz_ Specific implementation of server side protocol with different management of TLS. c C s t td�S )zS @return: the global C{tempcert} certificate as local certificate. r� )r� r r' r r r! r� L r( zSlightlySmartTLS.getTLSVarsN)r4 r5 r6 r7 r� r r� r� r r r r! r F s r c @ r� )�PlainVanillaLiveFireTestsc s � fdd�}� j �tj��|�S )z� Verify that out of the box, we can start TLS to at least encrypt the connection, even if we don't have any certificates to use. c � � j �t�S r# �r� r� r� rd r' r r! r b � zBPlainVanillaLiveFireTests.test_liveFireDefaultTLS.<locals>.secured)r� r� r r� r� r r r' r! �test_liveFireDefaultTLS\ s z1PlainVanillaLiveFireTests.test_liveFireDefaultTLSN)r4 r5 r6 r� r` r^ r r r r r! r U r r c @ s e Zd ZeZeZdd� ZdS )� WithServerTLSVerificationTestsc s& � fdd�}� j jtjtgd��|�S )zO Verify that anonymous clients can verify server certificates. c r r# r rd r' r r! r t r zMWithServerTLSVerificationTests.test_anonymousVerifyingClient.<locals>.secured)r� )r� r� r r� r r� r r r' r! � test_anonymousVerifyingCliento s ��z<WithServerTLSVerificationTests.test_anonymousVerifyingClientN)r4 r5 r6 r� r` r r^ r r r r r! r h r r c @ r$ )�ProtocolIncludingArgumentz� An L{amp.Argument} which encodes its parser and serializer arguments *including the protocol* into its parsed and serialized forms. c C s ||fS )z� Don't decode anything; just return all possible information. @return: A two-tuple of the input string and the protocol. r )r �stringr r r r! r[ � s z)ProtocolIncludingArgument.fromStringProtoc C s dt |�t |�f }|�d�S )z� Encode identifying information about L{object} and protocol into a string for later verification. @type obj: L{object} @type protocol: L{amp.AMP} �%d:%dr7 )r/ r8 )r �objr �identr r r! � toStringProto� s z'ProtocolIncludingArgument.toStringProtoN)r4 r5 r6 r7 r[ r r r r r! r | s r c @ s( e Zd ZdZde� fgZde� fgZdS )�ProtocolIncludingCommandzf A command that has argument and response schemas which use L{ProtocolIncludingArgument}. � weirdN)r4 r5 r6 r7 r rq rr r r r r! r � s r c @ s4 e Zd ZdZedd� �Zedd� �Zedd� �ZdS ) �MagicSchemaCommandza A command which overrides L{parseResponse}, L{parseArguments}, and L{makeResponse}. c C � ||f|_ |S )z� Don't do any parsing, just jam the input strings and protocol onto the C{protocol.parseResponseArguments} attribute as a two-tuple. Return the original strings. )�parseResponseArguments)r r^ r r r r! � parseResponse� � z MagicSchemaCommand.parseResponsec C r )z� Don't do any parsing, just jam the input strings and protocol onto the C{protocol.parseArgumentsArguments} attribute as a two-tuple. Return the original strings. )�parseArgumentsArguments)�clsr^ r r r r! �parseArguments� r! z!MagicSchemaCommand.parseArgumentsc C r )z� Don't do any serializing, just jam the input strings and protocol onto the C{protocol.makeArgumentsArguments} attribute as a two-tuple. Return the original strings. )�makeArgumentsArguments)r# r_ r r r r! � makeArguments� r! z MagicSchemaCommand.makeArgumentsN)r4 r5 r6 r7 � classmethodr r$ r& r r r r! r � s r c @ s& e Zd ZdZdd� Ze�dd� � dS )�NoNetworkProtocola� An L{amp.AMP} subclass which overrides private methods to avoid testing the network. It also provides a responder for L{MagicSchemaCommand} that does nothing, so that tests can test aspects of the interaction of L{amp.Command}s and L{amp.AMP}. @ivar parseArgumentsArguments: Arguments that have been passed to any L{MagicSchemaCommand}, if L{MagicSchemaCommand} has been handled by this protocol. @ivar parseResponseArguments: Responses that have been returned from a L{MagicSchemaCommand}, if L{MagicSchemaCommand} has been handled by this protocol. @ivar makeArgumentsArguments: Arguments that have been serialized by any L{MagicSchemaCommand}, if L{MagicSchemaCommand} has been handled by this protocol. c C s t �|�S )zJ Return a Deferred which fires with the original strings. )r �succeed)r rm r^ rv r r r! �_sendBoxCommand� r( z!NoNetworkProtocol._sendBoxCommandc C s i S r# r )r �weirdr r r! r� � r� zNoNetworkProtocol.<lambda>N)r4 r5 r6 r7 r* r r� r r r r! r( � s r( c @ rJ )�MyBoxz! A unique dict subclass. NrM r r r r! r, � rN r, c @ s e Zd ZdZeZdS )�0ProtocolIncludingCommandWithDifferentCommandTypezN A L{ProtocolIncludingCommand} subclass whose commandType is L{MyBox} N)r4 r5 r6 r7 r, � commandTyper r r r! r- � rz r- c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&S )'� CommandTestsz7 Tests for L{amp.Argument} and L{amp.Command}. c C s | � ttjt�� �� dS )zE L{Argument} instances provide L{amp.IArgumentType}. N)r r r � IArgumentType�Argumentr' r r r! �test_argumentInterface� r� z#CommandTests.test_argumentInterfacec C �2 t � }d}d|i}| �t�||�d||fi� dS )z� There should be a class method of Command which accepts a mapping of argument names to serialized forms and returns a similar mapping whose values have been parsed via the Command's response schema. � whateverr r+ N)r� r� r r �r r r� r^ r r r! �test_parseResponse� � �z CommandTests.test_parseResponsec s6 t � � d�� jt�d�}� ��fdd�}|�|� |S )z� Making a remote call on a L{amp.Command} subclass which overrides the C{parseResponse} method should call that C{parseResponse} method to get the response. s weeoo�r+ c s �� � jd�i� f� d S �Nr+ )r� r r� ��clientr �thingyr r! � gotResponse � zCCommandTests.test_callRemoteCallsParseResponse.<locals>.gotResponse)r( r� r r� �r rr r= r r: r! �!test_callRemoteCallsParseResponse s z.CommandTests.test_callRemoteCallsParseResponsec C r3 )z� There should be a class method of L{amp.Command} which accepts a mapping of argument names to serialized forms and returns a similar mapping whose values have been parsed via the command's argument schema. r4 r r+ N)r� r� r r$ r5 r r r! �test_parseArguments r7 z CommandTests.test_parseArgumentsc s@ t � ���tj�}t� � |t� d��}|�� ��fdd�� |S )z� Making a remote call on a L{amp.Command} subclass which overrides the C{parseArguments} method should call that C{parseArguments} method to get the arguments. r8 c s �� �jd� i�f�S r9 )r� r" r� ��argumentr r r r! r� 5 s �z@CommandTests.test_responderCallsParseArguments.<locals>.<lambda>)r( r) r rm r� r� r� )r r� rr r rB r! �!test_responderCallsParseArguments* s �z.CommandTests.test_responderCallsParseArgumentsc C sJ t � }t � }d|i}dt|�t|�f }| �t�||�d|�d�i� dS )z� There should be a class method of L{amp.Command} which accepts a mapping of argument names to objects and returns a similar mapping whose values have been serialized via the command's argument schema. r+ r r r7 N)r� r/ r� r r& r8 )r r rC r_ r r r r! �test_makeArguments; s �z CommandTests.test_makeArgumentsc C s. t � }ddi}t�||�}| �t|�t� dS )z� L{amp.Command.makeArguments}'s return type should be the type of the result of L{amp.Command.commandType}. r+ r4 N)r� r- r& r� r� r, )r r r_ r� r r r! �!test_makeArgumentsUsesCommandTypeK s �z.CommandTests.test_makeArgumentsUsesCommandTypec s8 t � �t� � �jt� d�}� ��fdd�}|�|� |S )z� Making a remote call on a L{amp.Command} subclass which overrides the C{makeArguments} method should call that C{makeArguments} method to get the response. r8 c s �� �jd� i�f� d S r9 )r� r% r� �rC r; r r r! r= b s �zCCommandTests.test_callRemoteCallsMakeArguments.<locals>.gotResponse)r( r� r� r r� r? r rG r! �!test_callRemoteCallsMakeArgumentsX s z.CommandTests.test_callRemoteCallsMakeArgumentsc C s" | � tjtjtdt� d�d� dS )z� L{Command.makeArguments} raises L{amp.InvalidSignature} if the objects dictionary passed to it includes a key which does not correspond to the Python identifier for a defined argument. r> )r> � bogusArgumentN)r� r r[ ra r& r� r� r' r r r! � test_extraArgumentsDisallowedj s �z*CommandTests.test_extraArgumentsDisallowedc C s* | � tjtjtdddiddi��d� dS )a� If a command argument conflicts with a Python keyword, the untransformed argument name is not allowed as a key in the dictionary passed to L{Command.makeArguments}. If it is supplied, L{amp.InvalidSignature} is raised. This may be a pointless implementation restriction which may be lifted. The current behavior is tested to verify that such arguments are not silently dropped on the floor (the previous behavior). r> �required�printz print valueNr )r� r r[ ra r& r� r' r r r! �test_wireSpellingDisallowedw s �z(CommandTests.test_wireSpellingDisallowedc C s$ G dd� dt j�}| �d|j� dS )zj A L{Command} subclass without a defined C{commandName} that's not a byte string. c @ rJ )zPCommandTests.test_commandNameDefaultsToClassNameAsByteString.<locals>.NewCommandz( A new command. NrM r r r r! � NewCommand� rN rN s NewCommandN)r �Commandr� rm �r rN r r r! �/test_commandNameDefaultsToClassNameAsByteString� s z<CommandTests.test_commandNameDefaultsToClassNameAsByteStringc C s. | � ttdtjfddi�}| �t|�d� dS )zq A L{Command} subclass cannot be defined with a C{commandName} that's not a byte string. rN rm �FOOz2^Command names must be byte strings, got: u?'FOO'$N�r� r� r� r rO � assertRegexr8 �r r r r r! �!test_commandNameMustBeAByteString� s ��z.CommandTests.test_commandNameMustBeAByteStringc C �0 | � ttdtjfddgi�}| �t|�d� dS )zS A L{Command} subclass's C{arguments} must have byte string names. rN rq �rY Nz3^Argument names must be byte strings, got: u?'foo'$NrS rU r r r! �/test_commandArgumentsMustBeNamedWithByteStrings� s ��z<CommandTests.test_commandArgumentsMustBeNamedWithByteStringsc C rW )zR A L{Command} subclass's C{response} must have byte string names. rN rr rX z3^Response names must be byte strings, got: u?'foo'$NrS rU r r r! �.test_commandResponseMustBeNamedWithByteStrings� s ��z;CommandTests.test_commandResponseMustBeNamedWithByteStringsc C �( G dd� dt j�}| �tdi|j� dS )zN A L{Command} subclass's C{errors} is coerced into a C{dict}. c @ � e Zd ZedfgZdS )zDCommandTests.test_commandErrorsIsConvertedToDict.<locals>.NewCommand� ZDEN)r4 r5 r6 r] rk r r r r! rN � � rN r] N)r rO r� r] rk rP r r r! �#test_commandErrorsIsConvertedToDict� � z0CommandTests.test_commandErrorsIsConvertedToDictc C �4 | � ttdtjfdtdfgi�}| �t|�d� dS )zX A L{Command} subclass's C{errors} must map exceptions to byte strings. rN rk rY z0^Error names must be byte strings, got: u?'foo'$N�r� r� r� r rO r] rT r8 rU r r r! �5test_commandErrorsMustUseBytesForOnWireRepresentation� s �zBCommandTests.test_commandErrorsMustUseBytesForOnWireRepresentationc C r[ )zS A L{Command} subclass's C{fatalErrors} is coerced into a C{dict}. c @ r\ )zICommandTests.test_commandFatalErrorsIsConvertedToDict.<locals>.NewCommandr] N)r4 r5 r6 r] rl r r r r! rN � r^ rN r] N)r rO r� r] rl rP r r r! �(test_commandFatalErrorsIsConvertedToDict� r` z5CommandTests.test_commandFatalErrorsIsConvertedToDictc C ra )ze A L{Command} subclass's C{fatalErrors} must map exceptions to byte strings. rN rl rY z6^Fatal error names must be byte strings, got: u?'foo'$Nrb rU r r r! �:test_commandFatalErrorsMustUseBytesForOnWireRepresentation� s ��zGCommandTests.test_commandFatalErrorsMustUseBytesForOnWireRepresentationN)r4 r5 r6 r7 r2 r6 r@ rA rD rE rF rH rJ rM rQ rV rY rZ r_ rc rd re r r r r! r/ � s( r/ c @ r$ )�ListOfTestsMixina� Base class for testing L{ListOf}, a parameterized zero-or-more argument type. @ivar elementType: Subclasses should set this to an L{Argument} instance. The tests will make a L{ListOf} using this. @ivar strings: Subclasses should set this to a dictionary mapping some number of keys -- as BYTE strings -- to the correct serialized form for some example values. These should agree with what L{elementType} produces/accepts. @ivar objects: Subclasses should set this to a dictionary with the same keys as C{strings} -- as NATIVE strings -- and with values which are the lists which should serialize to the values in the C{strings} dictionary. c C sN t �| j�}t �� }| jD ]}|�|�d�|| j�� d� q | �|| j � dS )a� L{ListOf.toBox} extracts the list of objects from the C{objects} dictionary passed to it, using the C{name} key also passed to it, serializes each of the elements in that list using the L{Argument} instance previously passed to its initializer, combines the serialized results, and inserts the result into the C{strings} dictionary using the same C{name} key. r7 N) r �ListOf� elementTyper� r_ r` r8 �copyr� r^ )r � stringListr^ r� r r r! � test_toBox s zListOfTestsMixin.test_toBoxc C sD t �| j�}i }| jD ] }|�|| j�� |d� q | �|| j� dS )�X L{ListOf.fromBox} reverses the operation performed by L{ListOf.toBox}. N)r rg rh r^ �fromBoxri r� r_ )r rj r_ r� r r r! � test_fromBox s z ListOfTestsMixin.test_fromBoxN)r4 r5 r6 r7 rk rn r r r r! rf � s rf c @ �6 e Zd ZdZe�� Zdddd�Zg dgg d�d�Zd S ) �ListOfStringsTestsz: Tests for L{ListOf} combined with L{amp.String}. rS � foos bar baz quux�� empty� single� multipler� )r� s bazs quux��empty�single�multipleN) r4 r5 r6 r7 r rn rh r^ r_ r r r r! rp s �rp c @ sP e Zd ZdZe�� Zdd Zddddef dd�Zg d gg d �egd gd �Z d S )�ListOfIntegersTestsz; Tests for L{ListOf} combined with L{amp.Integer}. l ���� �]OS�t7s� Xg�&}rS s 10s 1 20 500s t%ds -1)rs rt ru s huges negativerR )r � i� r )rw rx ry �huge�negativeN) r4 r5 r6 r7 r r� rh r| r^ r_ r r r r! rz + s& ��� �rz c @ ro ) �ListOfUnicodeTestsz; Tests for L{ListOf} combined with L{amp.Unicode}. rS rq s ☃ Hello worldrr rY )u ☃ra rD rv N) r4 r5 r6 r7 r rp rh r^ r_ r r r r! r~ H s � �r~ c @ s� e Zd ZdZe�� Zddddde�d��� � d� d e�d ��� � d� d �Z g e�d �ge�d �e�d�ge�d�e�d�e�d�e�d�e�d�e�d�ge�d�e�d �gd�Z dS )�ListOfDecimalTestsz; Tests for L{ListOf} combined with L{amp.Decimal}. rS s 1.1s Infinity -InfinitysC 3.141E+5 0.00003141 3.141E-7 -3.141E+5 -0.00003141 -3.141E-7s �0e6r7 s z1.5E-9)rs rt s extremes scientists engineerz1.1�Infinityz -Infinityz3.141E5z3.141e-5z3.141E-7z-3.141e5z -3.141E-5z -3.141e-7)rw rx �extreme� scientist�engineerN) r4 r5 r6 r7 r �Decimalrh �decimal� to_eng_stringr8 r^ r_ r r r r! r \ s> ���� � � � �r c @ sP e Zd ZdZe�� ZddiZde�d�e�d�e�d�e�d�giZ d d � Z d S ) �ListOfDecimalNanTestszS Tests for L{ListOf} combined with L{amp.Decimal} for not-a-number values. s nans NaN -NaN sNaN -sNaN�nan�NaNz-NaN�sNaNz-sNaNc C s� dd� }dd� }dd� }t �| j�}i }| jD ] }|�|| j�� |d� q|d }| �||d �o7||d � � | �||d �oF||d �� | �||d �oV||d � � | �||d �oe||d �� dS ) rl c S s dt | �v o dt | �vS )Nr� r� �r8 �r� r r r! �is_qnan� r� z3ListOfDecimalNanTests.test_fromBox.<locals>.is_qnanc S � dt | �v S )Nr� r� r� r r r! �is_snan� r z3ListOfDecimalNanTests.test_fromBox.<locals>.is_snanc S r� )N�-r� r� r r r! � is_signed� r z5ListOfDecimalNanTests.test_fromBox.<locals>.is_signedNr� r r r� rr )r rg rh r^ rm ri r )r r� r� r� rj r_ r� �nr r r! rn � s "z"ListOfDecimalNanTests.test_fromBoxN) r4 r5 r6 r7 r r� rh r^ r� r_ rn r r r r! r� � s ��� r� c @ r� )� DecimalTestsz# Tests for L{amp.Decimal}. c C �< t �� }| �t|jd� | �t|jd� | �t|jd� dS )z� L{amp.Decimal.toString} raises L{ValueError} if passed an object which is not an instance of C{decimal.Decimal}. �1.234�X9��v��?�� N)r r� r� � ValueErrorr �r rC r r r! �test_nonDecimal� � z DecimalTests.test_nonDecimalN)r4 r5 r6 r7 r� r r r r! r� � s r� c @ r$ )� FloatTestsz! Tests for L{amp.Float}. c C r� )zq L{amp.Float.toString} raises L{ValueError} if passed an object which is not a L{float}. r� � 1.234r� N)r �Floatr� r� r r� r r r! � test_nonFloat� r� zFloatTests.test_nonFloatc C s t �� }| �|�d�d� dS )zY L{amp.Float.toString} returns a bytestring when it is given a L{float}. r� r� N)r r� r� r r� r r r! � test_float� s zFloatTests.test_floatN)r4 r5 r6 r7 r� r� r r r r! r� � s r� c @ s e Zd ZdZe�� Zdddddd�Zejdd d d d d ej d �ejdd d d d d e d d d �d �gejdd d d d d e d dd �d �gejdd d d d d e d dd�d �gejdd d d d d e ddd �d �gejdd d d d d e d dd�d �ejdd d d d d e ddd�d �gd�Z dS )�ListOfDateTimeTestsz< Tests for L{ListOf} combined with L{amp.DateTime}. sD 2010-12-25T00:00:00.000000-00:00 2010-12-25T00:00:00.000000-00:00s" 2010-12-25T00:00:00.000000+01:00s" 2010-12-25T00:00:00.000000+03:30s" 2010-12-25T00:00:00.000000-05:00sD 2010-12-25T00:00:00.000000+03:19 2010-12-25T00:00:00.000000-06:59)s christmass christmas in eus christmas in irans christmas in nycs previous tests�� � � r )�tzinfo�+r rr � r� � � � �; )� christmaszchristmas in euzchristmas in iranzchristmas in nyczprevious testsN) r4 r5 r6 r7 r �DateTimerh r^ �datetime�utc�tzr_ r r r r! r� � s, � � � � � � �r� c @ r� ) �ListOfOptionalTestszo Tests to ensure L{ListOf} AMP arguments can be omitted from AMP commands via the 'optional' flag. c C s0 t �t �� �}| �t|jdt �� ddid� dS )zn L{ListOf.toBox} raises C{TypeError} when passed a value of L{None} for the argument. � omitted�omittedN)r rg r� r� r� r` r� �r rj r r r! �1test_requiredArgumentWithNoneValueRaisesTypeError � �zEListOfOptionalTests.test_requiredArgumentWithNoneValueRaisesTypeErrorc C s> t jt �� dd�}t �� }|�d|ddid� | �|i � dS )z� L{ListOf.toBox} silently omits serializing any argument with a value of L{None} that is designated as optional for the protocol. Trc r� N)r rg r� r� r` r� )r rj r^ r r r! �)test_optionalArgumentWithNoneValueOmitted s z=ListOfOptionalTests.test_optionalArgumentWithNoneValueOmittedc C s0 t �t �� �}| �t|jdt �� ddid� dS )z| L{ListOf.toBox} raises C{KeyError} if the argument's key is not present in the objects dictionary. � ommited� someOtherKeyr N)r rg r� r� �KeyErrorr` r� r� r r r! �1test_requiredArgumentWithKeyMissingRaisesKeyError( r� zEListOfOptionalTests.test_requiredArgumentWithKeyMissingRaisesKeyErrorc C s. t jt �� dd�}|�dt �� ddid� dS )z� L{ListOf.toBox} silently omits serializing any argument designated as optional whose key is not present in the objects dictionary. Trc r� s someOtherKeyr N)r rg r� r` r� r� r r r! �*test_optionalArgumentWithKeyMissingOmitted7 s z>ListOfOptionalTests.test_optionalArgumentWithKeyMissingOmittedc C s: t jt �� dd�}i }|�di |d� | �|ddi� dS )z� L{ListOf.fromBox} correctly reverses the operation performed by L{ListOf.toBox} for optional arguments. Trc r� Nr� )r rg r� rm r� )r rj r_ r r r! �.test_omittedOptionalArgumentDeserializesAsNone? s zBListOfOptionalTests.test_omittedOptionalArgumentDeserializesAsNoneN) r4 r5 r6 r7 r� r� r� r� r� r r r r! r� s r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )�UNIXStringTransportar An in-memory implementation of L{interfaces.IUNIXTransport} which collects all data given to it for later inspection. @ivar _queue: A C{list} of the data which has been given to this transport, eg via C{write} or C{sendFileDescriptor}. Elements are two-tuples of a string (identifying the destination of the data) and the data itself. c C s || _ g | _dS )zp @param descriptorFuzz: An offset to apply to descriptors. @type descriptorFuzz: C{int} N)�_fuzz�_queue)r �descriptorFuzzr r r! r" U r� z UNIXStringTransport.__init__c C s | j �d|| j f� d S )N�fileDescriptorReceived)r� r* r� )r � descriptorr r r! �sendFileDescriptor] r> z&UNIXStringTransport.sendFileDescriptorc C s | j �d|f� d S )Nr+ )r� r* r� r r r! r& ` r� zUNIXStringTransport.writec C s |D ]}| � |� qd S r# )r& )r �seqr$ r r r! � writeSequencec s �z!UNIXStringTransport.writeSequencec C s | j �dtt�� �f� d S )Nr0 )r� r* r r r� r' r r r! rH g s z"UNIXStringTransport.loseConnectionc C � t �d�S )Nz/tmp/some-path�r � UNIXAddressr' r r r! r� j r\ zUNIXStringTransport.getHostc C r� )Nz/tmp/another-pathr� r' r r r! rY m r\ zUNIXStringTransport.getPeerN) r4 r5 r6 r7 r" r� r& r� rH r� rY r r r r! r� J s r� c @ r+ ) �DescriptorTestsz� Tests for L{amp.Descriptor}, an argument type for passing a file descriptor over an AMP connection over a UNIX domain socket. c C s< d| _ t| j d�| _t�t�t�� ��| _| j�| j� d S )Nrr )r� ) �fuzzr� r% r r� r; r r r� r' r r r! r> | s zDescriptorTests.setUpc C s� t �� }| j�d� | j�d� | j�d� | �d|�d| j�� | �d|�d| j�� | �d|�d| j�� | �i | jj� dS )a| L{Descriptor.fromStringProto} constructs a file descriptor value by extracting a previously received file descriptor corresponding to the wire value of the argument from the L{_DescriptorExchanger} state of the protocol passed to it. This is a whitebox test which involves direct L{_DescriptorExchanger} state inspection. r� rr r �0�1�2N)r � Descriptorr r� r� r[ � _descriptorsr� r r r! �test_fromStringProto� s z$DescriptorTests.test_fromStringProtoc C s� t �� }| �d|�d| j�� | �dd| j f| jj�d�� | �d|�d| j�� | �dd| j f| jj�d�� | �d|�d| j�� | �dd| j f| jj�d�� | �i | jj � d S ) a= To send a file descriptor, L{Descriptor.toStringProto} uses the L{IUNIXTransport.sendFileDescriptor} implementation of the transport of the protocol passed to it to copy the file descriptor. Each subsequent descriptor sent over a particular AMP connection is assigned the next integer value, starting from 0. The base ten string representation of this value is the byte encoding of the argument. This is a whitebox test which involves direct L{_DescriptorExchanger} state inspection and mutation. � 0r� r� r rS r| � 2r� N) r r� r� r r r� r% r� r r� r� r r r! �test_toStringProto� s ���z"DescriptorTests.test_toStringProtoc C s� d}|� d�}i }d}||i}t�� }|�|||�� | j� t�t�t�� ��}| j j D ]}t ||d �|dd� � q,i } |� ||�� | |� | � || j | | � dS )z� L{amp.Descriptor.fromBox} can interpret an L{amp.AmpBox} constructed by L{amp.Descriptor.toBox} to reconstruct a file descriptor value. �alphar7 � r r N)r8 r r� r` ri r r� r; r r% r� �getattrrm r� r� ) r rU � nameAsBytesr^ r� � sendObjectsrC �receiver�event�receiveObjectsr r r! �test_roundTrip� s z DescriptorTests.test_roundTripN)r4 r5 r6 r7 r> r� r� r� r r r r! r� v s r� c @ sX e Zd ZdZdZeddd�Ze�dddddd d e�Zd d � Z d d� Z dd� Z dd� Z dS )� DateTimeTestszO Tests for L{amp.DateTime}, L{amp._FixedOffsetTZInfo}, and L{amp.utc}. s 9876-01-23T12:34:56.054321-01:23r� r � i�& r� �" �8 i1� c C s t �� }| �t|jd� dS )z� L{amp.DateTime.fromString} raises L{ValueError} when passed a string which does not represent a timestamp in the proper format. �abcN)r r� r� r� r �r rT r r r! �test_invalidString� s z DateTimeTests.test_invalidStringc C s, t �� }| �t|jt�dddddd�� dS )z� L{amp.DateTime.toString} raises L{ValueError} when passed a naive datetime (a datetime with no timezone information). r� r� r� r N)r r� r� r� r r� r� r r r! �test_invalidDatetime� s �z"DateTimeTests.test_invalidDatetimec C �& t �� }|�| j�}| �|| j� dS )z� L{amp.DateTime.fromString} returns a C{datetime.datetime} with all of its fields populated from the string passed to it. N)r r� r r r� r� �r rC r� r r r! �test_fromString� s z DateTimeTests.test_fromStringc C r� )z� L{amp.DateTime.toString} returns a C{str} in the wire format including all of the information from the C{datetime.datetime} passed into it, including the timezone offset. N)r r� r r� r� r r� r r r! � test_toString� s zDateTimeTests.test_toStringN) r4 r5 r6 r7 r r� r� r� r� r� r� r� r� r r r r! r� � s r� c @ r+ ) �UTCTestsz Tests for L{amp.utc}. c C s | � tj�d�d� dS )z8 L{amp.utc.tzname} returns C{"+00:00"}. Nz+00:00)r� r r� �tznamer' r r r! � test_tzname� s zUTCTests.test_tznamec C � | � tj�d�t�d�� dS )z: L{amp.utc.dst} returns a zero timedelta. Nr )r� r r� �dstr� � timedeltar' r r r! �test_dst � zUTCTests.test_dstc C r� )z@ L{amp.utc.utcoffset} returns a zero timedelta. Nr )r� r r� � utcoffsetr� r� r' r r r! �test_utcoffset r� zUTCTests.test_utcoffsetc C s | � ttddd� dS )z� L{amp._FixedOffsetTZInfo.fromSignHoursMinutes} raises L{ValueError} if passed an offset sign other than C{'+'} or C{'-'}. �?r N)r� r� r� r' r r r! � test_badSign s zUTCTests.test_badSignN)r4 r5 r6 r7 r� r� r� r� r r r r! r� � s r� c @ r� ) �RemoteAmpErrorTestsz* Tests for L{amp.RemoteAmpError}. c C � t �dd�}| �dt|�� dS )z� L{amp.RemoteAmpError} renders the given C{errorCode} (C{bytes}) and C{description} into a native string. � BROKEN�Something has brokenz"Code<BROKEN>: Something has brokenN�r rL r� r8 rU r r r! �test_stringMessage � z&RemoteAmpErrorTests.test_stringMessagec C r� )z� When C{errorCode} contains non-ASCII characters, L{amp.RemoteAmpError} renders then as backslash-escape sequences. s BROKEN-�r� z'Code<BROKEN-\xff>: Something has brokenNr� rU r r r! �&test_stringMessageReplacesNonAsciiText& r� z:RemoteAmpErrorTests.test_stringMessageReplacesNonAsciiTextc C s0 t td��}tjdd|d�}| �t|�d� dS )zs L{amp.RemoteAmpError} renders local errors with a "(local)" marker and a brief traceback. zSomething came looser� r� )�localz}^Code<BROKEN> [(]local[)]: Something has broken Traceback [(]failure with no frames[)]: <.+Exception.>: Something came loose N)r rs r rL rT r8 )r r6 r r r r! �"test_stringMessageWithLocalFailure. s �z6RemoteAmpErrorTests.test_stringMessageWithLocalFailureN)r4 r5 r6 r7 r� r� r r r r r! r� s r� )�r7 r� r� �typingr r �unittestr �zope.interfacer �zope.interface.verifyr r r� r r r r r r �twisted.protocolsr �twisted.pythonr �twisted.python.failurer � twisted.testr �twisted.test.proto_helpersr �twisted.trial.unittestr r �_ssl� ImportError� supportedr� � IReactorSSLr� �reactorLacksSSL�_FixedOffsetTZInfo�fromSignHoursMinutesr� r� r r� r: rs rK rO rP r1 rQ rO ra ru rw rx ry r{ r~ r r� r� �ProtocolSwitchCommandr� r� r� r� r� r� r� r� r� r� r� r� r� r� r r% r, r8 rm r rn r{ r} r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r( r� r, r- r/ rf rp rz r~ r r� r� r� r� r� �IUNIXTransportr� � ITransportr� r� r� r� r r r r! �<module> s � ) X � F& _ N $ iB 3/ ) % ~- /1&A ' U/