File: //usr/lib/python3/dist-packages/twisted/names/test/__pycache__/test_dns.cpython-38.pyc
U
W[�Y � @ s d Z ddlmZmZ ddlmZ ddlZddlmZ ddl m
Z
ddl m Z m
Z
ddlmZmZ dd lmZmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ej ej ej ej ej ej!ej"ej#ej$ej%ej&ej'ej(ej)ej*ej+ej,ej-ej.ej/ej0ej1ej2ej3gZ4G dd� dej5�Z6G dd� dej5�Z7G dd� dej5�Z8G dd� dej5�Z9dZ:dZ;G dd� dej<�Z=G dd� deej<�Z>G d d � d e?�Z@G d d � d ej5�ZAG d d!� d!e@�ZBG d"d#� d#ej5�ZCG d$d%� d%ej5�ZDG d&d'� d'eej5�ZEG d(d)� d)ej5�ZFG d*d+� d+ej<�ZGd,d-� ZHd.d/� ZIG d0d1� d1ej<�ZJG d2d3� d3e?�ZKG d4d5� d5eej5�ZLG d6d7� d7eej5�ZMG d8d9� d9eN�ZOG d:d;� d;e?�ZPG d<d=� d=e?�ZQG d>d?� d?e?�ZRG d@dA� dAe?�ZSG dBdC� dC�ZTG dDdE� dEe?�ZUG dFdG� dGe?�ZVG dHdI� dIe?�ZWG dJdK� dKe e
e?�ZXdhdLdM�ZYG dNdO� dOe?�ZZG dPdQ� dQe?�Z[G dRdS� dSeZe[ej<�Z\G dTdU� dUeZe[ej<�Z]G dVdW� dWeZej<�Z^G dXdY� dYeej<�Z_G dZd[� d[e?�Z`G d\d]� d]e`ej<�ZaG d^d_� d_e`ej<�ZbG d`da� daej<�ZcG dbdc� dcej<�ZdG ddde� dee?�ZeG dfdg� dgej<�ZfdS )iz
Tests for twisted.names.dns.
� )�division�absolute_import)�BytesION)� verifyClass)�Failure)� FancyEqMixin�
FancyStrMixin)�address�task)�CannotListenError�ConnectionDone)�unittest)�dns)�
proto_helpers)�ComparisonTestsMixinc @ s e Zd ZdZdd� ZdS )�
Ord2ByteTestsz&
Tests for L{dns._ord2bytes}.
c C s | � dt�d�� dS )z�
L{dns._ord2byte} accepts an integer and returns a byte string of length
one with an ordinal value equal to the given integer.
� � N)� assertEqualr �
_ord2bytes��self� r �=/usr/lib/python3/dist-packages/twisted/names/test/test_dns.py�
test_ord2byte) s zOrd2ByteTests.test_ord2byteN)�__name__�
__module__� __qualname__�__doc__r 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 )�
Str2TimeTestsz$
Tests for L{dns.str2name}.
c C s t � }| �|t�|�� dS )zY
When passed a non-string object, L{dns.str2name} returns it unmodified.
N)�object�assertIsr �str2time)r �timer r r �test_nonString6 s z Str2TimeTests.test_nonStringc C s | � dt�d�� dS )z�
Passed a string giving a number of seconds, L{dns.str2time} returns the
number of seconds represented. For example, C{"10S"} represents C{10}
seconds.
�
Z10SN�r r r" r r r r � test_seconds> s zStr2TimeTests.test_secondsc C s | � dt�d�� dS )z�
Like C{test_seconds}, but for the C{"M"} suffix which multiplies the
time value by C{60} (the number of seconds in a minute!).
�x Z2MNr& r r r r � test_minutesG s zStr2TimeTests.test_minutesc C s | � dt�d�� dS )z�
Like C{test_seconds}, but for the C{"H"} suffix which multiplies the
time value by C{3600}, the number of seconds in an hour.
i0* Z3HNr& r r r r �
test_hoursO s zStr2TimeTests.test_hoursc C s | � dt�d�� dS )z�
Like L{test_seconds}, but for the C{"D"} suffix which multiplies the
time value by C{86400}, the number of seconds in a day.
i F Z4DNr& r r r r � test_daysW s zStr2TimeTests.test_daysc C s | � dt�d�� dS )z�
Like L{test_seconds}, but for the C{"W"} suffix which multiplies the
time value by C{604800}, the number of seconds in a week.
i�$. Z5WNr& r r r r �
test_weeks_ s zStr2TimeTests.test_weeksc C s | � dt�d�� dS )z�
Like L{test_seconds}, but for the C{"Y"} suffix which multiplies the
time value by C{31536000}, the number of seconds in a year.
i 5G Z6YNr& r r r r �
test_yearsg s zStr2TimeTests.test_yearsc C s | � ttjd� dS )zY
If a non-integer prefix is given, L{dns.str2time} raises L{ValueError}.
ZfooSN)� assertRaises�
ValueErrorr r" r r r r �test_invalidPrefixo s z Str2TimeTests.test_invalidPrefixN) r r r r r$ r' r) r* r+ r, r- r0 r r r r r 2 s r c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )� NameTestsz�
Tests for L{Name}, the representation of a single domain name with support
for encoding into and decoding from DNS message format.
c C s6 | � ttjd� | � ttjt� � | � ttjg � dS )zy
When constructed with a name which is neither C{bytes} nor C{str},
L{Name} raises L{TypeError}.
�{ N)r. � TypeErrorr �Namer r r r r �test_nonStringName| s z NameTests.test_nonStringNamec C s* t �d�}| �|jt� | �d|j� dS )zg
L{dns.Name} automatically encodes unicode domain name using C{idna}
encoding.
u échec.example.orgs xn--chec-9oa.example.orgN)r r4 �assertIsInstance�name�bytesr )r r7 r r r �test_unicodeName� s
zNameTests.test_unicodeNamec C s( t �� }|�td�� | �|jd� dS )z�
L{Name.decode} populates the L{Name} instance with name information read
from the file-like object passed to it.
�
examplecom � example.comN)r r4 �decoder r r7 )r �nr r r � test_decode� s zNameTests.test_decodec C s. t �d�}t� }|�|� | �|�� d� dS )zy
L{Name.encode} encodes its name information and writes it to the
file-like object passed to it.
� foo.example.coms fooexamplecom N)r r4 r �encoder �getvalue�r r7 �streamr r r � test_encode� s
zNameTests.test_encodec C st t �d�}ddi}d}t� }|�|� t|�t jj }|�||� | �d|� � t|�d� � | �d|d�|� dS )am
If a compression dictionary is passed to it, L{Name.encode} uses offset
information from it to encode its name with references to existing
labels in the stream instead of including another copy of them in the
output. It also updates the compression dictionary with the location of
the name it writes to the stream.
r? r; � s some prefix to change .tell()s foo�N)r; r? )
r r4 r �write�len�MessageZ
headerSizer@ r rA )r r7 Z compressionZpreviousrC �expectedr r r �test_encodeWithCompression� s
��z$NameTests.test_encodeWithCompressionc C s� d}t �� }|�|� | �|jt jdddd�g� | �|jt jddddt jddd�d �g� | �|j t jd
t j
t j dt j d dd�d �g� |�
� }| �||� d S )
z�
A resource record of unknown type and class is parsed into an
L{UnknownRecord} instance with its data preserved, and an
L{UnknownRecord} instance is serialized to a string equal to the one it
was parsed from.
sD foobar ޭ�� ޭ�� somedatabazban � foo.bari�� i� )�type�clsi s somedata��ttl�rL rM rO �payloads baz.ban�1.2.3.4N)r rH �fromStrr �queries�Query�answers�RRHeader�
UnknownRecord�
additional�A�IN�Record_A�toStr)r Zwire�msg�encr r r � test_unknown� s& �
� �� ��zNameTests.test_unknownc C s� t d�}|�d� t�� }|�|� | �d|j� | �d|�� � |�|� | �|jd� | �d|�� � |�|� | �|jd� | �d|�� � d S )
a
If the leading byte of an encoded label (in bytes read from a stream
passed to L{Name.decode}) has its two high bits set, the next byte is
treated as a pointer to another label in the stream and that label is
included in the name being decoded.
s, xxxxxxxxxxxxxxxxxxxxfisiarpa foo�bar� � s
f.isi.arpa� s foo.f.isi.arpa�& s bar.foo.f.isi.arpa�, N)r �seekr r4 r< r r7 �tell)r rC r7 r r r �test_decodeWithCompression� s �
z$NameTests.test_decodeWithCompressionc C s$ t �� }td�}| �t|j|� dS )z�
L{Name.decode} raises L{ValueError} if the stream passed to it includes
a compression pointer which forms a loop, causing the name to be
undecodable.
s � N)r r4 r r. r/ r< rB r r r �test_rejectCompressionLoop s z$NameTests.test_rejectCompressionLoopc C s: t �d�}t �d�}| �||� t �d�}| �||� dS )z�
L{Name} instances are equal as long as they have the same value for
L{Name.name}, regardless of the case.
rK s fOO.barN)r r4 r )r �name1�name2Zname3r r r �
test_equality' s
zNameTests.test_equalityc C s$ t �d�}t �d�}| �||� dS )zq
L{Name} instances are not equal as long as they have different
L{Name.name} attributes.
rK s bar.fooN)r r4 ZassertNotEqual)r ri rj r r r �test_inequality4 s
zNameTests.test_inequalityN)r r r r r5 r9 r> rD rJ r` rg rh rk rl r r r r r1 w s
;
r1 c @ s� e Zd ZdZdddgZdd� Zdd� Zd d
� 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'd(� Zd)d*� Zd+d,� Zd-d.� Zd/S )0�RoundtripDNSTestsz5
Encoding and then decoding various objects.
� example.orgs go-away.fish.tvs 23strikesback.netc C sR | j D ]F}t� }t�|��|� |�dd� t�� }|�|� | �|j|� qd S )Nr ) �namesr r r4 r@ re r< r r7 �r r= �f�resultr r r � test_nameF s
zRoundtripDNSTests.test_namec C s� | j D ]�}tdd�D ]x}tdd�D ]h}t� }t�|||��|� |�dd� t�� }|�|� | �|j j |� | �|j
|� | �|j |� q"qqdS )z�
L{dns.Query.encode} returns a byte string representing the fields of the
query which can be decoded into a new L{dns.Query} instance using
L{dns.Query.decode}.
� � � r N) ro �ranger r rU r@ re r< r r7 rL rM )r r= ZdnstypeZdnsclsrq rr r r r �
test_queryR s
z RoundtripDNSTests.test_queryc C s| t � }t�dddd��|� |�dd� t�� }|�|� | �|jt�d�� | �|j d� | �|j
d� | �|j d� dS )a
L{dns.RRHeader.encode} encodes the record header's information and
writes it to the file-like object passed to it and
L{dns.RRHeader.decode} reads from a file-like object to re-construct a
L{dns.RRHeader} instance.
s test.org� � ru r N) r r rW r@ re r< r r7 r4 rL rM rO )r rq rr r r r �test_resourceRecordHeaderg s
z+RoundtripDNSTests.test_resourceRecordHeaderc C sZ d}|D ]L}t � }t�|��|� |�dd� t�� }|�|� | �|jt�|�� qdS )a
L{dns.SimpleRecord.encode} encodes the record's name information and
writes it to the file-like object passed to it and
L{dns.SimpleRecord.decode} reads from a file-like object to re-construct
a L{dns.SimpleRecord} instance.
)s this.are.test.names0 will.compress.will.this.will.name.will.hopefullys test.CASE.preSErVatIOn.YeAHs a.s.h.o.r.t.c.a.s.e.t.o.t.e.s.ts singletonr N) r r Z SimpleRecordr@ re r< r r7 r4 )r ro �srq rr r r r �test_resources| s
z RoundtripDNSTests.test_resourcesc C sD t D ]:}|� |� }}t|�}t|�}| �||d|||f � qdS )z=
Instances of all record types are hashable.
z%s != %s (for %s)N)� RECORD_TYPES�hashr )r �kZk1Zk2Zhk1Zhk2r r r �
test_hashable� s
z RoundtripDNSTests.test_hashablec C sR | j D ]F}t� }t�|��|� |�dd� t�� }|�|� | �|j|� qdS )z8
Test L{dns.Charstr} encode and decode.
r N) ro r r �Charstrr@ re r< r �stringrp r r r � test_Charstr� s
z RoundtripDNSTests.test_Charstrc C sH t � }|�|� |�� }|�dd� |�� }|�||� | �||� dS )z�
Assert that encoding C{record} and then decoding the resulting bytes
creates a record which compares equal to C{record}.
r N)r r@ rf re � __class__r< r )r �recordrC ZlengthZreplicar r r �_recordRoundtripTest� s
z&RoundtripDNSTests._recordRoundtripTestc C s" | � tjdddddddd�� d S )
z�
The byte stream written by L{dns.Record_SOA.encode} can be used by
L{dns.Record_SOA.decode} to reconstruct the state of the original
L{dns.Record_SOA} instance.
� foo� bar� �" �8 �N �Z )�mname�rname�serial�refresh�retry�expire�minimumN)r� r �
Record_SOAr r r r �test_SOA� s ��zRoundtripDNSTests.test_SOAc C s | � t�d�� dS )z�
The byte stream written by L{dns.Record_A.encode} can be used by
L{dns.Record_A.decode} to reconstruct the state of the original
L{dns.Record_A} instance.
rR N)r� r r\ r r r r �test_A� s zRoundtripDNSTests.test_Ac C s | � t�d�� dS )z�
The byte stream written by L{dns.Record_NULL.encode} can be used by
L{dns.Record_NULL.decode} to reconstruct the state of the original
L{dns.Record_NULL} instance.
s foo barN)r� r � Record_NULLr r r r � test_NULL� s zRoundtripDNSTests.test_NULLc C s | � t�ddd�� dS )z�
The byte stream written by L{dns.Record_WKS.encode} can be used by
L{dns.Record_WKS.decode} to reconstruct the state of the original
L{dns.Record_WKS} instance.
rR ry s xyzN)r� r �
Record_WKSr r r r �test_WKS� s zRoundtripDNSTests.test_WKSc C s | � t�d�� dS )z�
The byte stream written by L{dns.Record_AAAA.encode} can be used by
L{dns.Record_AAAA.decode} to reconstruct the state of the original
L{dns.Record_AAAA} instance.
z::1N)r� r � Record_AAAAr r r r � test_AAAA� s zRoundtripDNSTests.test_AAAAc C s | � t�ddd�� dS )z�
The byte stream written by L{dns.Record_A6.encode} can be used by
L{dns.Record_A6.decode} to reconstruct the state of the original
L{dns.Record_A6} instance.
� z::1:2r� N)r� r � Record_A6r r r r �test_A6� s zRoundtripDNSTests.test_A6c C s | � tjddddd�� dS )z�
The byte stream written by L{dns.Record_SRV.encode} can be used by
L{dns.Record_SRV.decode} to reconstruct the state of the original
L{dns.Record_SRV} instance.
rt � ry r; )ZpriorityZweightZport�targetN)r� r �
Record_SRVr r r r �test_SRV� s �zRoundtripDNSTests.test_SRVc C s� ddg}|D ]�\}}}}}}t �||||||�}t� } |�| � | �dd� t �� }
|
�| � | �|j|
j� | �|j|
j� | �|j |
j � | �|j
|
j
� | �|j |
j � | �|j j
|
j j
� | �|j|
j� q dS )z=
Test L{dns.Record_NAPTR} encode and decode.
)�d r% � u� sip+E2Us! !^.*$!sip:information@domain.tld!� )r� �2 � ss http+I2L+I2C+I2Rr� s _http._tcp.gatech.edur N)r � Record_NAPTRr r@ re r< r �order�
preference�flags�service�regexp� replacementr7 rO ) r Znaptrsr� r� r� r� r� r� Zrin�eZroutr r r �
test_NAPTR s& ��
z RoundtripDNSTests.test_NAPTRc C s | � tjddd�� dS )z�
The byte stream written by L{dns.Record_AFSDB.encode} can be used by
L{dns.Record_AFSDB.decode} to reconstruct the state of the original
L{dns.Record_AFSDB} instance.
ry r; )ZsubtypeZhostnameN)r� r � Record_AFSDBr r r r �
test_AFSDB s �z RoundtripDNSTests.test_AFSDBc C s | � tjddd�� dS )z�
The byte stream written by L{dns.Record_RP.encode} can be used by
L{dns.Record_RP.decode} to reconstruct the state of the original
L{dns.Record_RP} instance.
� alice.example.comr; )ZmboxZtxtN)r� r � Record_RPr r r r �test_RP% s �zRoundtripDNSTests.test_RPc C s | � tjddd�� dS )z�
The byte stream written by L{dns.Record_HINFO.encode} can be used by
L{dns.Record_HINFO.decode} to reconstruct the state of the original
L{dns.Record_HINFO} instance.
s fasts great)Zcpu�osN)r� r � Record_HINFOr r r r �
test_HINFO/ s z RoundtripDNSTests.test_HINFOc C s | � tjddd�� dS )z�
The byte stream written by L{dns.Record_MINFO.encode} can be used by
L{dns.Record_MINFO.decode} to reconstruct the state of the original
L{dns.Record_MINFO} instance.
r� r� )ZrmailbxZemailbxN)r� r � Record_MINFOr r r r �
test_MINFO8 s �z RoundtripDNSTests.test_MINFOc C s | � tjddd�� dS )z�
The byte stream written by L{dns.Record_MX.encode} can be used by
L{dns.Record_MX.decode} to reconstruct the state of the original
L{dns.Record_MX} instance.
rt r; )r� r7 N)r� r � Record_MXr r r r �test_MXB s �zRoundtripDNSTests.test_MXc C s | � t�dd�� dS )z�
The byte stream written by L{dns.Record_TXT.encode} can be used by
L{dns.Record_TXT.decode} to reconstruct the state of the original
L{dns.Record_TXT} instance.
r� r� N)r� r �
Record_TXTr r r r �test_TXTL s zRoundtripDNSTests.test_TXTN)r r r r ro rs rx r{ r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rm ? s.
rm s s 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 )'� MessageTestsz1
Tests for L{twisted.names.dns.Message}.
c C s | � t�� jd� dS )zC
L{dns.Message.authenticData} has default value 0.
r N�r r rH �
authenticDatar r r r �test_authenticDataDefaults s z&MessageTests.test_authenticDataDefaultc C s | � tjdd�jd� dS )z�
L{dns.Message.__init__} accepts a C{authenticData} argument which
is assigned to L{dns.Message.authenticData}.
rt �r� Nr� r r r r �test_authenticDataOverridez s z'MessageTests.test_authenticDataOverridec C s | � tjdd��� t� dS )zr
L{dns.Message.toStr} encodes L{dns.Message.authenticData} into
byte4 of the byte string.
rt r� N)r r rH r] � MESSAGE_AUTHENTIC_DATA_BYTESr r r r �test_authenticDataEncode� s �z%MessageTests.test_authenticDataEncodec C s$ t �� }|�t� | �|jd� dS )zp
L{dns.Message.fromStr} decodes byte4 and assigns bit3 to
L{dns.Message.authenticData}.
rt N)r rH rS r� r r� �r �mr r r �test_authenticDataDecode� s
z%MessageTests.test_authenticDataDecodec C s | � t�� jd� dS )zF
L{dns.Message.checkingDisabled} has default value 0.
r N�r r rH �checkingDisabledr r r r � test_checkingDisabledDefault� s z)MessageTests.test_checkingDisabledDefaultc C s | � tjdd�jd� dS )z�
L{dns.Message.__init__} accepts a C{checkingDisabled} argument which
is assigned to L{dns.Message.checkingDisabled}.
rt �r� Nr� r r r r � test_checkingDisabledOverride� s �z*MessageTests.test_checkingDisabledOverridec C s | � tjdd��� t� dS )zu
L{dns.Message.toStr} encodes L{dns.Message.checkingDisabled} into
byte4 of the byte string.
rt r� N)r r rH r] � MESSAGE_CHECKING_DISABLED_BYTESr r r r �test_checkingDisabledEncode� s �z(MessageTests.test_checkingDisabledEncodec C s$ t �� }|�t� | �|jd� dS )zs
L{dns.Message.fromStr} decodes byte4 and assigns bit4 to
L{dns.Message.checkingDisabled}.
rt N)r rH rS r� r r� r� r r r �test_checkingDisabledDecode� s
z(MessageTests.test_checkingDisabledDecodec C s | � dtt�� �� dS )z�
L{dns.Message.__repr__} omits field values and sections which are
identical to their defaults. The id field value is always shown.
z<Message id=0>N)r �reprr rH r r r r �test_reprDefaults� s
�z MessageTests.test_reprDefaultsc C s, t jdddddddd�}| �dt|�� dS )zM
L{dns.Message.__repr__} displays flags if they are L{True}.
T)�answer�auth�trunc�recDes�recAvr� r� zR<Message id=0 flags=answer,auth,trunc,recDes,recAv,authenticData,checkingDisabled>N�r rH r r� r� r r r �test_reprFlagsIfSet� s ��z MessageTests.test_reprFlagsIfSetc C s& t jddddd�}| �dt|�� dS )zk
L{dns.Message.__repr__} displays field values if they differ from their
defaults.
r% ra � �( )�id�opCode�rCode�maxSizez-<Message id=10 opCode=20 rCode=30 maxSize=40>Nr� r� r r r �test_reprNonDefautFields� s
�z%MessageTests.test_reprNonDefautFieldsc C sL t �� }dddg|_dddg|_ddd g|_d
d d g|_| �d
t|�� dS )�e
L{dns.Message.__repr__} displays sections which differ from their
defaults.
rt r� ry rz rv � � r� � r% � r� z^<Message id=0 queries=[1, 2, 3] answers=[4, 5, 6] authority=[7, 8, 9] additional=[10, 11, 12]>N)r rH rT rV � authorityrY r r� r� r r r �test_reprNonDefaultSections� s �z(MessageTests.test_reprNonDefaultSectionsc C s t �� }| �t|jd� dS )zy
Test that a message which has been truncated causes an EOFError to
be raised when it is parsed.
r� N)r rH r. �EOFErrorrS �r r^ r r r �test_emptyMessage s z MessageTests.test_emptyMessagec C s� t �� }|�d� | �|jd� | �|jd� | �|jt j� | �|j d� | �|j
d� | �|j g � | �|j g � | �|j
g � | �|jg � dS )ze
Test that bytes representing an empty query message can be decoded
as such.
s � z)Message was not supposed to be an answer.z-Message was not supposed to be authoritative.z)Message was not supposed to be truncated.N)r rH rS r r� � assertFalser� r� �OP_QUERYr� r� rT rV r� rY r� r r r �test_emptyQuery s, �
� � �z MessageTests.test_emptyQueryc C s� d� dd� td�D ��}t�|�}tjdtj|d�}t�� }|j�|� t � }|�
|� |� dd� t�� }|� |� | �
|jd jtj� | �|jd jj|� dS ) z|
A I{NULL} record with an arbitrary payload can be encoded and decoded as
part of a L{dns.Message}.
r� c S s g | ]}t �|��qS r �r r ��.0�ir r r �
<listcomp>, s z*MessageTests.test_NULL.<locals>.<listcomp>r� � testname�rQ r N)�joinrw r r� rW �NULLrH rV �appendr r@ re r< r6 rQ r �r r8 ZrecZrrZmsg1r| Zmsg2r r r r� ' s
zMessageTests.test_NULLc C s | � t�� �d�tj� dS )z�
L{Message.lookupRecordType} returns C{dns.UnknownRecord} if it is
called with an integer which doesn't correspond to any known record
type.
i � N)r! r rH ZlookupRecordTyperX r r r r � test_lookupRecordTypeDefault; s z)MessageTests.test_lookupRecordTypeDefaultc C sf t � }tjtjddd�d�}|�|� t�� }|�d|�� � | �|j |g� | �
|j d j � dS )z�
The L{RRHeader} instances created by L{Message} from a non-authoritative
message are marked as not authoritative.
rR r rN r� s N) r r rW r\ r@ rH rS rA r rV r� r� �r Zbufr� �messager r r � test_nonAuthoritativeMessageG s
��
z)MessageTests.test_nonAuthoritativeMessagec C sl t � }tjtjddd�d�}|�|� t�� }|�d|�� � d|_| � |j
|g� | � |j
d j� dS )z�
The L{RRHeader} instances created by L{Message} from an authoritative
message are marked as authoritative.
rR r rN r� s TN) r r rW r\ r@ rH rS rA r� r rV �
assertTruer r r r �test_authoritativeMessagea s
��
z&MessageTests.test_authoritativeMessageN)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� n s& 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"S )#�MessageComparisonTestszD
Tests for the rich comparison of L{dns.Message} instances.
c O sh |� dg �}|� dg �}|� dg �}|� dg �}tjf |�}|rF||_|rP||_|rZ||_|rd||_|S )a�
Create a L{dns.Message}.
The L{dns.Message} constructor doesn't accept C{queries}, C{answers},
C{authority}, C{additional} arguments, so we extract them from the
kwargs supplied to this factory function and assign them to the message.
@param args: Positional arguments.
@param kwargs: Keyword arguments.
@return: A L{dns.Message} instance.
rT rV r� rY )�popr rH rT rV r� rY )r �args�kwargsrT rV r� rY r� r r r �messageFactory� s z%MessageComparisonTests.messageFactoryc C s* | � | jdd�| jdd�| jdd�� dS )zd
Two L{dns.Message} instances compare equal if they have the same id
value.
r% �r� ra N��"assertNormalEqualityImplementationr r r r r �test_id� s
�z MessageComparisonTests.test_idc C s* | � | jdd�| jdd�| jdd�� dS )zg
Two L{dns.Message} instances compare equal if they have the same answer
flag.
rt �r� r Nr r r r r � test_answer� s
�z"MessageComparisonTests.test_answerc C s* | � | jdd�| jdd�| jdd�� dS )zh
Two L{dns.Message} instances compare equal if they have the same opCode
value.
r% �r� ra Nr r r r r � test_opCode� s
�z"MessageComparisonTests.test_opCodec C s* | � | jdd�| jdd�| jdd�� dS )zg
Two L{dns.Message} instances compare equal if they have the same recDes
flag.
rt �r� r Nr r r r r � test_recDes� s
�z"MessageComparisonTests.test_recDesc C s* | � | jdd�| jdd�| jdd�� dS )zf
Two L{dns.Message} instances compare equal if they have the same recAv
flag.
rt �r� r Nr r r r r �
test_recAv� s
�z!MessageComparisonTests.test_recAvc C s* | � | jdd�| jdd�| jdd�� dS )ze
Two L{dns.Message} instances compare equal if they have the same auth
flag.
rt �r� r Nr r r r r � test_auth� s
�z MessageComparisonTests.test_authc C s* | � | jdd�| jdd�| jdd�� dS )zg
Two L{dns.Message} instances compare equal if they have the same rCode
value.
r% �r� ra Nr r r r r �
test_rCode� s
�z!MessageComparisonTests.test_rCodec C s* | � | jdd�| jdd�| jdd�� dS )zf
Two L{dns.Message} instances compare equal if they have the same trunc
flag.
rt �r� r Nr r r r r �
test_trunc� s
�z!MessageComparisonTests.test_truncc C s* | � | jdd�| jdd�| jdd�� dS )zi
Two L{dns.Message} instances compare equal if they have the same
maxSize value.
r% �r� ra Nr r r r r � test_maxSize� s
�z#MessageComparisonTests.test_maxSizec C s* | � | jdd�| jdd�| jdd�� dS )zn
Two L{dns.Message} instances compare equal if they have the same
authenticData flag.
rt r� r Nr r r r r �test_authenticData
s
�z)MessageComparisonTests.test_authenticDatac C s* | � | jdd�| jdd�| jdd�� dS )zq
Two L{dns.Message} instances compare equal if they have the same
checkingDisabled flag.
rt r� r Nr r r r r �test_checkingDisabled s
�z,MessageComparisonTests.test_checkingDisabledc C sB | � | jt�d�gd�| jt�d�gd�| jt�d�gd�� dS )zc
Two L{dns.Message} instances compare equal if they have the same
queries.
r; �rT rn N�r r r rU r r r r � test_queries" s
�z#MessageComparisonTests.test_queriesc
C s` | � | jtjdt�d�d�gd�| jtjdt�d�d�gd�| jtjdt�d�d�gd�� dS )zc
Two L{dns.Message} instances compare equal if they have the same
answers.
r; rR r� �rV rn �4.3.2.1N�r r r rW r\ r r r r � test_answers. s �
�
�
�z#MessageComparisonTests.test_answersc
C sf | � | jtjdtjt�� d�gd�| jtjdtjt�� d�gd�| jtjdtjt�� d�gd�� dS )zm
Two L{dns.Message} instances compare equal if they have the same
authority records.
r; �rL rQ �r� rn N�r r r rW �SOAr� r r r r �test_authority= s" �
�
�
�z%MessageComparisonTests.test_authorityc
C s` | � | jtjdt�d�d�gd�| jtjdt�d�d�gd�| jtjdt�d�d�gd�� dS )zn
Two L{dns.Message} instances compare equal if they have the same
additional records.
r; rR r� �rY rn Nr% r r r r �test_additionalO s �
�
�
�z&MessageComparisonTests.test_additionalN)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 } s" r c @ s" e Zd ZdZdd� Zddd�ZdS )�TestControllerz�
Pretend to be a DNS query processor for a DNSDatagramProtocol.
@ivar messages: the list of received messages.
@type messages: C{list} of (msg, protocol, address)
c C s
g | _ dS )zG
Initialize the controller: create a list of messages.
N)�messagesr r r r �__init__g s zTestController.__init__Nc C s | j �|||f� dS )zN
Save the message so that it can be checked during the tests.
N)r/ r� )r r^ �proto�addrr r r �messageReceivedn s z TestController.messageReceived)N)r r r r r0 r3 r r r r r. _ s r. c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�DatagramProtocolTestsz=
Test various aspects of L{dns.DNSDatagramProtocol}.
c C sD t �� | _t� | _t�| j�| _t� � }| j�
|� | jj | j_ dS )zQ
Create a L{dns.DNSDatagramProtocol} with a deterministic clock.
N) r
�Clock�clockr. �
controllerr ZDNSDatagramProtocolr1 r ZFakeDatagramTransport�makeConnection� callLater)r � transportr r r �setUp{ s
zDatagramProtocolTests.setUpc C s, | j �dt�ddd�� | �| jjg � dS )z�
Test that when a short datagram is received, datagramReceived does
not raise an exception while processing it.
r� ZUDP� 127.0.0.1i90 N)r1 �datagramReceivedr Z IPv4Addressr r7 r/ r r r r �test_truncatedPacket� s
�z*DatagramProtocolTests.test_truncatedPacketc s� � j �dt�d�g�}� �t� j j�� �d� t�� }t t
� j j�� ��|_ tj tj
dd�d�g|_� fdd�}|�|� � j �|�� d� |S ) �6
Test content received after a query.
�r< iaS r� rt rR �r r� c s � � | jd j�� d� d S �Nr rR �r rV rQ Z
dottedQuad�rr r r r �cb� s z2DatagramProtocolTests.test_simpleQuery.<locals>.cb)r1 �queryr rU r rG � liveMessages�keysrH �next�iterr� rW r\ rV � addCallbackr= r] )r �dr� rE r r r �test_simpleQuery� s
z&DatagramProtocolTests.test_simpleQueryc C s\ | j �dt�d�g�}| �t| j j�d� | j�d� | � |tj
� | �t| j j�d� |S )�>
Test that query timeouts after some seconds.
r@ r� rt r% r � r1 rF r rU r rG rG r6 Zadvance�
assertFailureZDNSQueryTimeoutError�r rL r r r �test_queryTimeout� s z'DatagramProtocolTests.test_queryTimeoutc C s4 dd� }|| j j_| j �dt�d�g�}| �|t�S )z�
Exceptions raised by the transport's write method should be turned into
C{Failure}s passed to errbacks of the C{Deferred} returned by
L{DNSDatagramProtocol.query}.
c S s t d��d S �N�bar�� RuntimeError)r r2 r r r �
writeError� s z9DatagramProtocolTests.test_writeError.<locals>.writeErrorr@ r� �r1 r: rF rF r rU rP rV �r rW rL r r r �test_writeError� s
z%DatagramProtocolTests.test_writeErrorc C s: dd� }|| j _d| j _| j �dt�d�g�}| �|t�S )z�
Exception L{CannotListenError} raised by C{listenUDP} should be turned
into a C{Failure} passed to errback of the C{Deferred} returned by
L{DNSDatagramProtocol.query}.
c S s t d d d ��d S �N)r r r r r �startListeningError� s zCDatagramProtocolTests.test_listenError.<locals>.startListeningErrorNr@ r� )r1 ZstartListeningr: rF r rU rP r )r r\ rL r r r �test_listenError� s
z&DatagramProtocolTests.test_listenErrorc C s\ t �� }d|_t jt jdd�d�g|_| j�|�� d� | � | j
j d d �� |�� � dS ) z�
When receiving a message whose id is not in
L{DNSDatagramProtocol.liveMessages} or L{DNSDatagramProtocol.resends},
the message will be received by L{DNSDatagramProtocol.controller}.
rt rR rA r� r@ ���r N) r rH r� rW r\ rV r1 r= r] r r7 r/ )r r r r r �$test_receiveMessageNotInLiveMessages� s
�
�z:DatagramProtocolTests.test_receiveMessageNotInLiveMessagesN) r r r r r; r> rM rR rZ r] r_ r r r r r4 v s
r4 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �TestTCPControllera
Pretend to be a DNS query processor for a DNSProtocol.
@ivar connections: A list of L{DNSProtocol} instances which have
notified this controller that they are connected and have not
yet notified it that their connection has been lost.
c C s t �| � g | _d S r[ )r. r0 � connectionsr r r r r0 � s
zTestTCPController.__init__c C s | j �|� d S r[ )ra r� �r r1 r r r �connectionMade� s z TestTCPController.connectionMadec C s | j �|� d S r[ )ra �removerb r r r �connectionLost� s z TestTCPController.connectionLostN)r r r r r0 rc re r r r r r` � s r` c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�DNSProtocolTestsz5
Test various aspects of L{dns.DNSProtocol}.
c C s@ t �� | _t� | _t�| j�| _| j�t �
� � | jj | j_ dS )zI
Create a L{dns.DNSProtocol} with a deterministic clock.
N) r
r5 r6 r` r7 r Z DNSProtocolr1 r8 r ZStringTransportr9 r r r r r; � s
zDNSProtocolTests.setUpc C s< | � | jj| jg� | j�ttd��� | � | jjg � dS )z�
L{dns.DNSProtocol} calls its controller's C{connectionMade}
method with itself when it is connected to a transport and its
controller's C{connectionLost} method when it is disconnected.
zFake Connection DoneN)r r7 ra r1 re r r r r r r �test_connectionTracking s
�z(DNSProtocolTests.test_connectionTrackingc C sZ | j �t�d�g�}| �t| j j�d� | j�d� | � |tj
� | �t| j j�d� |S )rN r� rt �< r rO rQ r r r rR
s z"DNSProtocolTests.test_queryTimeoutc s� � j �t�d�g�}� �t� j j�� �d� t�� }t t
� j j�� ��|_ tj tj
dd�d�g|_� fdd�}|�|� |�� }t�dt|��| }� j �|� |S ) r? r� rt rR rA r� c s � � | jd j�� d� d S rB rC rD r r r rE " s z-DNSProtocolTests.test_simpleQuery.<locals>.cb�!H)r1 rF r rU r rG rG rH rH rI rJ r� rW r\ rV rK r] �struct�pack� dataReceived)r rL r� rE r| r r r rM s
z!DNSProtocolTests.test_simpleQueryc C s2 dd� }|| j j_| j �t�d�g�}| �|t�S )z�
Exceptions raised by the transport's write method should be turned into
C{Failure}s passed to errbacks of the C{Deferred} returned by
L{DNSProtocol.query}.
c S s t d��d S rS rU )r r r r rW 1 s z4DNSProtocolTests.test_writeError.<locals>.writeErrorr� rX rY r r r rZ + s
z DNSProtocolTests.test_writeErrorc C sr t �� }d|_t jt jdd�d�g|_|�� }t�dt |��| }| j
� |� | � | j
jd d �� |�� � dS ) z�
When receiving a message whose id is not in L{DNSProtocol.liveMessages}
the message will be received by L{DNSProtocol.controller}.
rt rR rA r� ri r^ r N)r rH r� rW r\ rV r] rj rk rG r1 rl r r7 r/ )r r r� r r r r_ 9 s
�
�z5DNSProtocolTests.test_receiveMessageNotInLiveMessagesN)
r r r r r; rg rR rM rZ r_ r r r r rf � s rf 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&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4S )5� ReprTestszE
Tests for the C{__repr__} implementation of record classes.
c C s | � tt�dd��d� dS )z�
The repr of a L{dns.Record_NS} instance includes the name of the
nameserver and the TTL of the record.
r; �� z <NS name=example.com ttl=4321>N)r r� r � Record_NSr r r r �test_nsN s �zReprTests.test_nsc C s | � tt�dd��d� dS )z�
The repr of a L{dns.Record_MD} instance includes the name of the
mail destination and the TTL of the record.
r; rn z <MD name=example.com ttl=4321>N)r r� r � Record_MDr r r r �test_mdX s �zReprTests.test_mdc C s | � tt�dd��d� dS )z�
The repr of a L{dns.Record_MF} instance includes the name of the
mail forwarder and the TTL of the record.
r; rn z <MF name=example.com ttl=4321>N)r r� r � Record_MFr r r r �test_mfb s �zReprTests.test_mfc C s | � tt�dd��d� dS )z�
The repr of a L{dns.Record_CNAME} instance includes the name of the
mail forwarder and the TTL of the record.
r; rn z!<CNAME name=example.com ttl=4321>N)r r� r � Record_CNAMEr r r r �
test_cnamel s �zReprTests.test_cnamec C s | � tt�dd��d� dS )z}
The repr of a L{dns.Record_MB} instance includes the name of the
mailbox and the TTL of the record.
r; rn z <MB name=example.com ttl=4321>N)r r� r � Record_MBr r r r �test_mbv s �zReprTests.test_mbc C s | � tt�dd��d� dS )z�
The repr of a L{dns.Record_MG} instance includes the name of the
mail group member and the TTL of the record.
r; rn z <MG name=example.com ttl=4321>N)r r� r � Record_MGr r r r �test_mg� s �zReprTests.test_mgc C s | � tt�dd��d� dS )z�
The repr of a L{dns.Record_MR} instance includes the name of the
mail rename domain and the TTL of the record.
r; rn z <MR name=example.com ttl=4321>N)r r� r � Record_MRr r r r �test_mr� s �zReprTests.test_mrc C s | � tt�dd��d� dS )z~
The repr of a L{dns.Record_PTR} instance includes the name of the
pointer and the TTL of the record.
r; rn z <PTR name=example.com ttl=4321>N)r r� r �
Record_PTRr r r r �test_ptr� s �zReprTests.test_ptrc C s | � tt�dd��d� dS )z�
The repr of a L{dns.Record_DNAME} instance includes the name of the
non-terminal DNS name redirection and the TTL of the record.
r; rn z!<DNAME name=example.com ttl=4321>N)r r� r � Record_DNAMEr r r r �
test_dname� s �zReprTests.test_dnamec C s | � tt�dd��d� dS )z�
The repr of a L{dns.Record_A} instance includes the dotted-quad
string representation of the address it is for and the TTL of the
record.
rR i7 z<A address=1.2.3.4 ttl=567>N)r r� r r\ r r r r �test_a� s �zReprTests.test_ac
C s* | � ttjddddddddd ��d
� d S ) zh
The repr of a L{dns.Record_SOA} instance includes all of the
authority fields.
s mNames rNamer2 �� � r% r� r� )r� r� r� r� r� r� r� rO zZ<SOA mname=mName rname=rName serial=123 refresh=456 retry=789 expire=10 minimum=11 ttl=12>N)r r� r r� r r r r �test_soa� s ��zReprTests.test_soac C s | � tt�dd��d� dS )z
The repr of a L{dns.Record_NULL} instance includes the repr of its
payload and the TTL of the record.
s abcdr2 z <NULL payload='abcd' ttl=123>N)r r� r r� r r r r � test_null� s �zReprTests.test_nullc C s | � ttjdddd��d� dS )z�
The repr of a L{dns.Record_WKS} instance includes the dotted-quad
string representation of the address it is for, the IP protocol
number it is for, and the TTL of the record.
z2.3.4.5r� r� rN z&<WKS address=2.3.4.5 protocol=7 ttl=8>N)r r� r r� r r r r �test_wks� s �zReprTests.test_wksc C s | � ttjddd��d� dS )z�
The repr of a L{dns.Record_AAAA} instance includes the colon-separated
hex string representation of the address it is for and the TTL of the
record.
z
8765::1234r% rN z <AAAA address=8765::1234 ttl=10>N)r r� r r� r r r r � test_aaaa� s �zReprTests.test_aaaac C s" | � ttjddddd��d� dS )z�
The repr of a L{dns.Record_A6} instance includes the colon-separated
hex string representation of the address it is for and the TTL of the
record.
r z
1234::5678rK r% rN z,<A6 suffix=1234::5678 prefix=foo.bar ttl=10>N)r r� r r� r r r r �test_a6� s �zReprTests.test_a6c
C s" | � tt�ddddd��d� dS )z�
The repr of a L{dns.Record_SRV} instance includes the name and port of
the target and the priority, weight, and TTL of the record.
rt r� ry rn rz z9<SRV priority=1 weight=2 target=example.org port=3 ttl=4>N)r r� r r� r r r r �test_srv� s �zReprTests.test_srvc C s* t �ddddddd�}| �t|�d� d S )
z�
The repr of a L{dns.Record_NAPTR} instance includes the order,
preference, flags, service, regular expression, replacement, and TTL of
the record.
rv r� � S� https
/foo/bar/i� bazry zY<NAPTR order=5 preference=9 flags=S service=http regexp=/foo/bar/i replacement=baz ttl=3>N)r r� r r� �r r� r r r �
test_naptr� s ��zReprTests.test_naptrc C s | � tt�ddd��d� dS )z{
The repr of a L{dns.Record_AFSDB} instance includes the subtype,
hostname, and TTL of the record.
ry rn rv z,<AFSDB subtype=3 hostname=example.org ttl=5>N)r r� r r� r r r r �
test_afsdb s �zReprTests.test_afsdbc C s | � tt�ddd��d� dS )zw
The repr of a L{dns.Record_RP} instance includes the mbox, txt, and TTL
fields of the record.
r� s admin.example.comry z7<RP mbox=alice.example.com txt=admin.example.com ttl=3>N)r r� r r� r r r r �test_rp
s �zReprTests.test_rpc C s | � tt�ddd��d� dS )zx
The repr of a L{dns.Record_HINFO} instance includes the cpu, os, and
TTL fields of the record.
s sparcs minixr� z%<HINFO cpu='sparc' os='minix' ttl=12>N)r r� r r� r r r r �
test_hinfo s �zReprTests.test_hinfoc C s" t �ddd�}| �t|�d� dS )z�
The repr of a L{dns.Record_MINFO} instance includes the rmailbx,
emailbx, and TTL fields of the record.
r� � bob.example.com� zF<MINFO responsibility=alice.example.com errors=bob.example.com ttl=15>N)r r� r r� r� r r r �
test_minfo! s ��zReprTests.test_minfoc C s | � tt�ddd��d� dS )z~
The repr of a L{dns.Record_MX} instance includes the preference, name,
and TTL fields of the record.
�
s mx.example.comr� z,<MX preference=13 name=mx.example.com ttl=2>N)r r� r r� r r r r �test_mx. s �zReprTests.test_mxc C s | � ttjdddd��d� dS )zr
The repr of a L{dns.Record_TXT} instance includes the data and ttl
fields of the record.
r� r� r� rN z <TXT data=['foo', 'bar'] ttl=15>N)r r� r r� r r r r �test_txt8 s �zReprTests.test_txtc C s | � ttjdddd��d� dS )zr
The repr of a L{dns.Record_SPF} instance includes the data and ttl
fields of the record.
r� r� r� rN z <SPF data=['foo', 'bar'] ttl=15>N)r r� r �
Record_SPFr r r r �test_spfB s �zReprTests.test_spfc C s | � tt�dd��d� dS )zu
The repr of a L{dns.UnknownRecord} instance includes the data and ttl
fields of the record.
s foo barr� z"<UNKNOWN data='foo\x1fbar' ttl=12>N)r r� r rX r r r r r` L s �zReprTests.test_unknownN) r r r r rp rr rt rv rx rz r| r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r` r r r r rm J s4
rm 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&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>S )?�
EqualityTestszM
Tests for the equality and non-equality behavior of record classes.
c C s | � |||�S r[ )r )r Z
firstValueOneZsecondValueOneZvalueTwor r r �
_equalityTest[ s
�zEqualityTests._equalityTestc C s$ | � t�d�t�d�t�d�� dS )zt
Two L{dns.Charstr} instances compare equal if and only if they have the
same string value.
� abc� defN)r� r r� r r r r � test_charstr` s
�zEqualityTests.test_charstrc C s$ | � t�d�t�d�t�d�� dS )zo
Two L{dns.Name} instances compare equal if and only if they have the
same name value.
r� r� N)r� r r4 r r r r rs i s
�zEqualityTests.test_namec C sD | � |dd�|dd�|dd�� | � |dd�|dd�|dd�� dS )z�
Assert that instances of C{cls} with the same attributes compare equal
to each other and instances with different attributes compare as not
equal.
@param cls: A L{dns.SimpleRecord} subclass.
r; r2 iA rn N)r� )r rM r r r �_simpleEqualityTestr s ��z!EqualityTests._simpleEqualityTestc
C s~ | � tjdt�d�d�tjdt�d�d�tjdt�d�d�� | � tjdt�d�d�tjdt�d�d�tjdt�d�d�� | � t�dtj�t�dtj�t�dtj�� | � tjdtjt�d�d�tjdtjt�d�d�tjdtjt�d�d�� | � tjddt�d�d�tjddt�d�d�tjdd t�d�d�� | � tjdd
t�d�d �tjdd
t�d�d �tjdd t�d�d �� d
S )z�
Two L{dns.RRHeader} instances compare equal if and only if they have
the same name, type, class, time to live, payload, and authoritative
bit.
r; rR r� rn �1.2.3.5)rM rQ rh )rO rQ r( rt )r� rQ r N)r� r rW r\ rZ ZMXr[ ZCSr r r r �
test_rrheader� s< �� ����zEqualityTests.test_rrheaderc C s | � tj� dS )zv
Two L{dns.Record_NS} instances compare equal if and only if they have
the same name and TTL.
N)r� r ro r r r r rp � s zEqualityTests.test_nsc C s | � tj� dS �zv
Two L{dns.Record_MD} instances compare equal if and only if they have
the same name and TTL.
N)r� r rq r r r r rr � s zEqualityTests.test_mdc C s | � tj� dS )zv
Two L{dns.Record_MF} instances compare equal if and only if they have
the same name and TTL.
N)r� r rs r r r r rt � s zEqualityTests.test_mfc C s | � tj� dS )zy
Two L{dns.Record_CNAME} instances compare equal if and only if they
have the same name and TTL.
N)r� r ru r r r r rv � s zEqualityTests.test_cnamec C s | � tj� dS )zv
Two L{dns.Record_MB} instances compare equal if and only if they have
the same name and TTL.
N)r� r rw r r r r rx � s zEqualityTests.test_mbc C s | � tj� dS )zv
Two L{dns.Record_MG} instances compare equal if and only if they have
the same name and TTL.
N)r� r ry r r r r rz � s zEqualityTests.test_mgc C s | � tj� dS )zv
Two L{dns.Record_MR} instances compare equal if and only if they have
the same name and TTL.
N)r� r r{ r r r r r| � s zEqualityTests.test_mrc C s | � tj� dS )zw
Two L{dns.Record_PTR} instances compare equal if and only if they have
the same name and TTL.
N)r� r r} r r r r r~ � s zEqualityTests.test_ptrc C s | � tj� dS r� )r� r r r r r r r� � s zEqualityTests.test_dnamec C sP | � t�dd�t�dd�t�dd�� | � t�dd�t�dd�t�dd�� dS )zx
Two L{dns.Record_A} instances compare equal if and only if they have
the same address and TTL.
rR rv r� r� N)r� r r\ r r r r r� � s
�
�zEqualityTests.test_ac C sT | � t�dddddddd�t�dddddddd�t�d ddddddd�� | � t�dddddddd�t�dddddddd�t�dd dddddd�� | � t�dddddddd�t�dddddddd�t�ddd
ddddd�� | � t�dddddddd�t�dddddddd�t�dddd
dddd�� | � t�dddddddd�t�dddddddd�t�ddddd
ddd�� | � t�dddddddd�t�dddddddd�t�dddddd
dd�� | � t�dddddddd�t�dddddddd�t�ddddddd
d�� | � t�dddddddd�t�dddddddd�t�dd dddddd
�� d S ) z�
Two L{dns.Record_SOA} instances compare equal if and only if they have
the same mname, rname, serial, refresh, minimum, expire, retry, and
ttl.
s mnames rnamer2 r� r� r% ra r� s xnamert N)r� r r� r r r r r� sP ��������zEqualityTests.test_soac C sP | � t�dd�t�dd�t�dd�� | � t�dd�t�dd�t�dd�� dS )z{
Two L{dns.Record_NULL} instances compare equal if and only if they have
the same payload and ttl.
zfoo barr% zbar foor� N)r� r r� r r r r r� ; s
�
�zEqualityTests.test_nullc
C s� | � t�dddd�t�dddd�t�dddd�� | � t�dddd�t�dddd�t�dddd�� | � t�dddd�t�dddd�t�dddd�� | � t�dddd�t�dddd�t�dddd�� d S )
z�
Two L{dns.Record_WKS} instances compare equal if and only if they have
the same address, protocol, map, and ttl.
rR rt �foor� r$ r� rT �� N)r� r r� r r r r r� L s( ����zEqualityTests.test_wksc C sP | � t�dd�t�dd�t�dd�� | � t�dd�t�dd�t�dd�� dS )z{
Two L{dns.Record_AAAA} instances compare equal if and only if they have
the same address and ttl.
z1::2rt z2::1r% N)r� r r� r r r r r� g s
�
�zEqualityTests.test_aaaac
C s� | � t�dddd�t�dddd�t�dddd�� | � t�dddd�t�dddd�t�dddd�� | � t�dddd�t�dddd�t�dddd�� | � t�dddd�t�dddd�t�dddd�� d S )
z�
Two L{dns.Record_A6} instances compare equal if and only if they have
the same prefix, prefix length, suffix, and ttl.
r z::abcdr; r% rb z::abcd:0rn r� N)r� r r� r r r r r� x s( ����zEqualityTests.test_a6c C s | � t�ddddd�t�ddddd�t�ddddd�� | � t�ddddd�t�ddddd�t�ddddd�� | � t�ddddd�t�ddddd�t�ddddd�� | � t�ddddd�t�ddddd�t�dddd d�� | � t�ddddd�t�ddddd�t�ddddd
�� d S ) z�
Two L{dns.Record_SRV} instances compare equal if and only if they have
the same priority, weight, port, target, and ttl.
r% ra r� r; r� r� r� i, rn i� N)r� r r� r r r r r� � s2 �����zEqualityTests.test_srvc
C s� | � t�ddddddd�t�ddddddd�t�ddddddd�� | � t�ddddddd�t�ddddddd�t�ddddddd�� | � t�ddddddd�t�ddddddd�t�ddd dddd�� | � t�ddddddd�t�ddddddd�t�dddd
ddd�� | � t�ddddddd�t�ddddddd�t�ddddd dd�� | � t�ddddddd�t�ddddddd�t�ddddd d d�� | � t�ddddddd�t�ddddddd�t�ddddd dd
�� dS )z�
Two L{dns.Record_NAPTR} instances compare equal if and only if they
have the same order, preference, flags, service, regexp, replacement,
and ttl.
rt r� r� r� s /foo/bar/r� r� ry � pr� s /bar/foo/s quuxrv N)r� r r� r r r r r� � sF �������zEqualityTests.test_naptrc C s� | � t�ddd�t�ddd�t�ddd�� | � t�ddd�t�ddd�t�ddd�� | � t�ddd�t�ddd�t�ddd�� dS )z�
Two L{dns.Record_AFSDB} instances compare equal if and only if they
have the same subtype, hostname, and ttl.
rt r; r� rn ry N)r� r r� r r r r r� � s � � �zEqualityTests.test_afsdbc C s� | � t�ddd�t�ddd�t�ddd�� | � t�ddd�t�ddd�t�ddd�� | � t�ddd�t�ddd�t�ddd�� dS )zx
Two L{Record_RP} instances compare equal if and only if they have the
same mbox, txt, and ttl.
r� s
alice is nicer% r� s alice is not nicer� N)r� r r� r r r r r� � s � � �zEqualityTests.test_rpc C s� | � t�ddd�t�ddd�t�ddd�� | � t�ddd�t�ddd�t�ddd�� | � t�ddd�t�ddd�t�ddd�� dS )z}
Two L{dns.Record_HINFO} instances compare equal if and only if they
have the same cpu, os, and ttl.
zx86-64Zplan9r% Zi386Zplan11r� N)r� r r� r r r r r�
s � � �zEqualityTests.test_hinfoc C s� | � t�ddd�t�ddd�t�ddd�� | � t�ddd�t�ddd�t�ddd�� | � t�ddd�t�ddd�t�ddd�� dS )z�
Two L{dns.Record_MINFO} instances compare equal if and only if they
have the same rmailbx, emailbx, and ttl.
s rmailboxs emailboxr% s someplaces somethingr� N)r� r r� r r r r r� # s � � �zEqualityTests.test_minfoc C s� | � t�ddd�t�ddd�t�ddd�� | � t�ddd�t�ddd�t�ddd�� | � t�ddd�t�ddd�t�ddd�� dS )z�
Two L{dns.Record_MX} instances compare equal if and only if they have
the same preference, name, and ttl.
r% rn ra r� s example.netr� N)r� r r� r r r r r� 9 s � � �zEqualityTests.test_mxc
C s� | � tjdddd�tjdddd�tjddddd�� | � tjdddd�tjdddd�tjdddd�� | � tjdddd�tjdddd�tjdddd�� dS )zw
Two L{dns.Record_TXT} instances compare equal if and only if they have
the same data and ttl.
r� rT r% rN �bazr� N)r� r r� r r r r r� O s ���zEqualityTests.test_txtc
C s� | � tjdddd�tjdddd�tjddddd�� | � tjdddd�tjdddd�tjdddd�� | � tjdddd�tjdddd�tjdddd�� dS )zs
L{dns.Record_SPF} instances compare equal if and only if they have the
same data and ttl.
r� rT r% rN r� r� N)r� r r� r r r r r� e s ���zEqualityTests.test_spfc C s� | � tjddd�tjddd�tjddd�� | � tjddd�tjddd�tjddd�� | � tjddd�tjddd�tjddd�� dS )zv
L{dns.UnknownRecord} instances compare equal if and only if they have
the same data and ttl.
r� r% rN ZfoobarrT r� N)r� r rX r r r r r` { s � � �zEqualityTests.test_unknownN)"r r r r r� r� rs r� r� rp rr rt rv rx rz r| r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r` r r r r r� W s> ,0 +r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �
RRHeaderTestsz2
Tests for L{twisted.names.dns.RRHeader}.
c C s& | � ttjdtjtjdt�d�� dS )z�
Attempting to create a L{dns.RRHeader} instance with a negative TTL
causes L{ValueError} to be raised.
� example.comr^ r< N�r. r/ r rW rZ r[ r\ r r r r �test_negativeTTL� s �z RRHeaderTests.test_negativeTTLc C s6 d}t �dt jt j|t �d��}| �|jt|�� dS )z<
L{dns.RRHeader} converts TTLs to integers.
g������^@r� r< N)r rW rZ r[ r\ r rO �int)r Z
ttlAsFloat�headerr r r �test_nonIntegralTTL� s �z!RRHeaderTests.test_nonIntegralTTLc C s& | � ttjdtjtjdt�d�� dS )z�
Attempting to create a L{dns.RRHeader} instance with a TTL
that L{int} cannot convert to an integer raises a L{TypeError}.
r� zthis is not a numberr< Nr� r r r r �!test_nonNumericTTLRaisesTypeError� s �z/RRHeaderTests.test_nonNumericTTLRaisesTypeErrorN)r r r r r� r� r� r r r r r� � s
r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�NameToLabelsTestsz7
Tests for L{twisted.names.dns._nameToLabels}.
c C s | � t�d�dg� dS )zp
L{dns._nameToLabels} returns a list containing a single
empty label for an empty name.
r� N�r r Z
_nameToLabelsr r r r �
test_empty� s z NameToLabelsTests.test_emptyc C s | � t�d�dg� dS )z
L{dns._nameToLabels} returns a list containing a single
empty label for a name containing only a dot.
� .r� Nr� r r r r � test_onlyDot� s z NameToLabelsTests.test_onlyDotc C s | � t�d�ddg� dS )z{
L{dns._nameToLabels} returns a list ending with an empty
label for a name without a trailing dot.
� comr� Nr� r r r r �test_withoutTrailingDot� s z)NameToLabelsTests.test_withoutTrailingDotc C s | � t�d�ddg� dS )zx
L{dns._nameToLabels} returns a list ending with an empty
label for a name with a trailing dot.
s com.r� r� Nr� r r r r �test_withTrailingDot� s z&NameToLabelsTests.test_withTrailingDotc C s" | � t�d�ddddddg� dS ) zt
L{dns._nameToLabels} returns a list containing entries
for all labels in a subdomain name.
s foo.bar.baz.example.com.r� r� r� s exampler� r� Nr� r r r r �test_subdomain� s �z NameToLabelsTests.test_subdomainc C s | � t�d�dddg� dS )z`
L{dns._nameToLabels} preserves the case of ascii
characters in labels.
� EXAMPLE.COMs EXAMPLEs COMr� Nr� r r r r �test_casePreservation� s �z'NameToLabelsTests.test_casePreservationN)
r r r r r� r� r� r� r� r� r r r r r� � s
r� c C s | � t�||�d||f � dS )a\
Assert that C{descendant} *is* a subdomain of C{ancestor}.
@type testCase: L{unittest.SynchronousTestCase}
@param testCase: The test case on which to run the assertions.
@type descendant: C{str}
@param descendant: The subdomain name to test.
@type ancestor: C{str}
@param ancestor: The superdomain name to test.
z%r is not a subdomain of %rN)r r �_isSubdomainOf��testCaseZ
descendantZancestorr r r �assertIsSubdomainOf� s
�r� c C s | � t�||�d||f � dS )a`
Assert that C{descendant} *is not* a subdomain of C{ancestor}.
@type testCase: L{unittest.SynchronousTestCase}
@param testCase: The test case on which to run the assertions.
@type descendant: C{str}
@param descendant: The subdomain name to test.
@type ancestor: C{str}
@param ancestor: The superdomain name to test.
z%r is a subdomain of %rN)r� r r� r� r r r �assertIsNotSubdomainOf s
�r� c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�IsSubdomainOfTestsz8
Tests for L{twisted.names.dns._isSubdomainOf}.
c C s t | dd� dS )z[
L{dns._isSubdomainOf} returns C{True} for identical
domain names.
r; N�r� r r r r �test_identical s z!IsSubdomainOfTests.test_identicalc C s t | dd� dS )z�
L{dns._isSubdomainOf} returns C{True} when the first
name is an immediate descendant of the second name.
r? r; Nr� r r r r � test_parent' s z IsSubdomainOfTests.test_parentc C s t | dd� dS )z
L{dns._isSubdomainOf} returns C{True} when the first
name is a distant descendant of the second name.
s foo.bar.baz.example.comr� Nr� r r r r �test_distantAncestor/ s z'IsSubdomainOfTests.test_distantAncestorc C s t | dd� dS )zw
L{dns._isSubdomainOf} returns C{False} when the first
name is an ancestor of the second name.
r; r? N�r� r r r r �test_superdomain7 s z#IsSubdomainOfTests.test_superdomainc C s t | dd� dS )zs
L{dns._isSubdomainOf} returns C{False} if the first name
is a sibling of the second name.
r? s bar.example.comNr� r r r r � test_sibling? s z IsSubdomainOfTests.test_siblingc C s t | dd� dS )zx
L{dns._isSubdomainOf} returns C{False} even when domain
names happen to share a common suffix.
s foo.myexample.comr; Nr� r r r r �test_unrelatedCommonSuffixG s z-IsSubdomainOfTests.test_unrelatedCommonSuffixc C s t | dd� dS )z�
L{dns._isSubdomainOf} returns C{True} if the first name
is a subdomain of the second name but the first name has a
trailing ".".
� foo.example.com.r; Nr� r r r r � test_subdomainWithTrailingDotO s z0IsSubdomainOfTests.test_subdomainWithTrailingDotc C s t | dd� dS )z�
L{dns._isSubdomainOf} returns C{True} if the first name
is a subdomain of the second name but the second name has a
trailing ".".
r? � example.com.Nr� r r r r � test_superdomainWithTrailingDotX s z2IsSubdomainOfTests.test_superdomainWithTrailingDotc C s t | dd� dS )z�
L{dns._isSubdomainOf} returns C{True} if the first name
is a subdomain of the second name and both names have a
trailing ".".
r� r� Nr� r r r r �test_bothWithTrailingDota s z+IsSubdomainOfTests.test_bothWithTrailingDotc C s t | dd� dS )zw
L{dns._isSubdomainOf} returns C{False} if the first name
is empty and the second name is not.
r� r; Nr� r r r r �test_emptySubdomainj s z&IsSubdomainOfTests.test_emptySubdomainc C s t | dd� dS )zv
L{dns._isSubdomainOf} returns C{True} if the second name
is empty and the first name is not.
r? r� Nr� r r r r �test_emptySuperdomainr s z(IsSubdomainOfTests.test_emptySuperdomainc C s t | dd� t | dd� dS )z`
L{dns._isSubdomainOf} does case-insensitive comparison
of name labels.
r? r� s FOO.EXAMPLE.COMr; Nr� r r r r � test_caseInsensitiveComparisonz s z1IsSubdomainOfTests.test_caseInsensitiveComparisonN)r r r r r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� s r� c @ s* e Zd ZdZeddd��Zedd� �ZdS ) �OPTNonStandardAttributesz�
Generate byte and instance representations of an L{dns._OPTHeader}
where all attributes are set to non-default values.
For testing whether attributes have really been read from the byte
string during decoding.
Fc C s d}|r d}d| S )aT
Return L{bytes} representing an encoded OPT record.
@param excludeName: A flag that controls whether to exclude
the name field. This allows a non-standard name to be
prepended during the test.
@type excludeName: L{bool}
@param excludeOptions: A flag that controls whether to exclude
the RDLEN field. This allows encoded variable options to be
appended during the test.
@type excludeOptions: L{bool}
@return: L{bytes} representing the encoded OPT record returned
by L{object}.
s r� s ) � r )rM � excludeName�excludeOptionsZrdlenr r r r8 � s �z OPTNonStandardAttributes.bytesc C s t jddddd�S )z�
Return a new L{dns._OPTHeader} instance.
@return: A L{dns._OPTHeader} instance with attributes that
match the encoded record returned by L{bytes}.
� ry rz T)�udpPayloadSize�
extendedRCODE�version�dnssecOK)r �
_OPTHeader�rM r r r r � s �z OPTNonStandardAttributes.objectN)FF)r r r r � classmethodr8 r r r r r r� � s
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&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@S )A�OPTHeaderTestsz4
Tests for L{twisted.names.dns._OPTHeader}.
c C s t tjtj� dS )zA
L{dns._OPTHeader} implements L{dns.IEncodable}.
N)r r �
IEncodabler� r r r r �test_interface� s z OPTHeaderTests.test_interfacec C s | � t�� jt�d�� dS )zq
L{dns._OPTHeader.name} is an instance attribute whose value is
fixed as the root domain
r� N)r r r� r7 r4 r r r r rs � s zOPTHeaderTests.test_namec C s$ t �� }| �tt|dt �d�� dS )z5
L{dns._OPTHeader.name} is readonly.
r7 r; N)r r� r. �AttributeError�setattrr4 �r �hr r r �test_nameReadonly� s �z OPTHeaderTests.test_nameReadonlyc C s | � t�� jd� dS )z^
L{dns._OPTHeader.type} is an instance attribute with fixed value
41.
�) N)r r r� rL r r r r � test_type� s zOPTHeaderTests.test_typec C s t �� }| �tt|dt j� dS )z5
L{dns._OPTHeader.type} is readonly.
rL N)r r� r. r� r� rZ r� r r r �test_typeReadonly� s �z OPTHeaderTests.test_typeReadonlyc C s | � t�� jd� dS )zt
L{dns._OPTHeader.udpPayloadSize} defaults to 4096 as
recommended in rfc6891 section-6.2.5.
� N�r r r� r� r r r r �test_udpPayloadSize� s z"OPTHeaderTests.test_udpPayloadSizec C s | � tjdd�jd� dS )z`
L{dns._OPTHeader.udpPayloadSize} can be overridden in the
constructor.
r� �r� Nr� r r r r �test_udpPayloadSizeOverride� s z*OPTHeaderTests.test_udpPayloadSizeOverridec C s | � t�� jd� dS )z@
L{dns._OPTHeader.extendedRCODE} defaults to 0.
r N�r r r� r� r r r r �test_extendedRCODE� s z!OPTHeaderTests.test_extendedRCODEc C s | � tjdd�jd� dS )z_
L{dns._OPTHeader.extendedRCODE} can be overridden in the
constructor.
rt �r� Nr� r r r r �test_extendedRCODEOverride
s z)OPTHeaderTests.test_extendedRCODEOverridec C s | � t�� jd� dS )z:
L{dns._OPTHeader.version} defaults to 0.
r N�r r r� r� r r r r � test_version
s zOPTHeaderTests.test_versionc C s | � tjdd�jd� dS )zY
L{dns._OPTHeader.version} can be overridden in the
constructor.
rt �r� Nr� r r r r �test_versionOverride
s z#OPTHeaderTests.test_versionOverridec C s | � t�� j� dS )z?
L{dns._OPTHeader.dnssecOK} defaults to False.
N)r� r r� r� r r r r �
test_dnssecOK
s z OPTHeaderTests.test_dnssecOKc C s | � tjdd�j� dS )zZ
L{dns._OPTHeader.dnssecOK} can be overridden in the
constructor.
T�r� N)r r r� r� r r r r �test_dnssecOKOverride
s z$OPTHeaderTests.test_dnssecOKOverridec C s | � t�� jg � dS )zC
L{dns._OPTHeader.options} defaults to empty list.
N)r r r� �optionsr r r r � test_options'
s zOPTHeaderTests.test_optionsc C s" t jdgd�}| �|jdg� dS )zY
L{dns._OPTHeader.options} can be overridden in the
constructor.
)rt rt � �r� N)r r� r r� r� r r r �test_optionsOverride.
s z#OPTHeaderTests.test_optionsOverridec C s, t � }t�� �|� | �|�� t�� � dS )z�
L{dns._OPTHeader.encode} packs the header fields and writes
them to a file like object passed in as an argument.
N)r r� r r@ r rA r8 )r �br r r rD 7
s �zOPTHeaderTests.test_encodec C sR t �� }t�dd�t�dd�g|_t� }|�|� | �|�� t j dd�d � dS ) z�
L{dns._OPTHeader.options} is a list of L{dns._OPTVariableOption}
instances which are packed into the rdata area of the header.
rt � foobarbazr� � quxT�r� � foobarbaz quxN)
r� r r �_OPTVariableOptionr� r r@ r rA r8 )r r� r� r r r �test_encodeWithOptionsE
s
�
��z%OPTHeaderTests.test_encodeWithOptionsc C s. t �� }|�tt�� �� | �|t�� � dS )z�
L{dns._OPTHeader.decode} unpacks the header fields from a file
like object and populates the attributes of an existing
L{dns._OPTHeader} instance.
N)r r� r< r r� r8 r r )r �
decodedHeaderr r r r> b
s �zOPTHeaderTests.test_decodec C s: t t�� �}t�� }|�|� | �|�� t|� � �� dS )zk
L{dns._OPTHeader.decode} reads all the bytes of the record
that is being decoded.
N�
r r� r8 r r� r< r rf rG rA �r r� r r r r �test_decodeAllExpectedBytesp
s
z*OPTHeaderTests.test_decodeAllExpectedBytesc C sF t t�� d �}t�� }|�|� | �|�� t|� � �td� � dS )z�
L{dns._OPTHeader.decode} reads only the bytes from the current
file position to the end of the record that is being
decoded. Trailing bytes are not consumed.
s xxxxNr r r r r � test_decodeOnlyExpectedBytes~
s �
z+OPTHeaderTests.test_decodeOnlyExpectedBytesc C s> t tjdd�d �}t�� }|�|� | �|jt�d�� dS )z�
L{dns._OPTHeader.decode} discards the name which is encoded in
the supplied bytes. The name attribute of the resulting
L{dns._OPTHeader} instance will always be L{dns.Name(b'')}.
T)r� r: r� N) r r� r8 r r� r< r r7 r4 �r r� r� r r r �test_decodeDiscardsName�
s �
z&OPTHeaderTests.test_decodeDiscardsNamec C s0 t tjdd�d �}t�� }| �t|j|� dS )zj
L{dns._OPTHeader.decode} raises an exception if the supplied
RDLEN is too short.
Tr� s N�r r� r8 r r� r. r� r< r r r r �test_decodeRdlengthTooShort�
s
��z*OPTHeaderTests.test_decodeRdlengthTooShortc C s0 t tjdd�d �}t�� }| �t|j|� dS )zi
L{dns._OPTHeader.decode} raises an exception if the supplied
RDLEN is too long.
Tr� s Nr r r r r �test_decodeRdlengthTooLong�
s
�� z)OPTHeaderTests.test_decodeRdlengthTooLongc C sL t tjdd�d �}t�� }|�|� | �|jt�dd�t�dd�g� dS ) z�
If the OPT bytes contain variable options,
L{dns._OPTHeader.decode} will populate a list
L{dns._OPTHeader.options} with L{dns._OPTVariableOption}
instances.
Tr� r� rt r� r� r� N) r r� r8 r r� r< r r� r r r r r �test_decodeWithOptions�
s
��
��z%OPTHeaderTests.test_decodeWithOptionsc C sX t jdt jddt �d�d�}t j�|�}t jddddt jdd d
�gd �}| �||� d S )
z�
L{_OPTHeader.fromRRHeader} accepts an L{RRHeader} instance and
returns an L{_OPTHeader} instance whose attribute values have
been derived from the C{cls}, C{ttl} and C{payload} attributes
of the original header.
r; i�� l �} s �� abcrP � � Tr� )�code�data)r� r� r� r� r� N)r rW ZOPTrX r� Z fromRRHeaderr r )r Z
genericHeaderZdecodedOptHeaderZexpectedOptHeaderr r r �test_fromRRHeader�
s � �z OPTHeaderTests.test_fromRRHeaderc C s | � tt�� �d� dS )z�
L{dns._OPTHeader.__repr__} displays the name and type and all
the fixed and extended header values of the OPT record.
zb<_OPTHeader name= type=41 udpPayloadSize=4096 extendedRCODE=0 version=0 dnssecOK=False options=[]>N)r r� r r� r r r r � test_repr�
s
�zOPTHeaderTests.test_reprc C s* | � tjdd�tjdd�tjdd�� dS )zh
Two L{OPTHeader} instances compare equal if they have the same
udpPayloadSize.
r� r� r� N�r r r� r r r r �test_equalityUdpPayloadSize
s
�z*OPTHeaderTests.test_equalityUdpPayloadSizec C s* | � tjdd�tjdd�tjdd�� dS )zg
Two L{OPTHeader} instances compare equal if they have the same
extendedRCODE.
rt r� r� Nr r r r r �test_equalityExtendedRCODE s
�z)OPTHeaderTests.test_equalityExtendedRCODEc C s* | � tjdd�tjdd�tjdd�� dS )za
Two L{OPTHeader} instances compare equal if they have the same
version.
rt r� r� Nr r r r r �test_equalityVersion s
�z#OPTHeaderTests.test_equalityVersionc C s* | � tjdd�tjdd�tjdd�� dS )zh
Two L{OPTHeader} instances compare equal if they have the same
dnssecOK flags.
Tr� FNr r r r r �test_equalityDnssecOK+ s
�z$OPTHeaderTests.test_equalityDnssecOKc C sH | � tjt�dd�gd�tjt�dd�gd�tjt�dd�gd�� dS )za
Two L{OPTHeader} instances compare equal if they have the same
options.
rt � xr� r� � yN)r r r� r r r r r �test_equalityOptions6 s
�z#OPTHeaderTests.test_equalityOptionsN)#r r r r r� rs r� r� r� r� r� r� r� r� r� r� r� r� r� rD r r> r r r r
r r r r r r r r r r r r r r� � s@ r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�OPTVariableOptionTestsz.
Tests for L{dns._OPTVariableOption}.
c C s t tjtj� dS )zI
L{dns._OPTVariableOption} implements L{dns.IEncodable}.
N)r r r� r r r r r r� F s z%OPTVariableOptionTests.test_interfacec C s, t �dd�}| �|jd� | �|jd� dS )z�
L{dns._OPTVariableOption.__init__} requires code and data
arguments which are saved as public instance attributes.
rt r N)r r r r r r� r r r �test_constructorArgumentsM s z0OPTVariableOptionTests.test_constructorArgumentsc C s | � tt�dd��d� dS )zf
L{dns._OPTVariableOption.__repr__} displays the code and data
of the option.
rt r z"<_OPTVariableOption code=1 data=x>N)r r� r r r r r r r W s �z OPTVariableOptionTests.test_reprc C sP | � t�dd�t�dd�t�dd�� | � t�dd�t�dd�t�dd�� dS )zs
Two OPTVariableOption instances compare equal if they have the same
code and data values.
rt r r� r N)r r r r r r r rk d s
�
�z$OPTVariableOptionTests.test_equalityc C s0 t �dd�}t� }|�|� | �|�� d� dS )z�
L{dns._OPTVariableOption.encode} encodes the code and data
instance attributes to a byte string which also includes the
data length.
rt � foobar�
foobarN)r r r r@ r rA )r �or� r r r rD t s
�z"OPTVariableOptionTests.test_encodec C s: t d�}t�� }|�|� | �|jd� | �|jd� dS )z�
L{dns._OPTVariableOption.decode} is a classmethod that decodes
a byte string and returns a L{dns._OPTVariableOption} instance.
r rt r N)r r r r< r r r )r r� r r r r r> � s �
z"OPTVariableOptionTests.test_decodeN)
r r r r r� r r rk rD r> r r r r r B s
r c @ s e Zd ZdZdd� ZdS )�
RaisedArgszp
An exception which can be raised by fakes to test that the fake is called
with expected arguments.
c C s || _ || _dS )z�
Store the positional and keyword arguments as attributes.
@param args: The positional args.
@param kwargs: The keyword args.
N�r r �r r r r r r r0 � s zRaisedArgs.__init__N)r r r r r0 r r r r r � s r c @ s( e Zd ZdZedd� �Zedd� �ZdS )� MessageEmptyz^
Generate byte string and constructor arguments for an empty
L{dns._EDNSMessage}.
c C s dS )��
Bytes which are expected when encoding an instance constructed using
C{kwargs} and which are expected to result in an identical instance when
decoded.
@return: The L{bytes} of a wire encoded message.
s �� r r� r r r r8 � s
�zMessageEmpty.bytesc C s t ddtjddddddd� S )��
Keyword constructor arguments which are expected to result in an
instance which returns C{bytes} when encoded.
@return: A L{dict} of keyword arguments.
r� Tr� N� r� r� r� r� r� r� r� r� � ednsVersion)�dictr � OP_STATUSr� r r r r � s �zMessageEmpty.kwargsN�r r r r r� r8 r r r r r r# � s
r# c @ s( e Zd ZdZedd� �Zedd� �ZdS )�MessageTruncatedz=
An empty response message whose TR bit is set to 1.
c C s dS )r$ s � r r� r r r r8 � s
�zMessageTruncated.bytesc C s t dddddddddd� S )r% r� rt r Nr& )r( r� r r r r � s �zMessageTruncated.kwargsNr* r r r r r+ � s
r+ c @ s( e Zd ZdZedd� �Zedd� �ZdS )�MessageNonAuthoritativez.
A minimal non-authoritative message.
c C s dS )r$ s r r� r r r r8 s
�z MessageNonAuthoritative.bytesc
C s( t dddtjdtjddd�dd�gd �S )
r% r� r Nr� rR rN F�rQ r� �r� r� r' rV �r( r rW r\ r� r r r r # s ���z MessageNonAuthoritative.kwargsNr* r r r r r, s
r, c @ s( e Zd ZdZedd� �Zedd� �ZdS )�MessageAuthoritativez*
A minimal authoritative message.
c C s dS )r$ s r r� r r r r8 ; s
�zMessageAuthoritative.bytesc
C s( t dddtjdtjddd�dd �gd
�S ) r% r� rt Nr� rR r rN Tr- r. r/ r� r r r r V s ���zMessageAuthoritative.kwargsNr* r r r r r0 7 s
r0 c @ s( e Zd ZdZedd� �Zedd� �ZdS )�MessageCompletez�
An example of a fully populated non-edns response message.
Contains name compression, answers, authority, and additional records.
c C s dS )r$ sn �� examplecom � ���� 'ns1�
hostmaster� ������������������ ���� �)�) ���� r r� r r r r8 p s
�zMessageComplete.bytesc C s� t ddtjdddddt�dtj�gtjdtjddtjddd d
d d d
dd�d�gtjdtjddtjddd�d�gtjdtj ddtj
ddd�d�gd� S )r% r� rt r� Nr; � �� T� ns1.example.com� hostmaster.example.com� �� ������� �� �rO r� r� r� r� r� r� r� �rL rO r� rQ �ns1.example.comrN �5.6.7.8) r� r� r� r� r� r� r� r' rT rV r� rY � r( r r) rU r* rW r� ZNSro rZ r\ r� r r r r � sd ��� ��� ����zMessageComplete.kwargsNr* r r r r r1 j s
6r1 c @ s( e Zd ZdZedd� �Zedd� �ZdS )�MessageEDNSQueryz'
A minimal EDNS query message.
c C s dS )r$ s, wwwexamplecom ) r r� r r r r8 � s
�zMessageEDNSQuery.bytesc C s, t ddtjddddddt�dtj�gg d� S )r% r ry Fs www.example.com) r� r� r� r� r� r� r� r' r� rT rY )r( r r� rU rZ r� r r r r
s �zMessageEDNSQuery.kwargsNr* r r r r r? � s
!r? c @ s( e Zd ZdZedd� �Zedd� �ZdS )�MessageEDNSCompletez�
An example of a fully populated edns response message.
Contains name compression, answers, authority, and additional records.
c C s dS )r$ sy �� examplecom � ���� 'ns1�
hostmaster� ������������������ ���� �)�) ���� ) � r r� r r r r8
s
�zMessageEDNSComplete.bytesc C s� t ddtjddddddddddt�dtj�gtjdtjd dtjd d
d d d
dddd�d�gtjdtjd dtjdd d�d�gtjd
tj d dtj
dd d�d�gd�S )r% r� rt r r� ry T� r; r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< rN r= )r� r� r� r� r� r� r� r� r' r� r� r� r� rT rV r� rY r> r� r r r r _
sn ��� ��� ����zMessageEDNSComplete.kwargsNr* r r r r r@
s
?r@ c @ s( e Zd ZdZedd� �Zedd� �ZdS )�MessageEDNSExtendedRCODEz?
An example of an EDNS message with an extended RCODE.
c C s dS )r$ s ) � r r� r r r r8 �
s
�z MessageEDNSExtendedRCODE.bytesc C s( t ddtjddddddddg g g g d�S )r% r Fi�
r� )r� r� r� r� r� r� r� r� r' r� r� rT rV r� rY )r( r r� r� r r r r �
s" �z MessageEDNSExtendedRCODE.kwargsNr* r r r r rB �
s
rB c @ s( e Zd ZdZd ZZdd� Zdd� ZdS )�MessageComparablez�
A wrapper around L{dns.Message} which is comparable so that it can be tested
using some of the L{dns._EDNSMessage} tests.
) r� r� r� r� r� r� r� r� rT rV r� rY c C s
|| _ d S r[ )�original)r rD r r r r0 �
s zMessageComparable.__init__c C s t | j|�S r[ )�getattrrD )r �keyr r r � __getattr__�
s z MessageComparable.__getattr__N)r r r r ZshowAttributesZcompareAttributesr0 rG r r r r rC �
s rC c C sZ |dkr |}i }||d�}|� }t ||�|d< |f ||i�}t ||�|d< | �||� dS )a�
Verify that an attribute has the expected default value and that a
corresponding argument passed to a constructor is assigned to that
attribute.
@param testCase: The L{TestCase} whose assert methods will be
called.
@type testCase: L{unittest.TestCase}
@param cls: The constructor under test.
@type cls: L{type}
@param argName: The name of the constructor argument under test.
@type argName: L{str}
@param defaultVal: The expected default value of C{attrName} /
C{argName}
@type defaultVal: L{object}
@param altVal: A value which is different from the default. Used to
test that supplied constructor arguments are actually assigned to the
correct attribute.
@type altVal: L{object}
@param attrName: The name of the attribute under test if different
from C{argName}. Defaults to C{argName}
@type attrName: L{str}
N��
defaultVal�altValrI rJ )rE r ) r� rM �argNamerI rJ �attrNameZactualrI r r r r �verifyConstructorArgument�
s
rM c @ s e Zd ZdZdd� Zdd� ZdS )�ConstructorTestsMixinzl
Helper methods for verifying default attribute values and corresponding
constructor arguments.
c C s t | | j|||d� dS )a�
Wrap L{verifyConstructorArgument} to provide simpler interface for
testing Message and _EDNSMessage constructor arguments.
@param argName: The name of the constructor argument.
@param defaultVal: The expected default value.
@param altVal: An alternative value which is expected to be assigned to
a correspondingly named attribute.
�r� rM rK rI rJ N)rM r )r rK rI rJ r r r �_verifyConstructorArgument s
�z0ConstructorTestsMixin._verifyConstructorArgumentc C s& |dks t �t| | j||| d� dS )a
Wrap L{verifyConstructorArgument} to provide simpler interface for
testing _EDNSMessage constructor flags.
@param argName: The name of the constructor flag argument
@param defaultVal: The expected default value of the flag
)TFrO N)�AssertionErrorrM r )r rK rI r r r �_verifyConstructorFlag+ s �z,ConstructorTestsMixin._verifyConstructorFlagN)r r r r rP rR r r r r rN s rN c @ sx e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd S ) �CommonConstructorTestsMixina�
Tests for constructor arguments and their associated attributes that are
common to both L{twisted.names.dns._EDNSMessage} and L{dns.Message}.
TestCase classes that use this mixin must provide a C{messageFactory} method
which accepts any argment supported by L{dns.Message.__init__}.
TestCases must also mixin ConstructorTestsMixin which provides some custom
assertions for testing constructor arguments.
c C s | j dddd� dS )zk
L{dns._EDNSMessage.id} defaults to C{0} and can be overridden in
the constructor.
r� r rt rH N�rP r r r r r
E s z#CommonConstructorTestsMixin.test_idc C s | j ddd� dS )zs
L{dns._EDNSMessage.answer} defaults to C{False} and can be overridden in
the constructor.
r� F�rI N�rR r r r r r M s z'CommonConstructorTestsMixin.test_answerc C s | j dtjtjd� dS )zz
L{dns._EDNSMessage.opCode} defaults to L{dns.OP_QUERY} and can be
overridden in the constructor.
r� rH N)rP r r� r) r r r r r U s
�z'CommonConstructorTestsMixin.test_opCodec C s | j ddd� dS )zq
L{dns._EDNSMessage.auth} defaults to C{False} and can be overridden in
the constructor.
r� FrU NrV r r r r r ^ s z%CommonConstructorTestsMixin.test_authc C s | j ddd� dS )zr
L{dns._EDNSMessage.trunc} defaults to C{False} and can be overridden in
the constructor.
r� FrU NrV r r r r r f s z&CommonConstructorTestsMixin.test_truncc C s | j ddd� dS )zs
L{dns._EDNSMessage.recDes} defaults to C{False} and can be overridden in
the constructor.
r� FrU NrV r r r r r n s z'CommonConstructorTestsMixin.test_recDesc C s | j ddd� dS )zr
L{dns._EDNSMessage.recAv} defaults to C{False} and can be overridden in
the constructor.
r� FrU NrV r r r r r v s z&CommonConstructorTestsMixin.test_recAvc C s | j dddd� dS )zn
L{dns._EDNSMessage.rCode} defaults to C{0} and can be overridden in the
constructor.
r� r r2 rH NrT r r r r r ~ s z&CommonConstructorTestsMixin.test_rCodec C s | j dddd� dS )zr
L{dns._EDNSMessage.maxSize} defaults to C{512} and can be overridden in
the constructor.
r� r� rA rH NrT r r r r r � s z(CommonConstructorTestsMixin.test_maxSizec C s | � | �� jg � dS )z@
L{dns._EDNSMessage.queries} defaults to C{[]}.
N)r r rT r r r r r" � s z(CommonConstructorTestsMixin.test_queriesc C s | � | �� jg � dS )z@
L{dns._EDNSMessage.answers} defaults to C{[]}.
N)r r rV r r r r r& � s z(CommonConstructorTestsMixin.test_answersc C s | � | �� jg � dS )zB
L{dns._EDNSMessage.authority} defaults to C{[]}.
N)r r r� r r r r r+ � s z*CommonConstructorTestsMixin.test_authorityc C s | � | �� jg � dS )zC
L{dns._EDNSMessage.additional} defaults to C{[]}.
N)r r rY r r r r r- � s z+CommonConstructorTestsMixin.test_additionalN)r r r r r
r r r r r r r r r" r& r+ r- r r r r rS : s
rS c @ s e Zd ZdZejZdS )�EDNSMessageConstructorTestszt
Tests for L{twisted.names.dns._EDNSMessage} constructor arguments that are
shared with L{dns.Message}.
N�r r r r r � _EDNSMessager r r r r rW � s rW c @ s e Zd ZdZejZdS )�MessageConstructorTestszt
Tests for L{twisted.names.dns.Message} constructor arguments that are shared
with L{dns._EDNSMessage}.
N)r r r r r rH r r r r r rZ � s rZ c @ s� e Zd ZdZejZdd� Zdd� Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"S )#�EDNSMessageSpecificsTestsz�
Tests for L{dns._EDNSMessage}.
These tests are for L{dns._EDNSMessage} APIs which are not shared with
L{dns.Message}.
c C s | j dddd� dS )zt
L{dns._EDNSMessage.ednsVersion} defaults to C{0} and can be overridden
in the constructor.
r' r NrH rT r r r r �test_ednsVersion� s
�z*EDNSMessageSpecificsTests.test_ednsVersionc C s | j ddd� dS )zu
L{dns._EDNSMessage.dnssecOK} defaults to C{False} and can be overridden
in the constructor.
r� FrU NrV r r r r r� � s z'EDNSMessageSpecificsTests.test_dnssecOKc C s | j ddd� dS )zz
L{dns._EDNSMessage.authenticData} defaults to C{False} and can be
overridden in the constructor.
r� FrU NrV r r r r r � s z,EDNSMessageSpecificsTests.test_authenticDatac C s | j ddd� dS )z}
L{dns._EDNSMessage.checkingDisabled} defaults to C{False} and can be
overridden in the constructor.
r� FrU NrV r r r r r � s z/EDNSMessageSpecificsTests.test_checkingDisabledc C s. | j t�d�gd�}| �|jt�d�g� dS )zS
L{dns._EDNSMessage.queries} can be overridden in the constructor.
r; r N)r r rU r rT r� r r r �test_queriesOverride� s
�z.EDNSMessageSpecificsTests.test_queriesOverridec C sB | j tjdt�d�d�gd�}| �|jtjdt�d�d�g� dS )zS
L{dns._EDNSMessage.answers} can be overridden in the constructor.
r; rR r� r# N)r r rW r\ r rV r� r r r �test_answersOverride� s ����z.EDNSMessageSpecificsTests.test_answersOverridec C sF | j tjdtjt�� d�gd�}| �|jtjdtjt�� d�g� dS )�U
L{dns._EDNSMessage.authority} can be overridden in the constructor.
r; r' r( N)r r rW r* r� r r� r� r r r �test_authorityOverride s ���
��z0EDNSMessageSpecificsTests.test_authorityOverridec C sB | j tjdt�d�d�gd�}| �|jtjdt�d�d�g� dS )r_ r; rR r� r, N)r r rW r\ r rY r� r r r �test_additionalOverride s ����z1EDNSMessageSpecificsTests.test_additionalOverridec C s | � dt| �� �� dS )z�
L{dns._EDNSMessage.__repr__} omits field values and sections which are
identical to their defaults. The id field value is always shown.
z<_EDNSMessage id=0>N�r r� r r r r r r� & s
�z+EDNSMessageSpecificsTests.test_reprDefaultsc
C s. | j ddddddddd�}| �dt|�� dS )zR
L{dns._EDNSMessage.__repr__} displays flags if they are L{True}.
T)r� r� r� r� r� r� r� r� z`<_EDNSMessage id=0 flags=answer,auth,trunc,recDes,recAv,authenticData,checkingDisabled,dnssecOK>N�r r r� r� r r r r� 1 s ��z-EDNSMessageSpecificsTests.test_reprFlagsIfSetc C s( | j dddddd�}| �dt|�� dS ) zp
L{dns._EDNSMessage.__repr__} displays field values if they differ from
their defaults.
r% ra r� r� r� )r� r� r� r� r' zA<_EDNSMessage id=10 opCode=20 rCode=30 maxSize=40 ednsVersion=50>Nrc r� r r r r� B s ��z2EDNSMessageSpecificsTests.test_reprNonDefautFieldsc C sL | � � }dddg|_dddg|_ddd g|_d
d d g|_| �d
t|�� dS )r� rt r� ry rz rv r� r� r� r� r% r� r� zc<_EDNSMessage id=0 queries=[1, 2, 3] answers=[4, 5, 6] authority=[7, 8, 9] additional=[10, 11, 12]>N)r rT rV r� rY r r� r� r r r r� U s �z5EDNSMessageSpecificsTests.test_reprNonDefaultSectionsc C sR G dd� dt �}t�� }||_t � }| �t|j|�}| �|fi f|j|j f� dS )z�
L{dns._EDNSMessage.fromString} calls L{dns._EDNSMessage._messageFactory}
to create a new L{dns.Message} instance which is used to decode the
supplied bytes.
c @ s e Zd ZdZdd� ZdS )zUEDNSMessageSpecificsTests.test_fromStrCallsMessageFactory.<locals>.FakeMessageFactory�/
Fake message factory.
c _ s t ||��dS )z�
Fake fromStr method which raises the arguments it was passed.
@param args: positional arguments
@param kwargs: keyword arguments
N�r r" r r r rS u s z]EDNSMessageSpecificsTests.test_fromStrCallsMessageFactory.<locals>.FakeMessageFactory.fromStrN�r r r r rS r r r r �FakeMessageFactoryq s rg N)
r r rY �_messageFactoryr. r rS r r r )r rg r� �
dummyBytesr� r r r � test_fromStrCallsMessageFactoryk s
�z9EDNSMessageSpecificsTests.test_fromStrCallsMessageFactoryc sf t �� }G dd� d�}|� � � fdd�|_dd� }||_| �t|jd�}| �� fi f|j|j f� dS ) z~
L{dns._EDNSMessage.fromString} calls L{dns._EDNSMessage._fromMessage}
with a L{dns.Message} instance
c @ s e Zd ZdZdd� ZdS )zREDNSMessageSpecificsTests.test_fromStrCallsFromMessage.<locals>.FakeMessageFactoryrd c S s dS )zw
A noop fake version of fromStr
@param bytes: the bytes to be decoded
Nr )r r8 r r r rS � s zZEDNSMessageSpecificsTests.test_fromStrCallsFromMessage.<locals>.FakeMessageFactory.fromStrNrf r r r r rg � s rg c s � S r[ r r �Z fakeMessager r �<lambda>� r� zHEDNSMessageSpecificsTests.test_fromStrCallsFromMessage.<locals>.<lambda>c _ s t | |��d S r[ re r! r r r �fakeFromMessage� s zOEDNSMessageSpecificsTests.test_fromStrCallsFromMessage.<locals>.fakeFromMessager� N)
r rY rh � _fromMessager. r rS r r r )r r� rg rm r� r rk r � test_fromStrCallsFromMessage� s
�z6EDNSMessageSpecificsTests.test_fromStrCallsFromMessagec C s@ t �� }dd� }||_| �t|j�}| �di f|j|jf� dS )zP
L{dns._EDNSMessage.toStr} calls L{dns._EDNSMessage._toMessage}
c _ s t | |��d S r[ re r! r r r �
fakeToMessage� s zIEDNSMessageSpecificsTests.test_toStrCallsToMessage.<locals>.fakeToMessager N) r rY �
_toMessager. r r] r r r )r r� rp r� r r r �test_toStrCallsToMessage� s
�z2EDNSMessageSpecificsTests.test_toStrCallsToMessagec sH t �� }t� �G �fdd�dt�� � fdd�}||_| ��|�� � dS )z}
L{dns._EDNSMessage.toStr} calls C{toStr} on the message returned by
L{dns._EDNSMessage._toMessage}.
c s e Zd ZdZ� fdd�ZdS )zLEDNSMessageSpecificsTests.test_toStrCallsToMessageToStr.<locals>.FakeMessagez&
Fake Message
c s � S )zk
Fake toStr which returns dummyBytes.
@return: dummyBytes
r r �ri r r r] � s zREDNSMessageSpecificsTests.test_toStrCallsToMessageToStr.<locals>.FakeMessage.toStrN)r r r r r] r rs r r � FakeMessage� s rt c s � � S r[ r r! )rt r r rp � s zNEDNSMessageSpecificsTests.test_toStrCallsToMessageToStr.<locals>.fakeToMessageN)r rY r rq r r] )r r� rp r )rt ri r � test_toStrCallsToMessageToStr� s �z7EDNSMessageSpecificsTests.test_toStrCallsToMessageToStrN)r r r r r rY r r\ r� r r r] r^ r` ra r� r� r� r� rj ro rr ru r r r r r[ � s$ r[ c @ s� e Zd ZdZejZdd� Zdd� Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$S )%�EDNSMessageEqualityTestsz�
Tests for equality between L(dns._EDNSMessage} instances.
These tests will not work with L{dns.Message} because it does not use
L{twisted.python.util.FancyEqMixin}.
c C s* | � | jdd�| jdd�| jdd�� dS )zc
Two L{dns._EDNSMessage} instances compare equal if they have the same
id.
rt r
r� Nr r r r r r
� s
�z EDNSMessageEqualityTests.test_idc C s* | � | jdd�| jdd�| jdd�� dS )zl
Two L{dns._EDNSMessage} instances compare equal if they have the same
answer flag.
Tr FNr r r r r r � s
�z$EDNSMessageEqualityTests.test_answerc C s0 | � | jtjd�| jtjd�| jtjd�� dS )zg
Two L{dns._EDNSMessage} instances compare equal if they have the same
opCode.
r N)r r r r) Z
OP_INVERSEr r r r r � s
�z$EDNSMessageEqualityTests.test_opCodec C s* | � | jdd�| jdd�| jdd�� dS )zj
Two L{dns._EDNSMessage} instances compare equal if they have the same
auth flag.
Tr FNr r r r r r s
�z"EDNSMessageEqualityTests.test_authc C s* | � | jdd�| jdd�| jdd�� dS )zk
Two L{dns._EDNSMessage} instances compare equal if they have the same
trunc flag.
Tr FNr r r r r r
s
�z#EDNSMessageEqualityTests.test_truncc C s* | � | jdd�| jdd�| jdd�� dS )zl
Two L{dns._EDNSMessage} instances compare equal if they have the same
recDes flag.
Tr FNr r r r r r s
�z$EDNSMessageEqualityTests.test_recDesc C s* | � | jdd�| jdd�| jdd�� dS )zk
Two L{dns._EDNSMessage} instances compare equal if they have the same
recAv flag.
Tr FNr r r r r r % s
�z#EDNSMessageEqualityTests.test_recAvc C s* | � | jdd�| jdd�| jdd�� dS )zf
Two L{dns._EDNSMessage} instances compare equal if they have the same
rCode.
r r r� Nr r r r r r 1 s
�z#EDNSMessageEqualityTests.test_rCodec C s* | � | jdd�| jdd�| jdd�� dS )zl
Two L{dns._EDNSMessage} instances compare equal if they have the same
ednsVersion.
rt �r' Nr r r r r r\ = s
�z)EDNSMessageEqualityTests.test_ednsVersionc C s* | � | jdd�| jdd�| jdd�� dS )zi
Two L{dns._EDNSMessage} instances compare equal if they have the same
dnssecOK.
Tr� FNr r r r r r� I s
�z&EDNSMessageEqualityTests.test_dnssecOKc C s* | � | jdd�| jdd�| jdd�� dS )zt
Two L{dns._EDNSMessage} instances compare equal if they have the same
authenticData flags.
Tr� FNr r r r r r U s
�z+EDNSMessageEqualityTests.test_authenticDatac C s* | � | jdd�| jdd�| jdd�� dS )zw
Two L{dns._EDNSMessage} instances compare equal if they have the same
checkingDisabled flags.
Tr� FNr r r r r r a s
�z.EDNSMessageEqualityTests.test_checkingDisabledc C s* | � | jdd�| jdd�| jdd�� dS )zh
Two L{dns._EDNSMessage} instances compare equal if they have the same
maxSize.
i r rA Nr r r r r r m s
�z%EDNSMessageEqualityTests.test_maxSizec C sB | � | jt�d�gd�| jt�d�gd�| jt�d�gd�� dS )zh
Two L{dns._EDNSMessage} instances compare equal if they have the same
queries.
r; r rn Nr! r r r r r" y s
�z%EDNSMessageEqualityTests.test_queriesc
C s` | � | jtjdt�d�d�gd�| jtjdt�d�d�gd�| jtjdt�d�d�gd�� dS )zh
Two L{dns._EDNSMessage} instances compare equal if they have the same
answers.
r; rR r� r# rn r$ Nr% r r r r r& � s �
�
�
�z%EDNSMessageEqualityTests.test_answersc
C sf | � | jtjdtjt�� d�gd�| jtjdtjt�� d�gd�| jtjdtjt�� d�gd�� dS )zr
Two L{dns._EDNSMessage} instances compare equal if they have the same
authority records.
r; r' r( rn Nr) r r r r r+ � s" �
�
�
�z'EDNSMessageEqualityTests.test_authorityc
C s` | � | jtjdt�d�d�gd�| jtjdt�d�d�gd�| jtjdt�d�d�gd�� dS )zs
Two L{dns._EDNSMessage} instances compare equal if they have the same
additional records.
r; rR r� r, rn Nr% r r r r r- � s �
�
�
�z(EDNSMessageEqualityTests.test_additionalN)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 rv � s& rv c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�StandardEncodingTestsMixina�
Tests for the encoding and decoding of various standard (not EDNS) messages.
These tests should work with both L{dns._EDNSMessage} and L{dns.Message}.
TestCase classes that use this mixin must provide a C{messageFactory} method
which accepts any argment supported by L{dns._EDNSMessage.__init__}.
EDNS specific arguments may be discarded if not supported by the message
class under construction.
c C s$ | � | jf t�� ��� t�� � dS )z2
An empty message can be encoded.
N)r r r# r r] r8 r r r r �test_emptyMessageEncode� s �z2StandardEncodingTestsMixin.test_emptyMessageEncodec C s2 | � � }|�t�� � | �|| j f t�� �� dS )z@
An empty message byte sequence can be decoded.
N)r rS r# r8 r r r� r r r �test_emptyMessageDecode� s z2StandardEncodingTestsMixin.test_emptyMessageDecodec C s$ | � | jf t�� ��� t�� � dS )zA
A fully populated query message can be encoded.
N)r r r1 r r] r8 r r r r �test_completeQueryEncode� s �z3StandardEncodingTestsMixin.test_completeQueryEncodec C s4 | � � }|�t�� �f | �|| j f t�� �� dS )zG
A fully populated message byte string can be decoded.
N)r rS r1 r8 r r r� r r r �test_completeQueryDecode� s z3StandardEncodingTestsMixin.test_completeQueryDecodec C s� d� dd� td�D ��}t�|�}tjdtj|d�}| �� }|j�|� |� � }| �� }|�
|� | � |jd j tj� | �
|jd j j |� dS ) zu
A I{NULL} record with an arbitrary payload can be encoded and decoded as
part of a message.
r� c S s g | ]}t �|��qS r r� r� r r r r� � s z8StandardEncodingTestsMixin.test_NULL.<locals>.<listcomp>r� r� r� r N)r� rw r r� rW r� r rV r� r] rS r6 rQ r r� r r r r� � s
z$StandardEncodingTestsMixin.test_NULLc C s$ | � | jf t�� ��� t�� � dS )zv
If the message C{authoritative} attribute is set to 0, the encoded bytes
will have AA bit 0.
N)r r r, r r] r8 r r r r �"test_nonAuthoritativeMessageEncode� s �z=StandardEncodingTestsMixin.test_nonAuthoritativeMessageEncodec C s2 | � � }|�t�� � | �|| j f t�� �� dS )z�
The L{dns.RRHeader} instances created by a message from a
non-authoritative message byte string are marked as not authoritative.
N)r rS r, r8 r r r� r r r �"test_nonAuthoritativeMessageDecode s �z=StandardEncodingTestsMixin.test_nonAuthoritativeMessageDecodec C s$ | � | jf t�� ��� t�� � dS )zv
If the message C{authoritative} attribute is set to 1, the encoded bytes
will have AA bit 1.
N)r r r0 r r] r8 r r r r � test_authoritativeMessageEncode s �z:StandardEncodingTestsMixin.test_authoritativeMessageEncodec C s2 | � � }|�t�� � | �|| j f t�� �� dS )z�
The message and its L{dns.RRHeader} instances created by C{decode} from
an authoritative message byte string, are marked as authoritative.
N)r rS r0 r8 r r r� r r r � test_authoritativeMessageDecode s �z:StandardEncodingTestsMixin.test_authoritativeMessageDecodec C s$ | � | jf t�� ��� t�� � dS )zm
If the message C{trunc} attribute is set to 1 the encoded bytes will
have TR bit 1.
N)r r r+ r r] r8 r r r r �test_truncatedMessageEncode' s �z6StandardEncodingTestsMixin.test_truncatedMessageEncodec C s2 | � � }|�t�� � | �|| j f t�� �� dS )zn
The message instance created by decoding a truncated message is marked
as truncated.
N)r rS r+ r8 r r r� r r r �test_truncatedMessageDecode1 s z6StandardEncodingTestsMixin.test_truncatedMessageDecodeN)r r r r ry rz r{ r| r� r} r~ r r� r� r� r r r r rx � s
rx c @ s e Zd ZdZejZdS )� EDNSMessageStandardEncodingTestszq
Tests for the encoding and decoding of various standard (non-EDNS) messages
by L{dns._EDNSMessage}.
NrX r r r r r� = s r� c @ s e Zd ZdZedd� �ZdS )� MessageStandardEncodingTestszl
Tests for the encoding and decoding of various standard (non-EDNS) messages
by L{dns.Message}.
c K sh | � dg �}| � dg �}| � dg �}| � dg �}| � dd� tjf | �}||_||_||_||_t|�S )a�
This function adapts constructor arguments expected by
_EDNSMessage.__init__ to arguments suitable for use with the
Message.__init__.
Also handles the fact that unlike L{dns._EDNSMessage},
L{dns.Message.__init__} does not accept queries, answers etc as
arguments.
Also removes any L{dns._EDNSMessage} specific arguments.
@param args: The positional arguments which will be passed to
L{dns.Message.__init__}.
@param kwargs: The keyword arguments which will be stripped of EDNS
specific arguments before being passed to L{dns.Message.__init__}.
@return: An L{dns.Message} instance.
rT rV r� rY r' N)r r rH rT rV r� rY rC )r rT rV r� rY r� r r r r M s z+MessageStandardEncodingTests.messageFactoryN)r r r r � staticmethodr r r r r r� G s r� c @ sf e Zd ZdZejZdd� Zdd� Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� Zdd� Zdd� ZdS )� EDNSMessageEDNSEncodingTestszz
Tests for the encoding and decoding of various EDNS messages.
These test will not work with L{dns.Message}.
c C s( | � � }|�t�� � | �|jg � dS )z�
The L(_EDNSMessage} instance created by L{dns._EDNSMessage.decode} from
an EDNS query never includes OPT records in the additional section.
N)r rS r? r8 r rY r� r r r �&test_ednsMessageDecodeStripsOptRecordsz s zCEDNSMessageEDNSEncodingTests.test_ednsMessageDecodeStripsOptRecordsc C sH t �� }t jdd�t jdd�g|_t �� }|�|�� � | �|j� dS )a
An L(_EDNSMessage} instance created from a byte string containing
multiple I{OPT} records will discard all the C{OPT} records.
C{ednsVersion} will be set to L{None}.
@see: U{https://tools.ietf.org/html/rfc6891#section-6.1.1}
r� r� ry N) r rH r� rY rY rS r] Z assertIsNoner' )r r� � ednsMessager r r �(test_ednsMessageDecodeMultipleOptRecords� s
�zEEDNSMessageEDNSEncodingTests.test_ednsMessageDecodeMultipleOptRecordsc C sf t �� }|�t�� � t j�|�}g }dD ]"}t||�t||�kr*|�|� q*|rb| � d|f � dS )z�
L{dns._EDNSMessage._fromMessage} returns an L{_EDNSMessage} instance
whose queries, answers, authority and additional lists are copies (not
references to) the original message lists.
)rT rV r� rY z\Message and _EDNSMessage shared references to the following section lists after decoding: %sN)
r rH rS r? r8 rY rn rE r� Zfail)r �standardMessager� �
duplicatesrL r r r � test_fromMessageCopiesSections� s � ��z;EDNSMessageEDNSEncodingTests.test_fromMessageCopiesSectionsc C s& t jdd�}|�� | �|jg � dS )zf
L{dns._EDNSMessage.toStr} makes no in place changes to the message
instance.
rt rw N)r rY r] r rY )r r� r r r � test_toMessageCopiesSections� s z9EDNSMessageEDNSEncodingTests.test_toMessageCopiesSectionsc C s� t � }tjdd�}|�|� t�� }|�d� |�|� t�� }|g|_g }|� tj
� |�j � |j� tjtj
d�� |� tj
� |�j � |j�dtjtj
d�� |� tj
� |�j � | �dgd |� dS )a
L{dns._EDNSMessage} can decode OPT records, regardless of their position
in the additional records section.
"The OPT RR MAY be placed anywhere within the additional data section."
@see: U{https://tools.ietf.org/html/rfc6891#section-6.1.1}
rt r� r )rL ry N)r r r� r@ rW re r< rH rY r� rY rn r' rZ �insertr )r r� Z optRecordZ optRRHeaderr� ZactualMessagesr r r �test_optHeaderPosition� s,
� ��z3EDNSMessageEDNSEncodingTests.test_optHeaderPositionc C s2 | � � }|�t�� � | �|| j f t�� �� dS )z�
The L(_EDNSMessage} instance created by L{dns._EDNSMessage.fromStr}
derives its edns specific values (C{ednsVersion}, etc) from the supplied
OPT record.
N)r rS r@ r8 r r r� r r r �test_ednsDecode� s z,EDNSMessageEDNSEncodingTests.test_ednsDecodec C s$ | � | jf t�� ��� t�� � dS )z�
The L(_EDNSMessage} instance created by L{dns._EDNSMessage.toStr}
encodes its edns specific values (C{ednsVersion}, etc) into an OPT
record added to the additional section.
N)r r r@ r r] r8 r r r r �test_ednsEncode� s �z,EDNSMessageEDNSEncodingTests.test_ednsEncodec C s$ | � | jf t�� ��� t�� � dS )z�
The L(_EDNSMessage.toStr} encodes the extended I{RCODE} (>=16) by
assigning the lower 4bits to the message RCODE field and the upper 4bits
to the OPT pseudo record.
N)r r rB r r] r8 r r r r �test_extendedRcodeEncode� s �z5EDNSMessageEDNSEncodingTests.test_extendedRcodeEncodec C s2 | � � }|�t�� � | �|| j f t�� �� dS )z�
The L(_EDNSMessage} instance created by L{dns._EDNSMessage.fromStr}
derives RCODE from the supplied OPT record.
N)r rS rB r8 r r r� r r r �test_extendedRcodeDecode s �z5EDNSMessageEDNSEncodingTests.test_extendedRcodeDecodec C s4 | j ddd�}|�� }| �d|j|jd jf� dS )z�
Note that EXTENDED-RCODE value 0 indicates that an unextended RCODE is
in use (values 0 through 15).
https://tools.ietf.org/html/rfc6891#section-6.1.3
r� r )r� r' )r� r N)r rq r r� rY r� )r r� r� r r r �test_extendedRcodeZero s �z3EDNSMessageEDNSEncodingTests.test_extendedRcodeZeroN)r r r r r rY r r� r� r� r� r� r� r� r� r� r� r r r r r� r s
$ r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�ResponseFromMessageTestsz0
Tests for L{dns._responseFromMessage}.
c C s( t �� }t jt j|d�}| �||� dS )z�
L{dns.Message._responseFromMessage} is a constructor function which
generates a new I{answer} message from an existing L{dns.Message} like
instance.
��responseConstructorr N)r rH �_responseFromMessageZ assertIsNot�r �requestZresponser r r �$test_responseFromMessageResponseType! s
�z=ResponseFromMessageTests.test_responseFromMessageResponseTypec C s2 G dd� dt �}tj}| �tj||� d�|� dS )zN
L{dns._responseFromMessage} returns a new instance of C{cls}
c @ s e Zd ZdZg ZdS )zAResponseFromMessageTests.test_responseType.<locals>.SuppliedClassrt N)r r r r� rT r r r r �
SuppliedClass1 s r� r� N)r r rH r6 r� )r r� Z
expectedClassr r r �test_responseType- s ��z*ResponseFromMessageTests.test_responseTypec C s& | � dtjtjtjdd�d�j� dS )zi
L{dns._responseFromMessage} copies the C{id} attribute of the original
message.
i� r
r� N)r r r� rH r� r r r r �test_responseId> s
��z(ResponseFromMessageTests.test_responseIdc C s0 t �� }t jt j|d�}| �d|j|jf� dS )zP
L{dns._responseFromMessage} sets the C{answer} flag to L{True}
r� )FTN)r rH r� r r� r� r r r �test_responseAnswerJ s �
�z,ResponseFromMessageTests.test_responseAnswerc C sD t �� }t� t� t� g}|dd� |_| �|t jt j|d�j� dS )zn
L{dns._responseFromMessage} copies the C{queries} attribute of the
original message.
Nr� )r rH r rT r r� )r r� ZexpectedQueriesr r r �test_responseQueriesW s ��z-ResponseFromMessageTests.test_responseQueriesc C s$ | � dtjtjt�� dd�j� dS )z�
L{dns._responseFromMessage} accepts other C{kwargs} which are assigned
to the new message before it is returned.
r2 )r� r r� N)r r r� rH r� r r r r �test_responseKwargsg s ��z,ResponseFromMessageTests.test_responseKwargsN)
r r r r r� r� r� r� r� r� r r r r r� s
r� c @ s" e Zd ZdZd dd �Zd
d � ZdS )
�Fooz�
An example class for use in L{dns._compactRepr} tests.
It follows the pattern of initialiser settable flags, fields and sections
found in L{dns.Message} and L{dns._EDNSMessage}.
rt r� �ASTFNc C s4 || _ || _|| _|| _|| _|dkr*g }|| _dS )zK
Set some flags, fields and sections as public attributes.
N)�field1�field2�alwaysShowField�flagTrue� flagFalse�section1)r r� r� r� r� r� r� r r r r0 { s z Foo.__init__c C s$ t j| d�� d�� d�� d�� d�S )zO
Call L{dns._compactRepr} to generate a string representation.
r� z field1 field2 alwaysShowFieldzflagTrue flagFalsezsection1 section2)Z
alwaysShowZ
fieldNamesZ flagNamesZ sectionNames)r Z _compactRepr�splitr r r r �__repr__� s �z Foo.__repr__)rt r� r� TFN)r r r r r0 r� r r r r r� u s �
r� c @ s4 e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d
S ) �CompactReprTestsz(
Tests for L[dns._compactRepr}.
c C s | � dt| �� �� dS )z�
L{dns._compactRepr} omits field values and sections which have the
default value. Flags which are True are always shown.
z)<Foo alwaysShowField='AS' flags=flagTrue>Nrb r r r r �
test_defaults� s
�z CompactReprTests.test_defaultsc C s" | j ddd�}| �dt|�� dS )zV
L{dns._compactRepr} displays flags if they have a non-default value.
T)r� r� z3<Foo alwaysShowField='AS' flags=flagTrue,flagFalse>Nrc r� r r r �test_flagsIfSet� s
�z CompactReprTests.test_flagsIfSetc C s" | j ddd�}| �dt|�� dS )zg
L{dns._compactRepr} displays field values if they differ from their
defaults.
r% ra )r� r� z=<Foo field1=10 field2=20 alwaysShowField='AS' flags=flagTrue>Nrc r� r r r �test_nonDefautFields� s
�z%CompactReprTests.test_nonDefautFieldsc C s4 | � � }dddg|_dddg|_| �dt|�� dS )zY
L{dns._compactRepr} displays sections which differ from their defaults.
rt r� zO<Foo alwaysShowField='AS' flags=flagTrue section1=[1, 1, 1] section2=[2, 2, 2]>N)r r� Zsection2r r� r� r r r �test_nonDefaultSections� s �z(CompactReprTests.test_nonDefaultSectionsN)
r r r r r� r r� r� r� r� r r r r r� � s r� )N)gr Z
__future__r r �ior rj Zzope.interface.verifyr Ztwisted.python.failurer Ztwisted.python.utilr r Ztwisted.internetr r
Ztwisted.internet.errorr r Z
twisted.trialr
Z
twisted.namesr Z twisted.testr Ztwisted.test.testutilsr ro rq rs ru rw ry r{ r} r r\ r� r� r� r� r� r� r� r� r� r� r� r� r� rX r~ ZTestCaser r r1 rm r� r� ZSynchronousTestCaser� r r r. r4 r` rf rm r� r� r� r� r� r� r� r� r � Exceptionr r# r+ r, r0 r1 r? r@ rB rC rM rN rS rW rZ r[ rv rx r� r� r� r� r� r� r r r r �<module> s� �
E I �
� � ceY ?(:k8 U/-33r= = �
.#q � � � d �
�+ ,X%