File: //usr/lib/python3/dist-packages/twisted/names/test/__pycache__/test_cache.cpython-38.pyc
U
W[4 � @ sj d Z ddlmZmZ ddlZddlmZ ddlmZ ddl m
Z
m Z ddl m
Z
mZ G dd � d ej�ZdS )
z#
Tests for L{twisted.names.cache}.
� )�division�absolute_importN)� verifyClass)�unittest)�dns�cache)�task�
interfacesc @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� CachingTestsz+
Tests for L{cache.CacheResolver}.
c C s t tjtj� dS )zK
L{cache.CacheResolver} implements L{interfaces.IResolver}
N)r r Z IResolverr �
CacheResolver��self� r �?/usr/lib/python3/dist-packages/twisted/names/test/test_cache.py�test_interface s zCachingTests.test_interfacec C sF t �tjdtjtjd�t�� g g g ffi�}|�d��| j g g g f�S )N� example.com��name�type�cls)
r r r �QueryZMX�IN�timeZlookupMailExchange� addCallback� assertEqual)r
�cr r r � test_lookup s �
�zCachingTests.test_lookupc C s� t �dt jt jdt �dd��gt �dt jt jdt �dd��gt �dt jt jdt �dd��gf}t�� }t jdt jt jd�}tj ||�
� |fi|d�}|� d� | � ||j� | �
|�d�t j�S )z�
Cache entries passed into L{cache.CacheResolver.__init__} get
cancelled just like entries added with cacheResult
r �< � 127.0.0.1�2 �( r �Zreactor)r �RRHeader�Ar �Record_Ar �Clockr r r �seconds�advance� assertNotIn�
assertFailure�
lookupAddress� DomainError)r
�r�clock�queryr r r r �test_constructorExpires( s&
�
�
��
�z$CachingTests.test_constructorExpiresc s� t �dt jt jdt �dd��gt �dt jt jdt �dd��gt �dt jt jdt �dd��gf}t�� }tj|d�}|� t j
dt jt jd�|� |� d� � fd d
�}|� d��
|�S ) z�
When a cache lookup finds a cached entry from 1 second ago, it is
returned with a TTL of original TTL minus the elapsed 1 second.
r r r r r r! r � c s^ � � | d d jd� � � | d d jd� � � | d d jd� � � | d d jjd� d S )Nr �; r0 �1 � �' r )r Zttlr )�resultr r r �cbLookupV s z0CachingTests.test_normalLookup.<locals>.cbLookup)r r" r# r r$ r r% r r � cacheResultr r' r* r )r
r, r- r r6 r r r �test_normalLookupC s
�
�
��
z CachingTests.test_normalLookupc C s� t �dt jt jdt �dd��gt �dt jt jdt �dd��gt �dt jt jdt �dd��gf}t�� }tj|d�}t j dt jt jd�}|�
||� |� d� | � ||j� | �
|�d�t j�S )zW
Once the TTL has been exceeded, the result is removed from the cache.
r r r r r r! r )r r" r# r r$ r r% r r r r7 r' r( r) r* r+ )r
r, r- r r. r r r �test_cachedResultExpires_ s(
�
�
��
�z%CachingTests.test_cachedResultExpiresc C s� t �dt jt jdt �dd��gt �dt jt jdt �dd��gt �dt jt jdt �dd��gf}t�� }dd� |_tj t j
dt jt jd�|� � |fi|d �}|� d
� | �
|�d�t j�S ) z�
When the cache is queried exactly as the cached entry should expire but
before it has actually been cleared, the cache does not return the
expired entry.
r r r r r c _ s d S )Nr )�args�kwargsr r r �<lambda>� � z4CachingTests.test_expiredTTLLookup.<locals>.<lambda>r r! g����� N@)r r" r# r r$ r r% Z callLaterr r r r& r' r) r* r+ )r
r, r- r r r r �test_expiredTTLLookupx s.
�
�
��
��
�z"CachingTests.test_expiredTTLLookupN)
�__name__�
__module__� __qualname__�__doc__r r r/ r8 r9 r> r r r r r
s r
)rB Z
__future__r r r Zzope.interface.verifyr Z
twisted.trialr Z
twisted.namesr r Ztwisted.internetr r ZTestCaser
r r r r �<module> s