File: //usr/local/CyberCP/lib64/python3.10/site-packages/DNS/tests/__pycache__/test_base.cpython-310.pyc
o
�h�) � @ sr d dl Z d dlZzd dlZW n
ey d dlZY nw dd� ZG dd� dej�Zdd� Ze dkr7e�
� dS dS ) � Nc C s | dksJ �| dks J �d S )Nr � � )�br r �F/usr/local/CyberCP/lib/python3.10/site-packages/DNS/tests/test_base.py� assertIsByte s r c @ s� e Zd Zdd� Zdd� Zdd� 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'S )(�TestBasec C s� g t jd< dt jv rt jd= | �tt jd �d� g d�}t �|� | �tt jd �d� | �t jd d d� | �t jd d� d S )N�server�domainr )z # a commentzdomain example.orgznameserver 127.0.0.1� z 127.0.0.1� example.org)�DNS�defaults� assertEqual�len�ParseResolvConfFromIterable)�self�resolvr r r �testParseResolvConf s
z TestBase.testParseResolvConfc C s, t �d�}|jdddd�}| �|j� | �|jd d �d�d � | �|jd d d
� |jddd �}| �|j� | �|jd d t�d
�� |jdd dd�}| �|j� | �t |jd d �d
� |jd d D ]}t
|� qi| �|jd d d� |jdddd�}| �|j� | �|jd d d� d S )Nr �A�textr
��qtype�
resulttype�timeoutr �data�.� �
93.184.215.14�r r �binary� s ]���integeri]) r �
DnsRequest�qry�
assertTrue�answersr �count� ipaddress� IPv4Addressr r )r �dnsobj�
a_response� ad_response� ab_responser � ai_responser r r �testDnsRequestA s"
zTestBase.testDnsRequestAc C s( t �d�}|jdddd�}| �|j� | �d|jd d v � | �|jd d d � |jddd
�}| �|j� | �|jd d t�d �� |jdd dd�}| �|j� | �t|jd d �d � |jd d D ]}t |� qg| �|jd d d
� |jdddd�}| �|j� | �|jd d d� d S )Nr �AAAAr r
r �:r r z&2606:2800:21f:cb07:6820:80da:af6b:8b2cr r � � &( �h �گk�,r! l , �^jAv0�C & )
r r" r# r$ r% r r' � IPv6Addressr r )r r) �
aaaa_response�aaaad_response�aaaab_responser �aaaai_responser r r �testDnsRequestAAAA; s"
zTestBase.testDnsRequestAAAAc C �( t �d�}|jddd�}| �|j� d S �Nzmail.kitterman.org�MXr
r )r r" r# � assertFalser% )r r) �mx_empty_responser r r �testDnsRequestEmptyMXU �
z TestBase.testDnsRequestEmptyMXc C �j t �d�}|jddd�}| �|jd � | �|jd d d� t jddd�}| �|jd d |d � d S � Nzietf.orgr; r
r r r )r z
mail.ietf.org�r )r r" r# r$ r% r �mxlookup)r r) � mx_response�mr r r �testDnsRequestMX[ �
zTestBase.testDnsRequestMXc C �j t jdd�}|jddd�}| �|j� |jd d }| �t|�d� | �|d d
� | �d |d v � d S �
N�srv�r z_ldap._tcp.openldap.orgr
rB r r r � i� z openldap.orgr )r �Requestr# r$ r% r r )r r) �respdefr r r r �testDnsRequestSrvf � zTestBase.testDnsRequestSrvc C �d d}t j|dd�}|jdd�}| �|j� |jd d }| �t|d t�� | �|d �d�� d S � Nz 20161025._domainkey.google.com�txtrK r
rB r r s k=rsa) r rM r# r$ r% r< �
isinstance�str�
startswith)r �qr) �respr r r r �testDkimRequesto � zTestBase.testDkimRequestc C s� t �d�}|jddd�}| �|j� |jd d }| �|dg� |jdddd �}| �|j� |jd d }| �|d
g� |jdd dd �}| �|j� |jd d }| �|d g� d S )
N�fail.kitterman.org�TXTr
r r r � v=spf1 -allr r z v=spf1 -allr s v=spf1 -all�r r" r# r$ r% r )r r) rN r �resptext�respbinr r r �testDNSRequestTXTz s
zTestBase.testDNSRequestTXTc C �p t �d�}t �d�}|jdddd�}|jdddd�}| �|j� | �|j� | �|jd d |jd d � d S �
z/Can we lookup an internationalized domain name?z
xn--bb-eka.atu öbb.atr r r
r r r Nr^ )r r) � unidnsobj�a_resp�ua_respr r r �testIDN� �
�zTestBase.testIDNc C s~ t �d�}|jddd�}| �|j� |jd d d }| �|d� |jd|dd �}d
d � |jD �}|�� | �|dd g� d
S )�Lookup NS record from SOA�
kitterman.com�SOAr
r r r �ns1.pairnic.com�NS)r r �aac S � g | ]}|d � � �qS �r ��lower��.0�xr r r �
<listcomp>� � z#TestBase.testNS.<locals>.<listcomp>�ns2.pairnic.comN)r r" r# r$ r% r �sort�r �dnsobrX �primary�nslistr r r �testNS� s
zTestBase.testNSc C sZ t �d�}|jddd�}| �|j� | �|jd d �d�d� | �|jd d d � d S )
Nr r r
r r r r r r )r r" �reqr$ r% r r& )r r{ r+ r r r �testDnsRequestAD� s
zTestBase.testDnsRequestADc C st t �d�}|jddd�}| �|j� | �t|jd d �d� |jd d D ]}t|� q&| �|jd d d� d S ) Nr r/ r
r r r r1 r2 )r r" r r$ r% r r r )r r{ r5 r r r r �testDnsRequestAAAAD� s
z TestBase.testDnsRequestAAAADc C r9 r: )r r" r r< r% )r r{ r= r r r �testDnsRequestEmptyMXD� r? z TestBase.testDnsRequestEmptyMXDc C r@ rA )r r" r r$ r% r rC )r r{ rD rE r r r �testDnsRequestMXD� rG zTestBase.testDnsRequestMXDc C rH rI )r rM r r$ r% r r �r r{ rN r r r r �testDnsRequestSrvD� rP zTestBase.testDnsRequestSrvDc C rQ rR ) r rM r r$ r% r< rT rU rV )r rW r{ rX r r r r �testDkimRequestD� rZ zTestBase.testDkimRequestDc C sD t �d�}|jddd�}| �|j� |jd d }| �|dg� d S )Nr[ r\ r
r r r r] �r r" r r$ r% r r� r r r �testDNSRequestTXTD� s
zTestBase.testDNSRequestTXTDc C rb rc r� )r r{ �unidnsobre rf r r r �testIDND� rh zTestBase.testIDNDc C s� t �d�}|jddd�}| �|j� |jd d d }| �|d� |jd|ddd �}d
d � |jD �}|�� | �|dd g� d
S )ri rj rk r
r r r rl rm )r r rn r c S ro rp rq rs r r r rv � rw z$TestBase.testNSD.<locals>.<listcomp>rx N)r r" r r$ r% r ry rz r r r �testNSD� s
zTestBase.testNSDN)�__name__�
__module__� __qualname__r r. r8 r> rF rO rY ra rg r~ r� r� r� r� r� r� r� r� r� r r r r r s(
r c C s ddl m} | � �t�S )Nr ��
TestLoader)�unittestr� �loadTestsFromNamer� r� r r r �
test_suite� s r� �__main__) r r� r' � ImportError�ipaddrr �TestCaser r� r� �mainr r r r �<module> s � q �