File: //opt/alt/python37/lib/python3.7/site-packages/pydnsbl/__pycache__/tests.cpython-37.pyc
B
��fn � @ s� d dl Z d dlZd dlZddlmZmZmZ ddlmZ dd� Z dd� Z
d d
� Z d d � Z d
d� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � ZdS ) � N� )� DNSBLChecker�DNSBLIpChecker�DNSBLDomainChecker)�Providerc C sb t � } | �d�}|jst�|js$t�|js.t�| �ddg�}|d j|jksPt�|d jr^t�d S )Nz68.128.212.240z8.8.8.8r r )r �check� blacklisted�AssertionError�
categories� detected_by�
bulk_check)�checker�res�results� r �>/opt/alt/python37/lib/python3.7/site-packages/pydnsbl/tests.py� test_checker s
r c C st t � } | �d�}|jrt�|jr$t�|jr.t�|jr8t�t td�gd�} | �d�}|js\t�|jsft�|jspt�d S )Nz2001:4860:4860::8844zv6.fullbogons.cymru.com)� providersz::1)r r r r r
r �failed_providersr )r
r r r r �test_checker_ipv6 s
r c C s< t � } | �d�}|jrt�|jr$t�|jr.t�|jr8t�dS )z, Providers should not mark google ip as bad z8.8.8.8N)r r r r r
r r )r
r r r r �test_providers! s
r c
C s@ ddg} x2| D ]*}t � }t�t�� |�|� W d Q R X qW d S )N�abcz 8.8.8.256)r �pytest�raises�
ValueErrorr )�misformated_ips�ipr
r r r �test_wrong_ip_format* s
r c C sf t � } d}| �|�}|js t�|js(t�|js2t�| �|dg�}|d j|jksTt�|d jrbt�d S )Nz dbltest.comz
google.comr r )r r r r r
r r )r
Zmalicious_domainr r r r r �test_domain_checker2 s
r c C s< t � } | �d�}|jrt�|jr$t�|jr.t�|jr8t�dS )z4 Domain Providers should not mark google.com as bad z
google.comN)r r r r r
r r )r
r r r r �test_domain_providers> s
r c
C sF dddg} x6| D ].}t � }t�t�� t|�|�� W d Q R X qW d S )Nzabc-z 8.8.8.256Zbababa)r r r r �printr )r r r
r r r �test_wrong_domain_formatG s
r! c C sT dddg} xD| D ]<}t � }|�|�}|jr.t�|jr8t�|jrBt�|jrt�qW d S )Nz
Google.comu дом.рфzwww.digital.govt.nz)r r r r r
r r )Zcapitalized_domains�domainr
r r r r �test_domain_variantsO s
r# c s: d � � fdd�} t j| d�}|�� |�� � js6t�d S )Nc s t � } | �d�� d S )Nz68.128.212.240)r r )r
)�resultr r �test] s z test_main_thread.<locals>.test)�target)� threading�Thread�start�joinr r )r% Zthrr )r$ r �test_main_thread[ s r+ c C s t � } | �d�dkst�d S )Nz2001:db8:1:2:3:4:567:89abz?b.a.9.8.7.6.5.0.4.0.0.0.3.0.0.0.2.0.0.0.1.0.0.0.8.b.d.0.1.0.0.2)r Z
prepare_queryr )r
r r r �test_ipv6_convertingg s r, c C sb t � } | �d�}|jst�|js$t�|js.t�| �ddg�}|d j|jksPt�|d jr^t�d S )Nz68.128.212.240z8.8.8.8r r )r �check_ipr r r
r Z check_ips)r
r r r r r �test_checker_compat_0_6n s
r. c C s< t � } | �d�}|jrt�|jr$t�|jr.t�|jr8t�dS )z, Providers should not mark google ip as bad z8.8.8.8N)r r- r r r
r r )r
r r r r �test_providers_compat_0_6y s
r/ c
C s@ ddg} x2| D ]*}t � }t�t�� |�|� W d Q R X qW d S )Nr z 8.8.8.256)r r r r r- )r r r
r r r � test_wrong_ip_format_compat_0_6� s
r0 )Zasyncior' r r
r r r r r r r r r r r r! r# r+ r, r. r/ r0 r r r r �<module> s"