File: //lib/python3/dist-packages/twisted/names/test/__pycache__/test_dns.cpython-310.pyc
o
�b�s � @ s d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z m
Z
ddl m Z m
Z
ddlmZ dd lmZ dd
lmZmZ dd lmZ dd lmZ dd
lmZ ejej ej ej ej ej ej!ej"ej#ej$ej%ej&ej'ej(ej)ej*ej+ej,ej-ej.ej/ej0ej1ej2ej3ej4gZ5G dd� dej6�Z7G dd� dej8�Z9G dd� dej8�Z:G dd� dej8�Z;G dd� dej8�Z<dZ=dZ>G dd� dej6�Z?G d d � d eej6�Z@G d d � d �ZAG d d!� d!ej8�ZBG d"d#� d#eA�ZCG d$d%� d%ej8�ZDG d&d'� d'ej8�ZEG d(d)� d)eej8�ZFG d*d+� d+ej8�ZGG d,d-� d-ej6�ZHd.d/� ZId0d1� ZJG d2d3� d3ej6�ZKG d4d5� d5�ZLG d6d7� d7eej8�ZMG d8d9� d9eej8�ZNG d:d;� d;eO�ZPG d<d=� d=�ZQG d>d?� d?�ZRG d@dA� dA�ZSG dBdC� dC�ZTG dDdE� dE�ZUG dFdG� dG�ZVG dHdI� dI�ZWG dJdK� dK�ZXG dLdM� dMee�ZY djdNdO�ZZG dPdQ� dQ�Z[G dRdS� dS�Z\G dTdU� dUe[e\ej6�Z]G dVdW� dWe[e\ej6�Z^G dXdY� dYe[ej6�Z_G dZd[� d[eej6�Z`G d\d]� d]�ZaG d^d_� d_eaej6�ZbG d`da� daeaej6�ZcG dbdc� dcej6�ZdG ddde� deej6�ZeG dfdg� dg�ZfG dhdi� diej6�ZgdS )kz
Tests for twisted.names.dns.
� N)�BytesIO)�cast)� verifyClass)�address�task)�CannotListenError�ConnectionDone)�dns)�Failure)� FancyEqMixin�
FancyStrMixin)�
proto_helpers)�ComparisonTestsMixin)�unittestc @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �DomainStringTestsc C s | � dt�d�� dS )zA
L{dns.domainString} returns L{bytes} unchanged.
s twistedmatrix.comN�� assertEqualr � domainString��self� r �=/usr/lib/python3/dist-packages/twisted/names/test/test_dns.py�
test_bytes8 s �z DomainStringTests.test_bytesc C � | � dt�d�� dS )z`
L{dns.domainString} converts a native string to L{bytes}
if necessary.
� example.com� example.comNr r r r r � test_nativeA � z DomainStringTests.test_nativec C r )zS
L{dns.domainString} always converts a unicode string to L{bytes}.
s foo.examplez foo.exampleNr r r r r � test_textH � zDomainStringTests.test_textc C r )�A
L{dns.domainString} encodes Unicode using IDNA.
s xn--fwg.testu ‽.testNr r r r r � test_idnaN r zDomainStringTests.test_idnac C s* | � ttjd� | � ttjt�d�� dS )r i(# z bar.exampleN)� assertRaises� TypeErrorr r �Namer r r r �
test_nonsenseT s z DomainStringTests.test_nonsenseN)�__name__�
__module__� __qualname__r r r r! r% r r r r r 7 s r c @ � e Zd ZdZdd� ZdS )�
Ord2ByteTestsz&
Tests for L{dns._ord2bytes}.
c C r )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)r r �
_ord2bytesr r r r �
test_ord2bytea r zOrd2ByteTests.test_ord2byteN)r& r' r( �__doc__r. r r r r r* \ � 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_nonStringn s z Str2TimeTests.test_nonStringc C r )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.
�
�10SN�r r r4 r r r r � test_secondsu � zStr2TimeTests.test_secondsc C r )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 �2MNr9 r r r r � test_minutes} r zStr2TimeTests.test_minutesc C r )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* �3HNr9 r r r r �
test_hours� r zStr2TimeTests.test_hoursc C r )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 �4DNr9 r r r r � test_days� r zStr2TimeTests.test_daysc C r )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�$. �5WNr9 r r r r �
test_weeks� r zStr2TimeTests.test_weeksc C r )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 �6YNr9 r r r r �
test_years� r zStr2TimeTests.test_yearsc C s | � ttjd� dS )zY
If a non-integer prefix is given, L{dns.str2time} raises L{ValueError}.
�fooSN)r" �
ValueErrorr r4 r r r r �test_invalidPrefix� � z Str2TimeTests.test_invalidPrefixN) r& r' r( r/ r6 r: r> r@ rB rD rF rI r r r r r1 i s r1 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" r# r r$ r2 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 r$ �assertIsInstance�name�bytesr )r rO 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 r N)r r$ �decoder r rO )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 r$ r �encoder �getvalue�r rO �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.
rV r � s some prefix to change .tell()s foo�N)r rV )
r r$ r �write�len�Message�
headerSizerW r rX )r rO � compression�previousrZ �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�rf rg ri �payloads baz.ban�1.2.3.4N)r r_ �fromStrr �queries�Query�answers�RRHeader�
UnknownRecord�
additional�A�IN�Record_A�toStr)r �wire�msg�encr r r � test_unknown� sB �
�� ��� ���
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 r$ rS r rO �tell)r rZ rO r r r �test_decodeWithCompression5 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 r$ r r" rH rS rY r r r �test_rejectCompressionLoopT 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.
re s fOO.barN)r r$ r )r �name1�name2�name3r 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.
re s bar.fooN)r r$ �assertNotEqual)r r� r� r r r �test_inequalityj s
zNameTests.test_inequalityN)r& r' r( r/ rM rQ rU r[ rd r{ r� r� r� r� r r r r rK � s
J
rK c @ s� e Zd ZdZg d�Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d5S )6�RoundtripDNSTestsz5
Encoding and then decoding various objects.
)� example.orgs go-away.fish.tvs 23strikesback.netc C sR | j D ]#}t� }t�|��|� |�dd� t�� }|�|� | �|j|� qd S )Nr ) �namesr r r$ rW r� rS r rO �r rT �f�resultr r r � test_name{ s
�zRoundtripDNSTests.test_namec C s� | j D ]D}tdd�D ]<}tdd�D ]4}t� }t�|||��|� |�dd� t�� }|�|� | �|j j |� | �|j
|� | �|j |� qq
qdS )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) r� �ranger r ro rW r� rS r rO rf rg )r rT �dnstype�dnsclsr� r� r r r �
test_query� 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� � r� r N) r r rq rW r� rS r rO r$ rf rg ri )r r� r� r r r �test_resourceRecordHeader� s
z+RoundtripDNSTests.test_resourceRecordHeaderc C sF | � td�� tjtjt�� d� W d � dS 1 s w Y dS )zy
L{RRHeader()} raises L{ValueError} when the given type and the type
of the payload don't match.
z#Payload type \(AAAA\) .* type \(A\)�rf rk N)�assertRaisesRegexrH r rq rt � Record_AAAAr r r r �%test_resourceRecordHeaderTypeMismatch� s "�z7RoundtripDNSTests.test_resourceRecordHeaderTypeMismatchc C sZ d}|D ]&}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 � SimpleRecordrW r� rS r rO r$ )r r� �sr� r� r r r �test_resources� s
�z RoundtripDNSTests.test_resourcesc C sL t D ]!}|� |� }}t|�}t|�}| �|||� d|� d|� d�� qdS )z=
Instances of all record types are hashable.
z != z (for �)N)� RECORD_TYPES�hashr )r �k�k1�k2�hk1�hk2r r r �
test_hashable� s "�z RoundtripDNSTests.test_hashablec C sR | j D ]#}t� }t�|��|� |�dd� t�� }|�|� | �|j|� qdS )z8
Test L{dns.Charstr} encode and decode.
r N) r� r r �CharstrrW r� rS r �stringr� 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}.
@type record: L{dns.IEncodable}
@param record: A record instance to encode
r N)r rW r� r� � __class__rS r )r �recordrZ �length�replicar r r �_recordRoundtripTest� s
z&RoundtripDNSTests._recordRoundtripTestc C s$ t � }|�|� | �|�� |� dS )a3
Assert that encoding C{record} produces the expected bytes.
@type record: L{dns.IEncodable}
@param record: A record instance to encode
@type expectedEncoding: L{bytes}
@param expectedEncoding: The value which C{record.encode()}
should produce.
N)r rW r rX )r �expectedEncodingr� rZ r r r �assertEncodedFormat� s
z%RoundtripDNSTests.assertEncodedFormatc 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 � | � 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.
rl N)r� r rv r r r r �test_A � zRoundtripDNSTests.test_Ac C r� )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# r� zRoundtripDNSTests.test_NULLc C � | � 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.
rl r� s xyzN)r� r �
Record_WKSr r r r �test_WKS+ � zRoundtripDNSTests.test_WKSc C r� )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 r� r r r r � test_AAAA3 r� zRoundtripDNSTests.test_AAAAc C r� )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; r� 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.
r� � r� r )�priority�weight�port�targetN)r� r �
Record_SRVr r r r �test_SRVC � �zRoundtripDNSTests.test_SRVc C s: d}t jt jjt jj|d�}| �|� | �d| |� dS )z�
The byte stream written by L{dns.Record_SSHFP.encode} can be used by
L{dns.Record_SSHFP.decode} to reconstruct the state of the original
L{dns.Record_SSHFP} instance.
s �9��^kK
2U��`��� )� algorithm�fingerprintType� fingerprints N)r � Record_SSHFP�
ALGORITHM_DSS�FINGERPRINT_TYPE_SHA1r� r� )r �fp�rrr r r �
test_SSHFPM s ��
z RoundtripDNSTests.test_SSHFPc C s� ddg}|D ]c\}}}}}}t �||||||�}t� } |�| � | �dd� t �� }
|
�| � | �|j|
j� | �|j|
j� | �|j |
j � | �|j
|
j
� | �|j |
j � | �|j j
|
j j
� | �|j|
j� qdS )z=
Test L{dns.Record_NAPTR} encode and decode.
)�d r7 � u� sip+E2Us! !^.*$!sip:information@domain.tld!� )r� �2 � ss http+I2L+I2C+I2Rr� s _http._tcp.gatech.edur N)r � Record_NAPTRr rW r� rS r �order�
preference�flags�service�regexp� replacementrO ri ) r �naptrsr� r� r� r� r� r� �rin�e�routr r r �
test_NAPTR` s( � �
�z RoundtripDNSTests.test_NAPTRc C � | � 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.
r� r )�subtype�hostnameN)r� r � Record_AFSDBr r r r �
test_AFSDBz r� z RoundtripDNSTests.test_AFSDBc C r� )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 )�mbox�txtN)r� r � Record_RPr r r r �test_RP� s �zRoundtripDNSTests.test_RPc C r� )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)�cpu�osN)r� r � Record_HINFOr r r r �
test_HINFO� r� z RoundtripDNSTests.test_HINFOc C r� )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� )�rmailbx�emailbxN)r� r � Record_MINFOr r r r �
test_MINFO� r� z RoundtripDNSTests.test_MINFOc C r� )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.
r� r )r� rO N)r� r � Record_MXr r r r �test_MX� r� zRoundtripDNSTests.test_MXc C sx d}t jdddd|d�}| �|� d| d }| �||� t jd d
d t jd |d
�}| �|� d| d }| �||� dS )z�
The byte stream written by L{dns.Record_TSIG.encode} can be used by
L{dns.Record_TSIG.decode} to reconstruct the state of the original
L{dns.Record_TSIG} instance.
s !"#0123zhmac-md5.sig-alg.reg.int�/qUZ�* r� )r� �
timeSigned�
originalID�fudge�MACs$ hmac-md5sig-algregint ZUq/ s * z hmac-sha256l '� � s � )r� r r �error� otherDatar s hmac-sha256 �' s �� � N)r � Record_TSIGr� r� �EBADTIME)r �macr� �rdatar r r � test_TSIG� s@ �
��� �
���zRoundtripDNSTests.test_TSIGc 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_TXT� r; zRoundtripDNSTests.test_TXTN) r& r' r( r/ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r� t s8
)r� 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 � | � t�� jd� dS )zC
L{dns.Message.authenticData} has default value 0.
r N�r r r_ �
authenticDatar r r r �test_authenticDataDefault� r z&MessageTests.test_authenticDataDefaultc C � | � tjdd�jd� dS )z�
L{dns.Message.__init__} accepts a C{authenticData} argument which
is assigned to L{dns.Message.authenticData}.
r� �r Nr r r r r �test_authenticDataOverride� � z'MessageTests.test_authenticDataOverridec C � | � tjdd��� t� dS )zr
L{dns.Message.toStr} encodes L{dns.Message.authenticData} into
byte4 of the byte string.
r� r N)r r r_ rw � MESSAGE_AUTHENTIC_DATA_BYTESr r r r �test_authenticDataEncode� � �z%MessageTests.test_authenticDataEncodec C �$ t �� }|�t� | �|jd� dS )zp
L{dns.Message.fromStr} decodes byte4 and assigns bit3 to
L{dns.Message.authenticData}.
r� N)r r_ rm r# r r �r �mr r r �test_authenticDataDecode �
z%MessageTests.test_authenticDataDecodec C r )zF
L{dns.Message.checkingDisabled} has default value 0.
r N�r r r_ �checkingDisabledr r r r � test_checkingDisabledDefault r z)MessageTests.test_checkingDisabledDefaultc C r )z�
L{dns.Message.__init__} accepts a C{checkingDisabled} argument which
is assigned to L{dns.Message.checkingDisabled}.
r� �r, Nr+ r r r r � test_checkingDisabledOverride r! z*MessageTests.test_checkingDisabledOverridec C r" )zu
L{dns.Message.toStr} encodes L{dns.Message.checkingDisabled} into
byte4 of the byte string.
r� r. N)r r r_ rw � MESSAGE_CHECKING_DISABLED_BYTESr r r r �test_checkingDisabledEncode r% z(MessageTests.test_checkingDisabledEncodec C r& )zs
L{dns.Message.fromStr} decodes byte4 and assigns bit4 to
L{dns.Message.checkingDisabled}.
r� N)r r_ rm r0 r r, r' r r r �test_checkingDisabledDecode' r* 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 r_ r r r r �test_reprDefaults1 � 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 r_ r r3 r' r r r �test_reprFlagsIfSet8 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.
r7 r| � �( )�id�opCode�rCode�maxSizez-<Message id=10 opCode=20 rCode=30 maxSize=40>Nr; r' r r r �test_reprNonDefautFieldsN s
�z%MessageTests.test_reprNonDefautFieldsc C sD t �� }g d�|_g d�|_g d�|_g d�|_| �dt|�� dS )�e
L{dns.Message.__repr__} displays sections which differ from their
defaults.
�r� r� r� �r� r� � �� r� � �r7 � r� z^<Message id=0 queries=[1, 2, 3] answers=[4, 5, 6] authority=[7, 8, 9] additional=[10, 11, 12]>N)r r_ rn rp � authorityrs r r3 r' r r r �test_reprNonDefaultSectionsY �
�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 r_ r" �EOFErrorrm �r ry r r r �test_emptyMessagen 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 r_ rm r r? � assertFalser6 r@ �OP_QUERYr7 r8 rn rp rM rs rQ r r r �test_emptyQueryv 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 � g | ]}t �|��qS r �r r- ��.0�ir r r �
<listcomp>� � z*MessageTests.test_NULL.<locals>.<listcomp>rS � testname�rk r N)�joinr� r r� rq �NULLr_ rp �appendr rW r� rS rN rk r �r rP �recr� �msg1r� �msg2r 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)r3 r r_ �lookupRecordTyperr 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.
rl r rh r_ s N) r r rq rv rW r_ rm rX r rp rT r7 �r �bufr6 �messager r r � test_nonAuthoritativeMessage� 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.
rl r rh r_ s TN) r r rq rv rW r_ rm rX r7 r rp �
assertTrueri r r r �test_authoritativeMessage� s
�� z&MessageTests.test_authoritativeMessageN)r& r' r( r/ r r r$ r) r- r/ r1 r2 r4 r< rC rN rR rV r� rh rl rn 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"S )#�MessageComparisonTestszD
Tests for the rich comparison of L{dns.Message} instances.
c O sl |� dg �}|� dg �}|� dg �}|� dg �}tjdi |��}|r%||_|r*||_|r/||_|r4||_|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.
rn rp rM rs Nr )�popr r_ rn rp rM rs )r �args�kwargsrn rp rM rs r( r r r �messageFactory� s z%MessageComparisonTests.messageFactoryc C �* | � | jdd�| jdd�| jdd�� dS )zd
Two L{dns.Message} instances compare equal if they have the same id
value.
r7 �r? r| N��"assertNormalEqualityImplementationrs r r r r �test_id �
�z MessageComparisonTests.test_idc C rt )zg
Two L{dns.Message} instances compare equal if they have the same answer
flag.
r� �r6 r Nrv r r r r � test_answer ry z"MessageComparisonTests.test_answerc C rt )zh
Two L{dns.Message} instances compare equal if they have the same opCode
value.
r7 �r@ r| Nrv r r r r � test_opCode ry z"MessageComparisonTests.test_opCodec C rt )zg
Two L{dns.Message} instances compare equal if they have the same recDes
flag.
r� �r9 r Nrv r r r r � test_recDes! ry z"MessageComparisonTests.test_recDesc C rt )zf
Two L{dns.Message} instances compare equal if they have the same recAv
flag.
r� �r: r Nrv r r r r �
test_recAv, ry z!MessageComparisonTests.test_recAvc C rt )ze
Two L{dns.Message} instances compare equal if they have the same auth
flag.
r� �r7 r Nrv r r r r � test_auth7 ry z MessageComparisonTests.test_authc C rt )zg
Two L{dns.Message} instances compare equal if they have the same rCode
value.
r7 �rA r| Nrv r r r r �
test_rCodeB ry z!MessageComparisonTests.test_rCodec C rt )zf
Two L{dns.Message} instances compare equal if they have the same trunc
flag.
r� �r8 r Nrv r r r r �
test_truncM ry z!MessageComparisonTests.test_truncc C rt )zi
Two L{dns.Message} instances compare equal if they have the same
maxSize value.
r7 �rB r| Nrv r r r r � test_maxSizeX ry z#MessageComparisonTests.test_maxSizec C rt )zn
Two L{dns.Message} instances compare equal if they have the same
authenticData flag.
r� r r Nrv r r r r �test_authenticDatac ry z)MessageComparisonTests.test_authenticDatac C rt )zq
Two L{dns.Message} instances compare equal if they have the same
checkingDisabled flag.
r� r. r Nrv r r r r �test_checkingDisabledn ry z,MessageComparisonTests.test_checkingDisabledc C �B | � | 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 �rn r� N�rw rs r ro r r r r � test_queriesy �
�z#MessageComparisonTests.test_queriesc
C �` | � | 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 rl r_ �rp r� �4.3.2.1N�rw rs r rq rv r r r r � test_answers� � ����z#MessageComparisonTests.test_answersc
C �f | � | 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 r� �rM r� N�rw rs r rq �SOAr� r r r r �test_authority� � �������z%MessageComparisonTests.test_authorityc
C �` | � | 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 rl r_ �rs r� Nr� r r r r �test_additional� � �������z&MessageComparisonTests.test_additionalN)r& r' r( r/ rs rx r{ r} r r� r� r� r� r� r� r� r� r� r� r� r r r r ro � s$ ro 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__� s
zTestController.__init__Nc C s | j �|||f� dS )zN
Save the message so that it can be checked during the tests.
N)r� rb )r ry �proto�addrr r r �messageReceived� r z TestController.messageReceived�N)r& r' r( r/ r� r� r r r r r� � s r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�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 �DNSDatagramProtocolr� r
�FakeDatagramTransport�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� �UDP� 127.0.0.1i90 N)r� �datagramReceivedr � IPv4Addressr r� 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� r� rl �r r_ c � � � | jd j�� d� d S �Nr rl �r rp rk �
dottedQuad�r� r r r �cb� � z2DatagramProtocolTests.test_simpleQuery.<locals>.cb)r� �queryr ro r r^ � liveMessages�keysr_ �next�iterr? rq rv rp � addCallbackr� rw )r �dr( r� 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� r� r7 r � r� r� r ro r r^ r� r� �advance�
assertFailure�DNSQueryTimeoutError�r r� 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 � t d���N�bar�� RuntimeError)rk r� r r r �
writeError � z9DatagramProtocolTests.test_writeError.<locals>.writeErrorr� r� �r� r� r] r� r ro r� r� �r r� r� 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 ��r� )r r r r r �startListeningError# � zCDatagramProtocolTests.test_listenError.<locals>.startListeningErrorNr� r� )r� �startListeningr� r� r ro r� r )r r� r� 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}.
r� rl r� r_ r� ���r N) r r_ r? rq rv rp r� r� rw r r� r� )r rk r r r �$test_receiveMessageNotInLiveMessages- s
$z:DatagramProtocolTests.test_receiveMessageNotInLiveMessagesN) r& r' r( r/ r� r� r� r� r� r� r� r r r r r� � s r� c @ �( 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� r� � connectionsr r r r r� C s
zTestTCPController.__init__c C � | j �|� d S r� )r� rb �r r� r r r �connectionMadeG � z TestTCPController.connectionMadec C r� r� )r� �remover� r r r �connectionLostJ r� z TestTCPController.connectionLostN)r& r' r( r/ r� r� r� r r r r r� : s
r� c @ �@ 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 r� r� r� r� r � DNSProtocolr� r� r
�StringTransportr� r r r r r� S 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 r� r� r� r� 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 )r� r� r� �< r r� r� r r r r� g 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� r� rl r� r_ c r� r� r� r� r r r r� | r� z-DNSProtocolTests.test_simpleQuery.<locals>.cb�!H)r� r� r ro r r^ r� r� r_ r� r� r? rq rv rp r� rw �struct�pack� dataReceived)r r� r( r� r� r r r r� r 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 r� r� r� )rk r r r r� � r� z4DNSProtocolTests.test_writeError.<locals>.writeErrorr� r� r� r r r r� � 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}.
r� rl r� r_ r� r� r N)r r_ r? rq rv rp rw r� r� r^ r� r� r r� r� )r rk r� r r r r� � s $z5DNSProtocolTests.test_receiveMessageNotInLiveMessagesN)
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"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 � | � 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 r3 r � Record_NSr r r r �test_ns� r% zReprTests.test_nsc C r� )z�
The repr of a L{dns.Record_MD} instance includes the name of the
mail destination and the TTL of the record.
r r� z <MD name=example.com ttl=4321>N)r r3 r � Record_MDr r r r �test_md� r% zReprTests.test_mdc C r� )z�
The repr of a L{dns.Record_MF} instance includes the name of the
mail forwarder and the TTL of the record.
r r� z <MF name=example.com ttl=4321>N)r r3 r � Record_MFr r r r �test_mf� r% zReprTests.test_mfc C r� )z�
The repr of a L{dns.Record_CNAME} instance includes the name of the
mail forwarder and the TTL of the record.
r r� z!<CNAME name=example.com ttl=4321>N)r r3 r � Record_CNAMEr r r r �
test_cname� � �zReprTests.test_cnamec C r� )z}
The repr of a L{dns.Record_MB} instance includes the name of the
mailbox and the TTL of the record.
r r� z <MB name=example.com ttl=4321>N)r r3 r � Record_MBr r r r �test_mb� r% zReprTests.test_mbc C r� )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 r� z <MG name=example.com ttl=4321>N)r r3 r � Record_MGr r r r �test_mg� r% zReprTests.test_mgc C r� )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 r� z <MR name=example.com ttl=4321>N)r r3 r � Record_MRr r r r �test_mr� r% zReprTests.test_mrc C r� )z~
The repr of a L{dns.Record_PTR} instance includes the name of the
pointer and the TTL of the record.
r r� z <PTR name=example.com ttl=4321>N)r r3 r �
Record_PTRr r r r �test_ptr� r� zReprTests.test_ptrc C r� )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 r� z!<DNAME name=example.com ttl=4321>N)r r3 r � Record_DNAMEr r r r �
test_dname� r� zReprTests.test_dnamec C r� )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.
rl i7 z<A address=1.2.3.4 ttl=567>N)r r3 r rv r r r r �test_a� r� 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 rNamerL �� � r7 rL r� )r� r� r� r� r� r� r� ri zZ<SOA mname=mName rname=rName serial=123 refresh=456 retry=789 expire=10 minimum=11 ttl=12>N)r r3 r r� r r r r �test_soa s �� �zReprTests.test_soac C r� )z
The repr of a L{dns.Record_NULL} instance includes the repr of its
payload and the TTL of the record.
s abcdrL z <NULL payload='abcd' ttl=123>N)r r3 r r� r r r r � test_null r% zReprTests.test_nullc C � | � 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.5rI r� rh z&<WKS address=2.3.4.5 protocol=7 ttl=8>N)r r3 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::1234r7 rh z <AAAA address=8765::1234 ttl=10>N)r r3 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::5678re r7 rh z,<A6 suffix=1234::5678 prefix=foo.bar ttl=10>N)r r3 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.
r� r� r� r� r� z9<SRV priority=1 weight=2 target=example.org port=3 ttl=4>N)r r3 r r� r r r r �test_srvE 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.
r� rJ � S� https
/foo/bar/i� bazr� zY<NAPTR order=5 preference=9 flags=S service=http regexp=/foo/bar/i replacement=baz ttl=3>N)r r� r r3 �r r� r r r �
test_naptrO s
�zReprTests.test_naptrc C � | � tt�ddd��d� dS )z{
The repr of a L{dns.Record_AFSDB} instance includes the subtype,
hostname, and TTL of the record.
r� r� r� z,<AFSDB subtype=3 hostname=example.org ttl=5>N)r r3 r r� r r r r �
test_afsdb\ � �zReprTests.test_afsdbc C r )zw
The repr of a L{dns.Record_RP} instance includes the mbox, txt, and TTL
fields of the record.
r� s admin.example.comr� z7<RP mbox=alice.example.com txt=admin.example.com ttl=3>N)r r3 r r� r r r r �test_rpf r zReprTests.test_rpc C r )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 r3 r r r r r r �
test_hinfop r 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 r3 r r r r �
test_minfoz s
�zReprTests.test_minfoc C r )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 r3 r r r r r r �test_mx� r zReprTests.test_mxc C r )zr
The repr of a L{dns.Record_TXT} instance includes the data and ttl
fields of the record.
r� r� r rh z <TXT data=['foo', 'bar'] ttl=15>N)r r3 r r r r r r �test_txt� � �zReprTests.test_txtc C r )zr
The repr of a L{dns.Record_SPF} instance includes the data and ttl
fields of the record.
r� r� r rh z <SPF data=['foo', 'bar'] ttl=15>N)r r3 r �
Record_SPFr r r r �test_spf� r# zReprTests.test_spfc C r� )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 r3 r rr r r r r r{ � r� zReprTests.test_unknownN) r& r' r( r/ r� r� r� r� r r r r r r r r
r r r r r r r r r r! r" r% r{ r r r r r� � s6
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@dA� Z#dBS )C�
EqualityTestszM
Tests for the equality and non-equality behavior of record classes.
c C s | � |||�S r� )rw )r �
firstValueOne�secondValueOne�valueTwor r r �
_equalityTest� s �zEqualityTests._equalityTestc C �$ | � 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 r+ )zo
Two L{dns.Name} instances compare equal if and only if they have the
same name value.
r, r- N)r* r r$ r r r r r� � 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 rL iA r� N)r* )r rg r r r �_simpleEqualityTest� 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 rl r_ r� �1.2.3.5)rg rk r� )ri rk r< r� )r7 rk r N)r* r rq rv rt �MXru �CSr r r r �
test_rrheader� s< �� ����zEqualityTests.test_rrheaderc C � | � 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 r� r r r r r�
� zEqualityTests.test_nsc C r4 �zv
Two L{dns.Record_MD} instances compare equal if and only if they have
the same name and TTL.
N)r/ r r� r r r r r� r5 zEqualityTests.test_mdc C r4 )zv
Two L{dns.Record_MF} instances compare equal if and only if they have
the same name and TTL.
N)r/ r r� r r r r r� r5 zEqualityTests.test_mfc C r4 )zy
Two L{dns.Record_CNAME} instances compare equal if and only if they
have the same name and TTL.
N)r/ r r� r r r r r� r5 zEqualityTests.test_cnamec C r4 )zv
Two L{dns.Record_MB} instances compare equal if and only if they have
the same name and TTL.
N)r/ r r� r r r r r & r5 zEqualityTests.test_mbc C r4 )zv
Two L{dns.Record_MG} instances compare equal if and only if they have
the same name and TTL.
N)r/ r r r r r r r - r5 zEqualityTests.test_mgc C r4 )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 4 r5 zEqualityTests.test_mrc C r4 )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 ; r5 zEqualityTests.test_ptrc C r4 r6 )r/ r r r r r r r B r5 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.
rl r� rG r0 N)r* r rv r r r r r I �
�
�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 rnamerL r
r r7 r| r= s xnamer� N)r* r r� r r r r r [ sP ��������zEqualityTests.test_soac C �P | � 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 barr7 zbar foor� N)r* r r� r r r r r
� r7 zEqualityTests.test_nullc
C �� | � 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.
rl r� �foor� r� r� r� �� N)r* r r� r r r r r � �( ����zEqualityTests.test_wksc C r8 )z{
Two L{dns.Record_AAAA} instances compare equal if and only if they have
the same address and ttl.
z1::2r� z2::1r7 N)r* r r� r r r r r � r7 zEqualityTests.test_aaaac
C r9 )
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 r7 r} z::abcd:0r� r� N)r* r r� r r r r r � 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.
r7 r| r= r r> r� r; i, r� � N)r* r r� r r r r r � s2 �����zEqualityTests.test_srvc
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�� dS )z�
Two L{dns.Record_SSHFP} instances compare equal if and only if
they have the same key type, fingerprint type, fingerprint, and ttl.
r� r� s happydayr> s happxday�- N)r* r r� r r r r �
test_sshfp r< zEqualityTests.test_sshfpc
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.
r� r� r� r� s /foo/bar/r r� r� � pr s /bar/foo/s quuxr� N)r* r r� r r r r r 7 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.
r� r r� r� r� N)r* r r� r r r r r h � � � �zEqualityTests.test_afsdbc C � | � 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 nicer7 r s alice is not nicer� N)r* r r� r r r r r � rA zEqualityTests.test_rpc C rB )z}
Two L{dns.Record_HINFO} instances compare equal if and only if they
have the same cpu, os, and ttl.
zx86-64�plan9r7 �i386�plan11r� N)r* r r r r r r r � rA zEqualityTests.test_hinfoc C rB )z�
Two L{dns.Record_MINFO} instances compare equal if and only if they
have the same rmailbx, emailbx, and ttl.
s rmailboxs emailboxr7 s someplaces somethingr� N)r* r r r r r r r � rA zEqualityTests.test_minfoc C rB )z�
Two L{dns.Record_MX} instances compare equal if and only if they have
the same preference, name, and ttl.
r7 r� r| r� s example.netr; N)r* r r r r r r r! � rA zEqualityTests.test_mxc
C � | � 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: r� r7 rh �bazr� N)r* r r r r r r r" � � ���zEqualityTests.test_txtc
C rF )zs
L{dns.Record_SPF} instances compare equal if and only if they have the
same data and ttl.
r: r� r7 rh rG r� N)r* r r$ r r r r r% � rH zEqualityTests.test_spfc C s� dddddt jddd�}d d
d d d
t jddd�}|�� D ]%}|�� }|| ||< | �t jdi |��t jdi |��t jdi |��� q dS )z�
L{dns.Record_TSIG} instances compare equal if and only if they have the
same RDATA (algorithm, timestamp, MAC, etc.) and ttl.
z hmac-sha224r r� s �c r� r> )r� r r
r r r r ri z hmac-sha512i�pUZr s i�� � r= Nr )r �OKr r� �copyr* r )r �baseargs�altargs�kw�alteredr r r � test_tsig s6 � �
��zEqualityTests.test_tsigc 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: r7 rh �foobarr� r� N)r* r rr r r r r r{ 2 rA zEqualityTests.test_unknownN)$r& r' r( r/ r* r. r� r/ r3 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% rQ r{ r r r r r& � sD 17 !$ 1 "r& c @ r� ) �
RRHeaderTestsz2
Tests for L{twisted.names.dns.RRHeader}.
c C �& | � ttjdtjtjdt�d�� dS )z�
Attempting to create a L{dns.RRHeader} instance with a negative TTL
causes L{ValueError} to be raised.
r r� r� N�r" rH r rq rt ru rv r r r r �test_negativeTTLP � �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 rq rt ru rv r ri �int)r �
ttlAsFloat�headerr r r �test_nonIntegralTTL_ s
�z!RRHeaderTests.test_nonIntegralTTLc C rT )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� NrU r r r r �!test_nonNumericTTLRaisesTypeErrori rW z/RRHeaderTests.test_nonNumericTTLRaisesTypeErrorN)r& r' r( r/ rV r[ r\ r r r r rS K s
rS c @ r� )�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 �
_nameToLabelsr r r r �
test_empty~ r5 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� r5 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� r! 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.rc r� Nr^ r r r r �test_withTrailingDot� r! z&NameToLabelsTests.test_withTrailingDotc C � | � t�d�g d�� 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 examplerc r� Nr^ r r r r �test_subdomain� s �z NameToLabelsTests.test_subdomainc C rf )z`
L{dns._nameToLabels} preserves the case of ascii
characters in labels.
� EXAMPLE.COM)s EXAMPLEs COMr� Nr^ r r r r �test_casePreservation� r! z'NameToLabelsTests.test_casePreservationN)
r& r' r( r/ r` rb rd re rg ri r r r r r] y s
r] c C �" | � t�||�|�d|��� 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 is not a subdomain of N)rm r �_isSubdomainOf��testCase�
descendant�ancestorr r r �assertIsSubdomainOf� �
�rp c C rj )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 is a subdomain of N)rT r rk rl r r r �assertIsNotSubdomainOf� rq rr 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�rp r r r r �test_identical� r5 z!IsSubdomainOfTests.test_identicalc C � t | dd� dS )z�
L{dns._isSubdomainOf} returns C{True} when the first
name is an immediate descendant of the second name.
rV r Nrt r r r r � test_parent� r5 z IsSubdomainOfTests.test_parentc C rv )z
L{dns._isSubdomainOf} returns C{True} when the first
name is a distant descendant of the second name.
s foo.bar.baz.example.comrc Nrt r r r r �test_distantAncestor� r5 z'IsSubdomainOfTests.test_distantAncestorc C rv )zw
L{dns._isSubdomainOf} returns C{False} when the first
name is an ancestor of the second name.
r rV N�rr r r r r �test_superdomain� r5 z#IsSubdomainOfTests.test_superdomainc C rv )zs
L{dns._isSubdomainOf} returns C{False} if the first name
is a sibling of the second name.
rV s bar.example.comNry r r r r � test_sibling� r5 z IsSubdomainOfTests.test_siblingc C rv )zx
L{dns._isSubdomainOf} returns C{False} even when domain
names happen to share a common suffix.
s foo.myexample.comr Nry r r r r �test_unrelatedCommonSuffix� r5 z-IsSubdomainOfTests.test_unrelatedCommonSuffixc C rv )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 Nrt r r r r � test_subdomainWithTrailingDot
� z0IsSubdomainOfTests.test_subdomainWithTrailingDotc C rv )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 ".".
rV � example.com.Nrt r r r r � test_superdomainWithTrailingDot
r z2IsSubdomainOfTests.test_superdomainWithTrailingDotc C rv )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� Nrt r r r r �test_bothWithTrailingDot
r z+IsSubdomainOfTests.test_bothWithTrailingDotc C rv )zw
L{dns._isSubdomainOf} returns C{False} if the first name
is empty and the second name is not.
r� r Nry r r r r �test_emptySubdomain
r5 z&IsSubdomainOfTests.test_emptySubdomainc C rv )zv
L{dns._isSubdomainOf} returns C{True} if the second name
is empty and the first name is not.
rV r� Nrt r r r r �test_emptySuperdomain
r5 z(IsSubdomainOfTests.test_emptySuperdomainc C s t | dd� t | dd� dS )z`
L{dns._isSubdomainOf} does case-insensitive comparison
of name labels.
rV rh s FOO.EXAMPLE.COMr Nrt r r r r � test_caseInsensitiveComparison'
s z1IsSubdomainOfTests.test_caseInsensitiveComparisonN)r& r' r( r/ ru rw rx rz r{ r| r~ r� r� r� r� r� r r r r rs � s rs 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}|rd}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}.
rJ r� s ) � r )rg � excludeName�excludeOptions�rdlenr r r rP :
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}.
� r� r� T)�udpPayloadSize�
extendedRCODE�version�dnssecOK)r �
_OPTHeader�rg r r r r2 Y
s �z OPTNonStandardAttributes.objectN)FF)r& r' r( r/ � classmethodrP r2 r r r r r� 1
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 � t tjtj� dS )zA
L{dns._OPTHeader} implements L{dns.IEncodable}.
N)r r �
IEncodabler� r r r r �test_interfacek
� 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� rO r$ r r r r r� q
s zOPTHeaderTests.test_namec C s$ t �� }| �tt|dt �d�� dS )z5
L{dns._OPTHeader.name} is readonly.
rO r N)r r� r" �AttributeError�setattrr$ �r �hr r r �test_nameReadonlyx
s z OPTHeaderTests.test_nameReadonlyc C r )z^
L{dns._OPTHeader.type} is an instance attribute with fixed value
41.
�) N)r r r� rf r r r r � test_type
r zOPTHeaderTests.test_typec C s t �� }| �tt|dt j� dS )z5
L{dns._OPTHeader.type} is readonly.
rf N)r r� r" r� r� rt r� r r r �test_typeReadonly�
s z OPTHeaderTests.test_typeReadonlyc C r )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�
r z"OPTHeaderTests.test_udpPayloadSizec C r )z`
L{dns._OPTHeader.udpPayloadSize} can be overridden in the
constructor.
r� �r� Nr� r r r r �test_udpPayloadSizeOverride�
r! z*OPTHeaderTests.test_udpPayloadSizeOverridec C r )z@
L{dns._OPTHeader.extendedRCODE} defaults to 0.
r N�r r r� r� r r r r �test_extendedRCODE�
r z!OPTHeaderTests.test_extendedRCODEc C r )z_
L{dns._OPTHeader.extendedRCODE} can be overridden in the
constructor.
r� �r� Nr� r r r r �test_extendedRCODEOverride�
r! z)OPTHeaderTests.test_extendedRCODEOverridec C r )z:
L{dns._OPTHeader.version} defaults to 0.
r N�r r r� r� r r r r � test_version�
r zOPTHeaderTests.test_versionc C r )zY
L{dns._OPTHeader.version} can be overridden in the
constructor.
r� �r� Nr� r r r r �test_versionOverride�
r! z#OPTHeaderTests.test_versionOverridec C s | � t�� j� dS )z?
L{dns._OPTHeader.dnssecOK} defaults to False.
N)rT r r� r� r r r r �
test_dnssecOK�
rJ z OPTHeaderTests.test_dnssecOKc C s | � tjdd�j� dS )zZ
L{dns._OPTHeader.dnssecOK} can be overridden in the
constructor.
T�r� N)rm r r� r� r r r r �test_dnssecOKOverride�
r5 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�
r zOPTHeaderTests.test_optionsc C s" t jdgd�}| �|jdg� dS )zY
L{dns._OPTHeader.options} can be overridden in the
constructor.
)r� r� � �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� r2 rW r rX rP )r �br r r r[ �
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.
r� � foobarbazr� � quxT�r� � foobarbaz quxN)
r� r2 r �_OPTVariableOptionr� r rW r rX rP )r r� r� r r r �test_encodeWithOptions�
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� rS r r� rP r r2 )r �
decodedHeaderr r r rU �
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� rP r r� rS r r� r^ rX �r r� r� r r r �test_decodeAllExpectedBytes 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� rR r� N) r r� rP r r� rS r rO r$ �r r� r� r r r �test_decodeDiscardsName s �
z&OPTHeaderTests.test_decodeDiscardsNamec C �0 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� rP r r� r" rP rS r� r r r �test_decodeRdlengthTooShort( �
�� z*OPTHeaderTests.test_decodeRdlengthTooShortc C r� )zi
L{dns._OPTHeader.decode} raises an exception if the supplied
RDLEN is too long.
Tr� s Nr� r� r r r �test_decodeRdlengthTooLong9 r� 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� r� r� r� r� N) r r� rP r r� rS r r� r� r� r r r �test_decodeWithOptionsJ 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 r l �} s �� abcrj � � Tr, )�code�data)r� r� r� r� r� N)r rq �OPTrr r� � fromRRHeaderr� r )r �
genericHeader�decodedOptHeader�expectedOptHeaderr r r �test_fromRRHeaderi 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 r3 r r� r r r r � test_repr� s
�zOPTHeaderTests.test_reprc C �* | � tjdd�tjdd�tjdd�� dS )zh
Two L{OPTHeader} instances compare equal if they have the same
udpPayloadSize.
r� r� r� N�rw r r� r r r r �test_equalityUdpPayloadSize� ry z*OPTHeaderTests.test_equalityUdpPayloadSizec C r� )zg
Two L{OPTHeader} instances compare equal if they have the same
extendedRCODE.
r� r� r� Nr� r r r r �test_equalityExtendedRCODE� ry z)OPTHeaderTests.test_equalityExtendedRCODEc C r� )za
Two L{OPTHeader} instances compare equal if they have the same
version.
r� r� r� Nr� r r r r �test_equalityVersion� ry z#OPTHeaderTests.test_equalityVersionc C r� )zh
Two L{OPTHeader} instances compare equal if they have the same
dnssecOK flags.
Tr� FNr� r r r r �test_equalityDnssecOK� ry 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.
r� � xr� r� � yN)rw r r� r� r r r r �test_equalityOptions� s
�z#OPTHeaderTests.test_equalityOptionsN)#r& r' r( r/ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r[ r� rU r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� f
sB
r� c @ r� )�OPTVariableOptionTestsz.
Tests for L{dns._OPTVariableOption}.
c C r� )zI
L{dns._OPTVariableOption} implements L{dns.IEncodable}.
N)r r r� r� r r r r r� � r� 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.
r� r� N)r r� r r� r� r� r r r �test_constructorArguments� s z0OPTVariableOptionTests.test_constructorArgumentsc C r� )zf
L{dns._OPTVariableOption.__repr__} displays the code and data
of the option.
r� r� z"<_OPTVariableOption code=1 data=x>N)r r3 r r� r r r r r� � r� z OPTVariableOptionTests.test_reprc C r8 )zs
Two OPTVariableOption instances compare equal if they have the same
code and data values.
r� r� r� r� N)rw r r� r r r r r� � 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.
r� � foobar�
foobarN)r r� r rW r rX )r �or� r r r r[ � 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� r� r� N)r r r� rS r r� r� )r r� r� r r r rU s �
z"OPTVariableOptionTests.test_decodeN)
r& r' r( r/ r� r� r� r� r[ rU r r r r r� � s
r� c @ r) )�
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�rq rr �r rq rr r r r r� # s
zRaisedArgs.__init__N)r& r' r( r/ r� r r r r r� s r� c @ �( e Zd ZdZedd� �Zedd� �ZdS )� MessageEmptyz^
Generate byte string and constructor arguments for an empty
L{dns._EDNSMessage}.
c C � 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 rP 4 �
�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.
rS Tr N� r? r6 r@ r7 r8 r9 r: rA � ednsVersion)�dictr � OP_STATUSr� r r r rr G s �zMessageEmpty.kwargsN�r& r' r( r/ r� rP rr r r r r r� . s
r� c @ r� )�MessageTruncatedz=
An empty response message whose TR bit is set to 1.
c C r� )r� s � r r� r r r rP a r� zMessageTruncated.bytesc C s t dddddddddd� S )r� rS r� r Nr� )r� r� r r r rr t s �zMessageTruncated.kwargsNr� r r r r r� \ s
r� c @ r� )�MessageNonAuthoritativez.
A minimal non-authoritative message.
c C r� )r� s r r� r r r rP � r� z MessageNonAuthoritative.bytesc
C s( t dddtjdtjddd�dd�gd �S )
r� rS r Nr� rl rh F�rk r7 �r? r7 r� rp �r� r rq rv r� r r r rr � � ��z MessageNonAuthoritative.kwargsNr� r r r r r� � �
r� c @ r� )�MessageAuthoritativez*
A minimal authoritative message.
c C r� )r� s r r� r r r rP � r� zMessageAuthoritative.bytesc
C s( t dddtjdtjddd�dd �gd
�S ) r� rS r� Nr� rl r rh Tr� r� r� r� r r r rr � r� zMessageAuthoritative.kwargsNr� r r r r r� � r� r� c @ r� )�MessageCompletez�
An example of a fully populated non-edns response message.
Contains name compression, answers, authority, and additional records.
c C r� )r� sn �� examplecom � ���� 'ns1�
hostmaster� ������������������ ���� �)�) ���� r r� r r r rP � r� zMessageComplete.bytesc C s� t ddtjdddddt�dtj�gtjdtjddtjddd d
d d d
dd�d�gtjdtjddtjddd�d�gtjdtj ddtj
ddd�d�gd� S )r� rS r� r Nr � �� T� ns1.example.com� hostmaster.example.com� �� ������� �� �ri r� r� r� r� r� r� r� �rf ri r7 rk �ns1.example.comrh �5.6.7.8) r? r6 r@ r7 r9 r: rA r� rn rp rM rs � r� r r� ro r� rq r� �NSr� rt rv r� r r r rr $
sX ��� ��
���zMessageComplete.kwargsNr� r r r r r� � s
1r� c @ r� )�MessageEDNSQueryz'
A minimal EDNS query message.
c C r� )r� s, wwwexamplecom ) r r� r r r rP b
r� zMessageEDNSQuery.bytesc C s, t ddtjddddddt�dtj�gg d� S )r� r r� Fs www.example.com) r? r6 r@ r7 r9 r: rA r� r� rn rs )r� r rU ro rt r� r r r rr �
s �zMessageEDNSQuery.kwargsNr� r r r r r ]
s
r c @ r� )�MessageEDNSCompletez�
An example of a fully populated edns response message.
Contains name compression, answers, authority, and additional records.
c C r� )r� sy �� examplecom � ���� 'ns1�
hostmaster� ������������������ ���� �)�) ���� ) � r r� r r r rP �
r� zMessageEDNSComplete.bytesc C s� t d(i dd�dd�dtj�dd�dd�d d�d
d�d d �d
d�dd�dd�dd�dd�dt�dtj�g�dtjdtjddtjddddd d d d d �d!�g�d"tjdtjddtjd#dd$�d!�g�d%tjdtj ddtj
d&dd$�d!�g��S ))r� r? rS r6 r� r@ r7 r8 r r9 r: rA r r� r� r� Tr r, rB � rn r rp r� r� r� r� r� r� r� r� r r rM r rh rs r Nr r r� r r r rr �
s� �������� �
� � �
������" ���+ ���zMessageEDNSComplete.kwargsNr� r r r r r �
s
9r c @ r� )�MessageEDNSExtendedRCODEz?
An example of an EDNS message with an extended RCODE.
c C r� )r� s ) � r r� r r r rP r� z MessageEDNSExtendedRCODE.bytesc C s( t ddtjddddddddg g g g d�S )r� r Fi�
r� )r? r6 r@ r7 r8 r9 r: rA r� r� rB rn rp rM rs )r� r rU r� r r r rr 7 s" �z MessageEDNSExtendedRCODE.kwargsNr� r r r r r s
r 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? r6 r@ r7 r8 r9 r: rA rn rp rM rs c C s
|| _ d S r� )�original)r r r r r r� g �
zMessageComparable.__init__c C s t | j|�S r� )�getattrr )r �keyr r r � __getattr__j r� z MessageComparable.__getattr__N)r& r' r( r/ �showAttributes�compareAttributesr� r r r r r r
R s
r
c C s^ |du r|}i }||d�}|� }t ||�|d< |di ||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�altValr r r )r
r ) rm rg �argNamer r �attrName�actualrc r� r r r �verifyConstructorArgumentn s
r 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.
�rm rg r r r N�r rs )r r r r r r r �_verifyConstructorArgument� s
�z0ConstructorTestsMixin._verifyConstructorArgumentc C s& |dv sJ �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
)TFr Nr )r r r r r r �_verifyConstructorFlag� s
�z,ConstructorTestsMixin._verifyConstructorFlagN)r& r' r( r/ r r r r r r r � s r 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 � | j dddd� dS )zk
L{dns._EDNSMessage.id} defaults to C{0} and can be overridden in
the constructor.
r? r r� r N�r r r r r rx � � z#CommonConstructorTestsMixin.test_idc C � | j ddd� dS )zs
L{dns._EDNSMessage.answer} defaults to C{False} and can be overridden in
the constructor.
r6 F�r N�r r r r r r{ � � 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@ r N)r r rU r� r r r r r} � s
�z'CommonConstructorTestsMixin.test_opCodec C r" )zq
L{dns._EDNSMessage.auth} defaults to C{False} and can be overridden in
the constructor.
r7 Fr# Nr$ r r r r r� � r% z%CommonConstructorTestsMixin.test_authc C r" )zr
L{dns._EDNSMessage.trunc} defaults to C{False} and can be overridden in
the constructor.
r8 Fr# Nr$ r r r r r� � r% z&CommonConstructorTestsMixin.test_truncc C r" )zs
L{dns._EDNSMessage.recDes} defaults to C{False} and can be overridden in
the constructor.
r9 Fr# Nr$ r r r r r � r% z'CommonConstructorTestsMixin.test_recDesc C r" )zr
L{dns._EDNSMessage.recAv} defaults to C{False} and can be overridden in
the constructor.
r: Fr# Nr$ r r r r r� � r% z&CommonConstructorTestsMixin.test_recAvc C r )zn
L{dns._EDNSMessage.rCode} defaults to C{0} and can be overridden in the
constructor.
rA r rL r Nr r r r r r� r! z&CommonConstructorTestsMixin.test_rCodec C r )zr
L{dns._EDNSMessage.maxSize} defaults to C{512} and can be overridden in
the constructor.
rB r� r r Nr r r r r r� r! z(CommonConstructorTestsMixin.test_maxSizec C � | � | �� jg � dS )z@
L{dns._EDNSMessage.queries} defaults to C{[]}.
N)r rs rn r r r r r� r z(CommonConstructorTestsMixin.test_queriesc C r& )z@
L{dns._EDNSMessage.answers} defaults to C{[]}.
N)r rs rp r r r r r� r z(CommonConstructorTestsMixin.test_answersc C r& )zB
L{dns._EDNSMessage.authority} defaults to C{[]}.
N)r rs rM r r r r r� r z*CommonConstructorTestsMixin.test_authorityc C r& )zC
L{dns._EDNSMessage.additional} defaults to C{[]}.
N)r rs rs r r r r r� % r z+CommonConstructorTestsMixin.test_additionalN)r& r' r( r/ rx r{ r} r� r� r r� r� r� r� r� r� r� r r r r r � s r c @ � 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 � _EDNSMessagers r r r r r( , �
r( c @ r' )�MessageConstructorTestszt
Tests for L{twisted.names.dns.Message} constructor arguments that are shared
with L{dns._EDNSMessage}.
N)r& r' r( r/ r r_ rs r r r r r, 7 r+ 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"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 Nr r r r r r �test_ednsVersionL r! z*EDNSMessageSpecificsTests.test_ednsVersionc C r" )zu
L{dns._EDNSMessage.dnssecOK} defaults to C{False} and can be overridden
in the constructor.
r� Fr# Nr$ r r r r r� S r% z'EDNSMessageSpecificsTests.test_dnssecOKc C r" )zz
L{dns._EDNSMessage.authenticData} defaults to C{False} and can be
overridden in the constructor.
r Fr# Nr$ r r r r r� Z r% z,EDNSMessageSpecificsTests.test_authenticDatac C r" )z}
L{dns._EDNSMessage.checkingDisabled} defaults to C{False} and can be
overridden in the constructor.
r, Fr# Nr$ r r r r r� a r% 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)rs r ro r rn rQ r r r �test_queriesOverrideh s z.EDNSMessageSpecificsTests.test_queriesOverridec C �B | j tjdt�d�d�gd�}| �|jtjdt�d�d�g� dS )zS
L{dns._EDNSMessage.answers} can be overridden in the constructor.
r rl r_ r� N)rs r rq rv r rp rQ r r r �test_answersOverridep 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)rs r rq r� r� r rM rQ r r r �test_authorityOverride| s ���z0EDNSMessageSpecificsTests.test_authorityOverridec C r0 )r2 r rl r_ r� N)rs r rq rv r rs rQ r r r �test_additionalOverride� s ��z1EDNSMessageSpecificsTests.test_additionalOverridec C � | � 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 r3 rs r r r r r4 � r5 z+EDNSMessageSpecificsTests.test_reprDefaultsc
C s. | j ddddddddd�}| �dt|�� dS )zR
L{dns._EDNSMessage.__repr__} displays flags if they are L{True}.
T)r6 r7 r8 r9 r: r r, r� z`<_EDNSMessage id=0 flags=answer,auth,trunc,recDes,recAv,authenticData,checkingDisabled,dnssecOK>N�rs r r3 r' r r r r< � 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.
r7 r| r= r> r� )r? r@ rA rB r� zA<_EDNSMessage id=10 opCode=20 rCode=30 maxSize=40 ednsVersion=50>Nr7 r' r r r rC � s
�z2EDNSMessageSpecificsTests.test_reprNonDefautFieldsc C sD | � � }g d�|_g d�|_g d�|_g d�|_| �dt|�� dS )rD rE rF rH rK zc<_EDNSMessage id=0 queries=[1, 2, 3] answers=[4, 5, 6] authority=[7, 8, 9] additional=[10, 11, 12]>N)rs rn rp rM rs r r3 r' r r r rN � rO z5EDNSMessageSpecificsTests.test_reprNonDefaultSectionsc C sP G dd� d�}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 @ r) )zUEDNSMessageSpecificsTests.test_fromStrCallsMessageFactory.<locals>.FakeMessageFactory�/
Fake message factory.
c _ s
t ||��)z�
Fake fromStr method which raises the arguments it was passed.
@param args: positional arguments
@param kwargs: keyword arguments
�r� r� r r r rm � s
z]EDNSMessageSpecificsTests.test_fromStrCallsMessageFactory.<locals>.FakeMessageFactory.fromStrN�r& r' r( r/ rm r r r r �FakeMessageFactory� r0 r; N)
r r* �_messageFactoryr2 r" r� rm r rq rr )r r; r( �
dummyBytesr� r r r � test_fromStrCallsMessageFactory� 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 @ r) )zREDNSMessageSpecificsTests.test_fromStrCallsFromMessage.<locals>.FakeMessageFactoryr8 c S s dS )zw
A noop fake version of fromStr
@param bytes: the bytes to be decoded
Nr )r rP r r r rm � zZEDNSMessageSpecificsTests.test_fromStrCallsFromMessage.<locals>.FakeMessageFactory.fromStrNr: r r r r r; � r0 r; c � � S r� r r �� fakeMessager r �<lambda> r? zHEDNSMessageSpecificsTests.test_fromStrCallsFromMessage.<locals>.<lambda>c _ �
t | |��r� r9 r� r r r �fakeFromMessage
r zOEDNSMessageSpecificsTests.test_fromStrCallsFromMessage.<locals>.fakeFromMessager� N)
r r* r< � _fromMessager" r� rm r rq rr )r r( r; rE r� r rA 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 _ rD r� r9 r� r r r �
fakeToMessage r zIEDNSMessageSpecificsTests.test_toStrCallsToMessage.<locals>.fakeToMessager N) r r* �
_toMessager" r� rw r rq rr )r r( rH r� r r r �test_toStrCallsToMessage s
z2EDNSMessageSpecificsTests.test_toStrCallsToMessagec sF t �� }t� �G �fdd�d�� � 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 r@ )zk
Fake toStr which returns dummyBytes.
@return: dummyBytes
r r �r= r r rw . s zREDNSMessageSpecificsTests.test_toStrCallsToMessageToStr.<locals>.FakeMessage.toStrN)r& r' r( r/ rw r rK r r � FakeMessage) s rL c s � � S r� r r� )rL r r rH 6 s zNEDNSMessageSpecificsTests.test_toStrCallsToMessageToStr.<locals>.fakeToMessageN)r r* r2 rI r rw )r r( rH r )rL r= r � test_toStrCallsToMessageToStr! s
z7EDNSMessageSpecificsTests.test_toStrCallsToMessageToStrN)r& r' r( r/ r r* rs r. r� r� r� r/ r1 r3 r4 r4 r< rC rN r> rG rJ rM r r r r r- B 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 rt )zc
Two L{dns._EDNSMessage} instances compare equal if they have the same
id.
r� ru r� Nrv r r r r rx H ry z EDNSMessageEqualityTests.test_idc C rt )zl
Two L{dns._EDNSMessage} instances compare equal if they have the same
answer flag.
Trz FNrv r r r r r{ S ry 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)rw rs r r� �
OP_INVERSEr r r r r} ^ s
�z$EDNSMessageEqualityTests.test_opCodec C rt )zj
Two L{dns._EDNSMessage} instances compare equal if they have the same
auth flag.
Tr� FNrv r r r r r� i ry z"EDNSMessageEqualityTests.test_authc C rt )zk
Two L{dns._EDNSMessage} instances compare equal if they have the same
trunc flag.
Tr� FNrv r r r r r� t ry z#EDNSMessageEqualityTests.test_truncc C rt )zl
Two L{dns._EDNSMessage} instances compare equal if they have the same
recDes flag.
Tr~ FNrv r r r r r ry z$EDNSMessageEqualityTests.test_recDesc C rt )zk
Two L{dns._EDNSMessage} instances compare equal if they have the same
recAv flag.
Tr� FNrv r r r r r� � ry z#EDNSMessageEqualityTests.test_recAvc C rt )zf
Two L{dns._EDNSMessage} instances compare equal if they have the same
rCode.
r, r� r Nrv r r r r r� � ry z#EDNSMessageEqualityTests.test_rCodec C s* | � | jdd�| jdd�| jdd�� dS )zl
Two L{dns._EDNSMessage} instances compare equal if they have the same
ednsVersion.
r� �r� Nrv r r r r r. � ry z)EDNSMessageEqualityTests.test_ednsVersionc C rt )zi
Two L{dns._EDNSMessage} instances compare equal if they have the same
dnssecOK.
Tr� FNrv r r r r r� � ry z&EDNSMessageEqualityTests.test_dnssecOKc C rt )zt
Two L{dns._EDNSMessage} instances compare equal if they have the same
authenticData flags.
Tr FNrv r r r r r� � ry z+EDNSMessageEqualityTests.test_authenticDatac C rt )zw
Two L{dns._EDNSMessage} instances compare equal if they have the same
checkingDisabled flags.
Tr. FNrv r r r r r� � ry z.EDNSMessageEqualityTests.test_checkingDisabledc C rt )zh
Two L{dns._EDNSMessage} instances compare equal if they have the same
maxSize.
i r� r Nrv r r r r r� � ry z%EDNSMessageEqualityTests.test_maxSizec C r� )zh
Two L{dns._EDNSMessage} instances compare equal if they have the same
queries.
r r� r� Nr� r r r r r� � r� z%EDNSMessageEqualityTests.test_queriesc
C r� )zh
Two L{dns._EDNSMessage} instances compare equal if they have the same
answers.
r rl r_ r� r� r� Nr� r r r r r� � r� z%EDNSMessageEqualityTests.test_answersc
C r� )zr
Two L{dns._EDNSMessage} instances compare equal if they have the same
authority records.
r r� r� r� Nr� r r r r r� � r� z'EDNSMessageEqualityTests.test_authorityc
C r� )zs
Two L{dns._EDNSMessage} instances compare equal if they have the same
additional records.
r rl r_ r� r� Nr� r r r r r�
r� z(EDNSMessageEqualityTests.test_additionalN)r& r' r( r/ r r* rs rx r{ r} r� r� r r� r� r. r� r� r� r� r� r� r� r� r r r r rN > s( rN 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 �( | � | jdi t�� ���� t�� � dS )z2
An empty message can be encoded.
Nr )r rs r� rr rw rP r r r r �test_emptyMessageEncode/ s �z2StandardEncodingTestsMixin.test_emptyMessageEncodec C �6 | � � }|�t�� � | �|| j di t�� ��� dS )z@
An empty message byte sequence can be decoded.
Nr )rs rm r� rP r rr r' r r r �test_emptyMessageDecode7 s z2StandardEncodingTestsMixin.test_emptyMessageDecodec C rR )zA
A fully populated query message can be encoded.
Nr )r rs r� rr rw rP r r r r �test_completeQueryEncode@ s �z3StandardEncodingTestsMixin.test_completeQueryEncodec C s8 | � � }|�t�� �f | �|| j di t�� ��� dS )zG
A fully populated message byte string can be decoded.
Nr )rs rm r� rP r rr r' r r r �test_completeQueryDecodeI 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 rW r rX rY r r r r\ W r] z8StandardEncodingTestsMixin.test_NULL.<locals>.<listcomp>rS r^ r_ r N)r` r� r r� rq ra rs rp rb rw rm rN rk r rc r r r r� R s
z$StandardEncodingTestsMixin.test_NULLc C rR )zv
If the message C{authoritative} attribute is set to 0, the encoded bytes
will have AA bit 0.
Nr )r rs r� rr rw rP r r r r �"test_nonAuthoritativeMessageEncoded � �z=StandardEncodingTestsMixin.test_nonAuthoritativeMessageEncodec C rT )z�
The L{dns.RRHeader} instances created by a message from a
non-authoritative message byte string are marked as not authoritative.
Nr )rs rm r� rP r rr r' r r r �"test_nonAuthoritativeMessageDecoden � z=StandardEncodingTestsMixin.test_nonAuthoritativeMessageDecodec C rR )zv
If the message C{authoritative} attribute is set to 1, the encoded bytes
will have AA bit 1.
Nr )r rs r� rr rw rP r r r r � test_authoritativeMessageEncodex rY z:StandardEncodingTestsMixin.test_authoritativeMessageEncodec C rT )z�
The message and its L{dns.RRHeader} instances created by C{decode} from
an authoritative message byte string, are marked as authoritative.
Nr )rs rm r� rP r rr r' r r r � test_authoritativeMessageDecode� r[ z:StandardEncodingTestsMixin.test_authoritativeMessageDecodec C rR )zm
If the message C{trunc} attribute is set to 1 the encoded bytes will
have TR bit 1.
Nr )r rs r� rr rw rP r r r r �test_truncatedMessageEncode� rY z6StandardEncodingTestsMixin.test_truncatedMessageEncodec C rT )zn
The message instance created by decoding a truncated message is marked
as truncated.
Nr )rs rm r� rP r rr r' r r r �test_truncatedMessageDecode� r[ z6StandardEncodingTestsMixin.test_truncatedMessageDecodeN)r& r' r( r/ rS rU rV rW r� rX rZ r\ r] r^ r_ r r r r rQ " s
rQ c @ r' )� EDNSMessageStandardEncodingTestszq
Tests for the encoding and decoding of various standard (non-EDNS) messages
by L{dns._EDNSMessage}.
Nr) r r r r r` � r+ 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 sl | � dg �}| � dg �}| � dg �}| � dg �}| � dd� tjdi | ��}||_||_||_||_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.
rn rp rM rs r� Nr )rp r r_ rn rp rM rs r
)rr rn rp rM rs r( r r r rs � s z+MessageStandardEncodingTests.messageFactoryN)r& r' r( r/ � staticmethodrs r r r r ra � s ra 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)rs rm r rP r rs r' r r r �&test_ednsMessageDecodeStripsOptRecords� 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� r� N) r r_ r� rs r* rm rw � assertIsNoner� )r r( � ednsMessager r r �(test_ednsMessageDecodeMultipleOptRecords� s
zEEDNSMessageEDNSEncodingTests.test_ednsMessageDecodeMultipleOptRecordsc C sj t �� }|�t�� � t j�|�}g }dD ]}t||�t||�u r&|�|� q|r3| � d|f � dS 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.
)rn rp rM rs z\Message and _EDNSMessage shared references to the following section lists after decoding: %sN)
r r_ rm r rP r* rF r
rb �fail)r �standardMessagerf �
duplicatesr 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.
r� rP N)r r* rw r rs )r rf 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}
r� r� r )rf r� N)r r r� rW rq r� rS r_ rs rb r* rF r� rt �insertr )r r� � optRecord� optRRHeaderr( �actualMessagesr r r �test_optHeaderPosition s
z3EDNSMessageEDNSEncodingTests.test_optHeaderPositionc C rT )z�
The L(_EDNSMessage} instance created by L{dns._EDNSMessage.fromStr}
derives its edns specific values (C{ednsVersion}, etc) from the supplied
OPT record.
Nr )rs rm r rP r rr r' r r r �test_ednsDecode9 s z,EDNSMessageEDNSEncodingTests.test_ednsDecodec C rR )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.
Nr )r rs r rr rw rP r r r r �test_ednsEncodeD � �z,EDNSMessageEDNSEncodingTests.test_ednsEncodec C rR )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.
Nr )r rs r rr rw rP r r r r �test_extendedRcodeEncodeO rt z5EDNSMessageEDNSEncodingTests.test_extendedRcodeEncodec C rT )z�
The L(_EDNSMessage} instance created by L{dns._EDNSMessage.fromStr}
derives RCODE from the supplied OPT record.
Nr )rs rm r rP r rr r' r r r �test_extendedRcodeDecodeZ r[ 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 )rA r� )r r N)rs rI r rA rs r� )r rf ri r r r �test_extendedRcodeZerod s �z3EDNSMessageEDNSEncodingTests.test_extendedRcodeZeroN)r& r' r( r/ r r* rs rd rg rk rl rq rr rs ru rv rw r r r r rc � s
rc c @ r� )�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.
��responseConstructorrk N)r r_ �_responseFromMessage� assertIsNot�r �request�responser r r �$test_responseFromMessageResponseTypey s
�z=ResponseFromMessageTests.test_responseFromMessageResponseTypec C s0 G dd� d�}t j}| �t j||� d�|� dS )zN
L{dns._responseFromMessage} returns a new instance of C{cls}
c @ s e Zd ZdZg ZdS )zAResponseFromMessageTests.test_responseType.<locals>.SuppliedClassr� N)r& r' r( r? rn r r r r �
SuppliedClass� s r� ry N)r r_ rN r{ )r r� �
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� ru ry N)r r r{ r_ 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}
ry )FTN)r r_ r{ r r6 r} r r r �test_responseAnswer� 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.
Nry )r r_ r2 rn r r{ )r r~ �expectedQueriesr r r �test_responseQueries� 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.
rL )rz rk rA N)r r r{ r_ rA r r r r �test_responseKwargs� s ��z,ResponseFromMessageTests.test_responseKwargsN)
r& r' r( r/ r� r� r� r� r� r� r r r r rx t s
rx c @ s4 e Zd ZdZ d
dd �Zd
efd 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}.
r� r� �ASTFNc C s4 || _ || _|| _|| _|| _|du rg }|| _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 r� � s
z Foo.__init__�returnc C s* t ttj| d�� d�� d�� d�� d��S )zO
Call L{dns._compactRepr} to generate a string representation.
r� z field1 field2 alwaysShowFieldzflagTrue flagFalsezsection1 section2)�
alwaysShow�
fieldNames� flagNames� sectionNames)r �strr � _compactRepr�splitr r r r �__repr__� s ��z Foo.__repr__)r� r� r� TFN)r& r' r( r/ r� r� r� r r r r r� � s
�r� c @ s4 e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d
S ) �CompactReprTestsz(
Tests for L{dns._compactRepr}.
c C r5 )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>Nr6 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>Nr7 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.
r7 r| )r� r� z=<Foo field1=10 field2=20 alwaysShowField='AS' flags=flagTrue>Nr7 r' r r r �test_nonDefautFields s
�z%CompactReprTests.test_nonDefautFieldsc C s0 | � � }g d�|_g d�|_| �dt|�� dS )zY
L{dns._compactRepr} displays sections which differ from their defaults.
)r� r� r� )r� r� r� zO<Foo alwaysShowField='AS' flags=flagTrue section1=[1, 1, 1] section2=[2, 2, 2]>N)rs r� �section2r r3 r' r r r �test_nonDefaultSections! s
�z(CompactReprTests.test_nonDefaultSectionsN)
r& r' r( r/ r� rs r� r� r� r� r r r r r� � s
r� r� )hr/ r� �ior �typingr �zope.interface.verifyr �twisted.internetr r �twisted.internet.errorr r �
twisted.namesr �twisted.python.failurer
�twisted.python.utilr r � twisted.testr
�twisted.test.testutilsr �
twisted.trialr r� r� r� r� r� r r r r rv r� r� r� r� r� r� r r r r r� r� r� r� r rr r� �SynchronousTestCaser �TestCaser* r1 rK r� r# r0 r ro r� r� r� r� r� r& rS r] rp rr rs r� r� r� � Exceptionr� r� r� r� r� r� r r r r
r r r r( r, r- rN rQ r` ra rc rx r� r� r r r r �<module> s� � %
> N e� � u eaT ".3_5 iP.-11r;;
�.*
f�
� } e
�
�, V-