← Back
Editing: test_common.cpython-38.pyc
U W[n � @ s� d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl m Z ddl m Z ddlmZmZmZmZmZmZ dd lmZmZmZ dd lmZmZ dd lmZ G d d � d e �Z G dd� de �Z dS )z$ Tests for L{twisted.names.common}. � )�division�absolute_import)� verifyClass)� IResolver)�SynchronousTestCase)�Failure)� ResolverBase)�EFORMAT�ESERVER�ENAME�ENOTIMP�EREFUSED�Query)�DNSFormatError�DNSServerError� DNSNameError)�DNSNotImplementedError�DNSQueryRefusedError)�DNSUnknownErrorc @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )�ExceptionForCodeTestsz5 Tests for L{ResolverBase.exceptionForCode}. c C s t � j| _d S �N)r �exceptionForCode��self� r �@/usr/lib/python3/dist-packages/twisted/names/test/test_common.py�setUp s zExceptionForCodeTests.setUpc C s | � | �t�t� dS )zd L{ResolverBase.exceptionForCode} converts L{EFORMAT} to L{DNSFormatError}. N)�assertIsr r r r r r r � test_eformat s z"ExceptionForCodeTests.test_eformatc C s | � | �t�t� dS )zd L{ResolverBase.exceptionForCode} converts L{ESERVER} to L{DNSServerError}. N)r r r r r r r r � test_eserver& s z"ExceptionForCodeTests.test_eserverc C s | � | �t�t� dS )zX L{ResolverBase.exceptionForCode} converts L{ENAME} to L{DNSNameError}. N)r r r r r r r r � test_ename. s z ExceptionForCodeTests.test_enamec C s | � | �t�t� dS )zl L{ResolverBase.exceptionForCode} converts L{ENOTIMP} to L{DNSNotImplementedError}. N)r r r r r r r r � test_enotimp5 s z"ExceptionForCodeTests.test_enotimpc C s | � | �t�t� dS )zk L{ResolverBase.exceptionForCode} converts L{EREFUSED} to L{DNSQueryRefusedError}. N)r r r r r r r r � test_erefused= s z#ExceptionForCodeTests.test_erefusedc C s | � | �t� �t� dS )zr L{ResolverBase.exceptionForCode} converts any other response code to L{DNSUnknownError}. N)r r �objectr r r r r � test_otherE s z ExceptionForCodeTests.test_otherN) �__name__� __module__� __qualname__�__doc__r r r r r! r" r$ r r r r r s r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � QueryTestsz* Tests for L{ResolverBase.query}. c C s t tt� dS )zF L{ResolverBase} provides the L{IResolver} interface. N)r r r r r r r �"test_resolverBaseProvidesIResolverS s z-QueryTests.test_resolverBaseProvidesIResolverc sF g � t � }d� fdd�i|_tddd�}|�|d� | �dg� � dS ) z� L{ResolverBase.query} looks up a method to invoke using the type of the query passed to it and the C{typeToMethod} mapping on itself. �90 c s � � | |f�S r )�append��queryZtimeout�Zresultsr r �<lambda>b � z6QueryTests.test_typeToMethodDispatch.<locals>.<lambda>� example.com��name�type�{ )r2 r6 N)r � typeToMethodr r. � assertEqual)r �resolverr. r r/ r �test_typeToMethodDispatchZ s � z$QueryTests.test_typeToMethodDispatchc sZ t � � t� }d� fdd�i|_tddd�}|�|d�}g }|�|j� | �� |d � dS ) z� L{ResolverBase.query} returns a L{Deferred} which fires with the result of the method found in the C{typeToMethod} mapping for the type of the query passed to it. i1� c s � S r r r- �Zexpectedr r r0 p r1 z4QueryTests.test_typeToMethodResult.<locals>.<lambda>r2 r3 r6 r N)r# r r7 r r. �addBothr, r8 �r r9 r. Z queryDeferred�resultr r; r �test_typeToMethodResulth s z"QueryTests.test_typeToMethodResultc C sV t � }i |_tddd�}|�|d�}g }|�|j� | �|d t� |d �t � dS )z� L{ResolverBase.query} returns a L{Deferred} which fails with L{NotImplementedError} when called with a query of a type not present in its C{typeToMethod} dictionary. r2 r+ r3 r6 r N) r r7 r r. r< r, ZassertIsInstancer Ztrap�NotImplementedErrorr= r r r �test_unknownQueryTypex s z QueryTests.test_unknownQueryTypeN)r% r& r' r( r* r: r? rA r r r r r) N s r) N) r( Z __future__r r Zzope.interface.verifyr Ztwisted.internet.interfacesr Ztwisted.trial.unittestr Ztwisted.python.failurer Ztwisted.names.commonr Ztwisted.names.dnsr r r r r r Ztwisted.names.errorr r r r r r r r) r r r r �<module> s 8
Save File
Cancel