__ __ __ __ _____ _ _ _____ _ _ _ | \/ | \ \ / / | __ \ (_) | | / ____| | | | | | \ / |_ __\ V / | |__) | __ ___ ____ _| |_ ___ | (___ | |__ ___| | | | |\/| | '__|> < | ___/ '__| \ \ / / _` | __/ _ \ \___ \| '_ \ / _ \ | | | | | | |_ / . \ | | | | | |\ V / (_| | || __/ ____) | | | | __/ | | |_| |_|_(_)_/ \_\ |_| |_| |_| \_/ \__,_|\__\___| |_____/|_| |_|\___V 2.1 if you need WebShell for Seo everyday contact me on Telegram Telegram Address : @jackleetFor_More_Tools:
a
f�Wc� � @ sn d Z ddlZddlmZ ddlZe�e�ZddlZddl Z
ddl m Z ddl
mZmZ ddlmZmZ ddl mZ ddlmZmZmZ d gZe ZZ ej d
kr�dd l m Z ej d kr�dd l m Z d
Z dZ!dZ"dZ#dZ$dZ%dZ&dZ'ej(j)dks�J d��ej(j*dk�s
J d��dd� Z+e+� Z,dd � Z-G d d � d e�Z.d �/d �Z0e0d dd!� Z1e0d" dd#� Z2G d$d%� d%e�Z3dS )&z"passlib.tests -- test passlib.totp� N)�partial)�exc)�unicode�u)�TestCase� time_call)�totp)�TOTP� AppWallet� AES_SUPPORTZ
EngineTest)� r )�Error)r r Zabcdefs �Z4AOGGDBBQSYHNTUZs
� c !��vΙs �]�9�0 D�y/�4&��ZS3JDVB7QD2R7JPXX�JBSWY3DPEHPK3PXPs
Hello!ޭ��� zunexpected float_info.radix�, z#double precision unexpectedly smallc
C sj d} d}| d> }zt j �|d �j}W n tttfyB Y qTY n0 ||k rNqT|} q| d8 } d}t| |�S )z8
helper to calc max_time_t constant (see below)
i @r � l ��� )�datetime�utcfromtimestamp�year�
ValueError�OSError�
OverflowError�min)�valuer Z
next_valueZ next_yearZmax_datetime_timestamp� r �;/usr/lib/python3.9/site-packages/passlib/tests/test_totp.py�_get_max_time_t8 s
r c C s | d d d S )N� � � r )Zraw_sizer r r � to_b32_sizei s r c @ s\ e Zd ZdZdd� Zdd� Zdd� Zdd d
�Zd d � Zd
d� Z ddd�Z
dd� Z dd� Z dS )�
AppWalletTestzpasslib.totp.AppWalletc C s� t � }| �|ji � | �|j� ddd�}t |�}| �|j|� | �|j� t d�}| �|j|� t d�}| �|jddd�� t d�}| �|j|� | �tt d� | �tt d � | �tt d
d d�� d S )
z-constructor -- 'secrets' param -- input types� aaa� bbb��1�2z
1: aaa
# comment
2: bbb z"1: aaa: bbb
# comment
2: bbb s aaa: bbbz{"1":"aaa","2":"bbb"}�{ z[123]�aaa� N) r
� assertEqual�_secrets� assertFalseZ has_secrets�
assertTrue� assertRaises� TypeErrorr )�self�wallet�refr r r �test_secrets_typesv s
z AppWalletTest.test_secrets_typesc C s dddd�}t |�}| �|j|� t td�dtd�dtd�di�}| �|j|� t dddd��}| �|j|� | �tt d d
i� t d di�}| �tt d d
i� | �tt d
d
i� t td
�ddd��}| �|j|� | �tt ddi� | �tt ddi� | �tt dg i� dS )z9constructor -- 'secrets' param -- tag/value normalizationr" r# s ccc)r% �02�Cr% r4 r5 )r r4 r5 )r r( z1-2_3.4z-abczab*$Zbbbr' N)r
r* r+ r r. r/ r )r0 r2 r1 r r r �test_secrets_tags� s z AppWalletTest.test_secrets_tagsc C s� t ddd��}| �|jd� | �|�|j�d� t dddd��}| �|jd� | �|�|j�d � t dddd�d
d �}| �|jd
� | �|�|j�d � | jtt dddd�d
d � t � }| �|jd� | �t|jd� dS )z"constructor -- 'default_tag' paramZoneZtwo)r% r4 r4 s twor( )r% r4 �Ar7 r" r% )� default_tags one�BN)r
r* r8 Z
get_secretr. �KeyError)r0 r1 r r r �test_default_tag� s �z AppWalletTest.test_default_tagNc C s. t r|o |� n|o | �t|� | �d��d S )Nz$'cryptography' package not installed)r r. � RuntimeErrorZskipTest)r0 �canaryr r r �require_aes_support� s z!AppWalletTest.require_aes_supportc C s t ttd��}tdddddd�}| jt|j|�d� | �|�|�d t� tddd
d dd�}| �|�|�d t� tdd d
ddd�}| �|�|�d t � |�
� }|j dd� | �|�|�d d� |�
� }|j dd� | � t
|j|� |�
� }|j dd� | � t|j|� dS )z.decrypt_key()r$ r �
Z6D7N7W53O7HHS37NLUFQZMHCTEGSNPFN5CGBJr% ��v�c�s�k�t�r= r ZSPZJ54Y6IPUD2BYA4C6AZZGDXXTVQOWYLC2AUr ZFCCTARTIJWE7CPQHUDKAZ D2DRS32YESGHHINWFFCELKN7Z6NAHM4Mr& �rE s
�D6.F7�Q�3�� )rA N)r
�PASS1�PASS2�dictr> r � decrypt_keyr* �KEY1_RAW�KEY2_RAW�copy�updater. r: r )r0 r1 ZCIPHER1ZCIPHER2ZCIPHER3�tempr r r �test_decrypt_key� s. ��� z AppWalletTest.test_decrypt_keyc C s� | � � tttd�dd�}tdddddd�}| �|�|�d � |�� }|jdd � | � |�|�d � |�� }|jd
d � | � |�|�d � d S )
z$.decrypt_key() -- needs_recrypt flagr$ r? �� encrypt_costr ZAAAAr& r@ r )rB r% rG N)
r> r
rJ rK rL r, rM rP rQ r- )r0 r1 r2 rR r r r � test_decrypt_key_needs_recrypt s z,AppWalletTest.test_decrypt_key_needs_recryptr% Fc C s� | � t|�tg d��� | � |d d� | � |d |� | � |d |j� | � t|d �t|j�� | � t|d �tt|��� |�|�\}}| � ||� | � ||� dS ) z.check encrypt_key() result has expected format)rA rE rB rC rD rA r rE rB rC rD N)r* �setrU �lenr � salt_sizerM )r0 �resultr1 �key�tagZ
needs_recryptZ
result_keyZresult_needs_recryptr r r �assertSaneResult0 s z AppWalletTest.assertSaneResultc C s$ t dtidd�}| jt|jt�d� |�t�}| �||t� |�t�}| �||t� | �|d |d � | �|d |d � t dtidd�}|�t�}| �||t� t ttd��}|�t�}| j||td d
� t dti�}d |_ |�t�}| �||t� |�t
�}| �||t
� | � t |jd � d
S )z.encrypt_key()r% r rT rF rC rD � r$ r& )r\ �@ � N)
r
rJ r> r � encrypt_keyrN r] �assertNotEqualrK rY rO r. r )r0 r1 rZ �otherZwallet2r r r �test_encrypt_keyA s*
z AppWalletTest.test_encrypt_keyc C s� | � � tddi�}| jd8 _tt|jt�dd�\}}| jd7 _tt|jt�dd�\}}| j||d |d d d � d
S ) z verify cost parameter via timingr% r( r r )Zmaxtimer r � �?)�deltaN)r> r
rU r r ra rN �assertAlmostEqual)r0 r1 rf �_Zdelta2r r r �test_encrypt_cost_timingj s z&AppWalletTest.test_encrypt_cost_timing)N)r% F)
�__name__�
__module__� __qualname__�descriptionPrefixr3 r6 r; r> rS rV r] rd ri r r r r r! o s 2%
& �
)r! Z12345678901234567890�ascii� r r_ c sX e Zd ZdZdZ� fdd�Zdd� Zd�dd �Zd
d � Ze d d
ddd�Z
e ddd�dddddge ddd�dddd d ge d dd�d d ge e d!d
d"�d#d$d%d&d'd(ge e d!d)d"�d*d+d,d-d.d/ge e
d!d0d"�d1d2d3d4d5d6ge d7dd�d8d9ge d7d:d;d<�d=ge e d!d>d?�d@ge e d!d)d>dA�dBge e
d!d0d>dA�dCgg ZdDdE� ZdFdG� ZdHdI� ZdJdK� ZdLdM� ZdNdO� ZdPdQ� ZdRdS� ZdTdU� ZdVdW� ZdXdY� Zd�dZd[�Zd\d]� Zd^d_� Z d`da� Z dbdc� Z ddde� Z dfdg� Z d�djdk�Z!dldm� Z"dndo� Z#dpdq� Z$drds� Z%d�dtdu�Z&d�dvdw�Z'dxdy� Z(dzd{� Z)d|d}� Z*d~d� Z+d�d�� Z,d�d�� Z-d�d�� Z.d�d�� Z/d�d�� Z0d�d�� Z1d�d�� Z2� Z3S )��TotpTestz3
common code shared by TotpTest & HotpTest
zpasslib.totp.TOTPc s8 t t| ��� ddlm} |�� | �td| �� � d S )Nr )� lookup_hash�rng) �superrp �setUpZpasslib.crypto.digestrq � clear_cacheZ patchAttr� totp_module� getRandom)r0 rq �� __class__r r rt � s zTotpTest.setUpc C s | � � �� t S )zY
helper to generate random epoch time
:returns float: epoch time
)rw Zrandom�
max_time_t�r0 r r r �randtime� s zTotpTest.randtimeNc K sh | � � }d|vrd|d< |�d|�dd�� |�d|�g d��� |�d |�dd
�� |p\tf i |��S ) z@
helper which generates a random TOTP instance.
r[ T�new�digitsr^ �
�alg)�sha1�sha256�sha512�period�x )rw �
setdefault�randint�choicer )r0 �cls�kwdsrr r r r �randotp� s zTotpTest.randotpc C s� | � � }| � � }| �|j|jd� td�D ] }|j|jkr> qR| � � }q*| �d� td�D ] }|j|jkrn q�| � � }qZ| �d� dS )z,
internal test -- randotp()
zkey not randomized:r zdigits not randomizedzalg not randomizedN)r� rb r[ �ranger~ Zfailr� )r0 �otp1�otp2rh r r r � test_randotp� s
zTotpTest.test_randotpZbase32r� � r )�formatr� r� r~ ZACDEFGHJKL234567r^ )r[ r~ )鷼6TZ221105)鸼6T�178491)鹼6Tr� )�ռ6Tr� )�ּ6TZ915114)r� Z20221105)r� �86178491)r� r� )r� r� )r� Z03915114zS3JD-VB7Q-D2R7-JPXX)i5��TZ000492)�S��T�897212�raw)r[ r� r� )�; Z94287082)��5:BZ07081804)��5:BZ14050471)���IZ89005924)� �5wZ69279037)� H/P Z65353130r� )r� Z46119246)r� Z68084774)r� Z67062674)r� Z91819424)r� Z90698825)r� Z77737706r� )r� Z90693936)r� Z25091201)r� Z99943326)r� Z93441116)r� Z38618901)r� Z47863826r )i�SZ727248)i.ѡTZ122419� �) )r[ r~ r� )i�ѡTZ 662331049�< )r[ r� r� )r� Z19360094)r[ r� r� r� )r� Z40857319)r� Z37023009c
c s� ddl m} | jD ]�}| j�� }|�|d � |dd� D ]h}t|�dkrV|\}}}n |\}}d}t�d||||� |f i |��}d|j ||f } ||||| fV q:qdS )zv
helper to iterate over test vectors.
yields ``(totp, time, token, expires, prefix)`` tuples.
r �r r Nr z+test vector: %r time=%r token=%r expires=%rzalg=%r time=%r token=%r: )
� passlib.totpr �vectors�vector_defaultsrP rQ rX �log�debugr� )
r0 r �rowr� �entry�time�token�expires�otp�prefixr r r �iter_test_vectors. s
zTotpTest.iter_test_vectorsc C sF | � tt� | j ttddd� tdd�}tdd�}| �|j|j� dS )z constructor -- 'new' parameterZ4aoggdbbqsyhntuzT)r[ r} �r} N)r. r/ r rb r[ )r0 r� r� r r r �test_ctor_w_newG s
zTotpTest.test_ctor_w_newc C s� | � ttddd�j�d� | � ttddd�j�d� | � ttddd�j�d� | � ttdd d
�j�d � | � ttdd d
�j�d � | jttdd dd
� | jttddd
� | �ttj dd�g�� tdd�}W d � n1 s�0 Y dS )z constructor -- 'size' parameterTr� )r} r� � r� ro r� r_ r �r} �size� � )r} r� r� r� z-.*for security purposes, secret key must be.*��categoryZ
message_reZ0A0A0A0A0A0A0A0A0A�hexN)
r* rX r r[ r. r �assertWarningListrL r ZPasslibSecurityWarning)r0 rh r r r �test_ctor_w_sizeS s �zTotpTest.test_ctor_w_sizec C s� | � tt�jt� | � tt�� �jt� | � td�jt� | �ttd� | � tdd�jt� | �ttdd� | � ttd�jt� dS )z,constructor -- 'key' and 'format' parametersz 4aog gdbb qsyh ntuz zao!ggdbbqsyhntuz�e01c630c2184b076ce99r� ZX01c630c2184b076ce99r� N) r* r �KEY1r[ rN �lowerr. �Base32DecodeError�Base16DecodeErrorr{ r r r �test_ctor_w_key_and_formatk s z#TotpTest.test_ctor_w_key_and_formatc C sB | � ttdd�jd� | � ttdd�jd� | jtttdd� dS )z constructor -- 'alg' parameterzSHA-256�r� r� ZSHA256zSHA-333N)r* r r� r� r. r r{ r r r �test_ctor_w_alg� s zTotpTest.test_ctor_w_algc C sT | j tttdd� | �ttdd�jd� | �ttdd�jd� | j tttdd� dS )z!constructor -- 'digits' parameterr �r~ r^ r � N)r. r r r� r* r~ r{ r r r �test_ctor_w_digits� s zTotpTest.test_ctor_w_digitsc C st | � tt�jd� | � ttdd�jd� | jtttdd� | jtttdd� | jtttdd� | jtttdd� dS ) z!constructor -- 'period' parameterr� �? �r� � �?�abcr ���N)r* r r� r� r. r/ r r{ r r r �test_ctor_w_period� s zTotpTest.test_ctor_w_periodc C s> | � tt�jd� | � ttdd�jd� | jtttdd� dS )z constructor -- 'label' parameterNzfoo@bar)�labelzfoo:bar)r* r r� r� r. r r{ r r r �test_ctor_w_label� s zTotpTest.test_ctor_w_labelc C s> | � tt�jd� | � ttdd�jd� | jtttdd� dS )z!constructor -- 'issuer' parameterNzfoo.com��issuerz foo.com:bar)r* r r� r� r. r r{ r r r �test_ctor_w_issuer� s zTotpTest.test_ctor_w_issuerc C sx | � tt�jd� | � tjdd�t�jd� | jttjdd� | jttjdd� | jttjdd� | jttjdd� dS ) z using() -- 'period' parameterr� r� r� r� r� r r� N)r* r r� r� �usingr. r/ r r{ r r r �test_using_w_period� s z TotpTest.test_using_w_periodc s� | � � }| �|jtj� | �|�d�tt�� �� dg� � fdd�}| j tj |d�d�}| �
|�d�d� | �
|�d�d� | j t tj d d� d
}| j
t|tj d d � d� | j
t|tj d
d � d� dS )zusing -- 'now' parameterNgH�z��^@c s � d d7 < � d S )Nr r r r ��counterr r �now� s z&TotpTest.test_using_w_now.<locals>.now�r� )r� �~ � r' z)now\(\) function must return non-negativec S s dS )Nr� r r r r r �<lambda>� r` z+TotpTest.test_using_w_now.<locals>.<lambda>c S s dS )Nr� r r r r r r� � r` )r� �assertIsr� �_timer� rg �normalize_time�intr r� r* r. r/ ZassertRaisesRegex�AssertionError)r0 r� r� Zmsg_rer r� r �test_using_w_now� s zTotpTest.test_using_w_nowc C s� |du r| j dd�}| �|�td��d� | �|�d�d� | �|�d�d� | �|�d�d� | �t|jd � | �t|jd� | �tj|jd
� | �tj|jd � | �tj|jd � dS )
z$normalize_token() -- instance methodN� r� Z1234567s 1234567i�� iG� Z0234567g ��2AZ123456Z01234567iNa� )r� r* Znormalize_tokenr r. r/ r �MalformedTokenError�r0 r� r r r � test_normalize_token_instance� s z&TotpTest.test_normalize_token_instancec C s | j tjdd�d� dS )z!normalize_token() -- class methodr� r� )r� N)r� r r� r{ r r r �test_normalize_token_class s z#TotpTest.test_normalize_token_classc s� t �� }| �|�}td�D ]�}| �� � t� �}| �|�� �|� | �|�|d �|� | �|�|�|� tj� � �}| �|�|�|� |j
}z.t � fdd��|_
| �|�d�|� W ||_
q||_
0 q| � t
|jd� dS )znormalize_time()r re c s � S �Nr r �r� r r r� r` z.TotpTest.test_normalize_time.<locals>.<lambda>NZ1234)r r� r� r� r| r� r* r� r r r� � staticmethodr. r/ )r0 � TotpFactoryr� rh Ztint�dtZorigr r� r �test_normalize_time s
z TotpTest.test_normalize_timec C s� | � � }ttd�}| �|jt� | �|jd� | �|jt� | �|�� d� | �|jdd�d� | �|jdd�t� | �|jdd �d
� td |� d d
�d�}|j}|j}|�� }dS )z pretty_key() and .key attributesr� r� z4AOG-GDBB-QSYH-NTUZ� )�sepz4AOG GDBB QSYH NTUZFr� )r� ze01c-630c-2184-b076-ce99Tr r� r� N)
rw r rN r* r[ Zhex_key�
base32_keyr� Z
pretty_keyr� )r0 rr r� rh r r r �test_key_attrs& s
zTotpTest.test_key_attrsc C sd ddl m}m} |d�}|�d�}| �||� | �|jd� | �|jd� | �|jd� | �|d� | �t |�d � | �|d d� | �|d
d� | �
t |j d � | �
t |j d � | �
|� d d
� |_| �|jd� | �
|j� dd
� |_| �|jd� | �|j� |�d�}| �||� | �||� |�d�}| �||� | �||� |�d�}| �||� dS )z generate() -- TotpToken() classr )r � TotpTokenZs3jdvb7qd2r7jpxxr� r� i���T��T)r� r� r r ���c S s dS )Ng �Tn'�Ar r r r r r� T r` z*TotpTest.test_totp_token.<locals>.<lambda>re c S s dS )NiU��Tr r r r r r� X r` i7��TiW��TN)r� r r� �generate�assertIsInstancer* r� r� � expire_timerX r. �
IndexError� __getitem__r- r� Z remainingZvalidr, � assertIsNotrb )r0 r r� r� rZ Zresult2Zresult3Zresult4r r r �test_totp_token? s8
zTotpTest.test_totp_tokenc s� ddl m} |dd�}| �� � |�� �}|j}| �|t� |jd }| �|�|d �j|� | � |�|d �j|� t
j
� � �}| �t |�
|��t � �� | �|�|�j|� |j� fdd�d �|jd
�}| �|�� j|� | �t|jd � d S )
z
generate()r r� Tr� r� � c s � S r� r r r� r r r� � r` z(TotpTest.test_generate.<locals>.<lambda>r� )r[ r� N)r� r r| r� r� r� r r� r* rb r r r� r� r� r� r. r )r0 r r� rZ r� Z
start_timer� r� r r� r �
test_generatej s
zTotpTest.test_generatec C sb | � � D ]T\}}}}}|�|�}| j|j||d� | j|j||j |d� |r| �|j|� qdS )z generate() -- reference vectors��msgN)r� r� r* r� r� r� r� )r0 r� r� r� r� r� rZ r r r �!test_generate_w_reference_vectors� s
z*TotpTest.test_generate_w_reference_vectorsr r) c C sJ ddl m} | �||� | �|jt� | �|jj|� | j|j||d d� || }|| } | j|j| |d d� | j|j ||d d� | j|j
||d d� | �|j || � | d | }
| �|j |
� | �|j
|
| � | �t|�d � | �|| |f� | �t|jd
� | �|d | � | �|d |� | �t|jd � | �|� d S ) Nr �� TotpMatchz matched time:r� z matched counter:z expected counter:z skipped:r r r� )r� r r� r r r* r� r� r� Zexpected_counter�skippedZ
cache_secondsr� Z
cache_timerX r. r� r� r- ) r0 �matchr� r r� �windowr r Zexpectedr� r� r r r �assertTotpMatch� s* zTotpTest.assertTotpMatchc s@ d� d}t j� fdd�d�t�}|�|� �}| j|� dd� dS ) z!match() -- valid TotpMatch object�k�781501c s � d S �Ni�Q r r r� r r r� � r` z8TotpTest.test_totp_match_w_valid_token.<locals>.<lambda>r� r �r� r N�r r� �KEY3r r �r0 r� r� rZ r r� r � test_totp_match_w_valid_token� s
z&TotpTest.test_totp_match_w_valid_tokenc sT ddl m} d� d}tj� fdd�d�t�}|�|� d �}| j|� d d d
� d S ) z3match() -- valid TotpMatch object with future tokenr r r r c s � d S r
r r r� r r r� � r` z8TotpTest.test_totp_match_w_older_token.<locals>.<lambda>r� r� r r N)r� r r r� r
r r )r0 r r� r� rZ r r� r � test_totp_match_w_older_token� s z&TotpTest.test_totp_match_w_older_tokenc sH d� d}t j� fdd�d�t�}|�|� d �}| j|� d dd� d S )
z1match() -- valid TotpMatch object with past tokenr r c s � d S r
r r r� r r r� � r` z6TotpTest.test_totp_match_w_new_token.<locals>.<lambda>r� r� r� r Nr r r r� r �test_totp_match_w_new_token� s
z$TotpTest.test_totp_match_w_new_tokenc s< d� d}t j� fdd�d�t�}| �tj|j|� d � dS )z#match() -- invalid TotpMatch objectr r c s � d S r
r r r� r r r� � r` z:TotpTest.test_totp_match_w_invalid_token.<locals>.<lambda>r� r� N)r r� r
r. r �InvalidTokenErrorr )r0 r� r� r r� r � test_totp_match_w_invalid_token� s z(TotpTest.test_totp_match_w_invalid_tokenc K sX d|j |j|j|||f }|j||fi |��}| j||�|�|j|�dd�||d� dS )z,helper to test otp.match() output is correct�5key=%r alg=%r period=%r token=%r gen_time=%r time=%r:r r� )r� r� r r r N)r� r� r� r r r� �get) r0 Zexpect_skippedr� r� r� �gen_timer� r rZ r r r �assertVerifyMatches� s �
�z TotpTest.assertVerifyMatchesc K s8 d|j |j|j|||f }| j||j||fd|i|��S )z/helper to test otp.match() throws correct errorr Z__msg__)r� r� r� r. r )r0 Z exc_classr� r� r� r r� r r r r �assertVerifyRaises� s ���zTotpTest.assertVerifyRaisesc C sb | � � }|j}| �� }|�|�j}t||d�}t| jfi |��}t| jfi |��}|t j
||| dd� |d||| |d� |d||| d| d� |d||dd� |t j
||| dd� |d||| |d� |d||| d| d� |t j
||d| dd� |t j
||d| |d� |d||d| d| d� t j � |�}|d||dd� |t
|d� dS ) z)match() -- 'time' and 'window' parameters�r� r r �r r r r� ���N)r� r� r| r� r� rL r r r r r r r r ) r0 r� r� r� r� �common�
assertMatchesr. r� r r r �test_match_w_window� s(
z TotpTest.test_match_w_windowc
C s� | � � }|j}| �� }t||d�}t| jfi |��}t| jfi |��}d| }|�|| �j}|t j
||dd� |d||d| d� |�|| �j} |t j
| |dd� |d| |d|d� dS )z match() -- 'skew' parametersr r r r r� )r �skewN) r� r� r| rL r r r r� r� r r )
r0 r� r� r� r r r. r Z behind_tokenZ ahead_tokenr r r �test_match_w_skew& s zTotpTest.test_match_w_skewc C s� | � � }|j}| �� }|�|�}|j}|j}|j}t||d�}t| j fi |��} t| j
fi |��}
| d||| |d� | d||| |d |d� |
t j ||d| ||d� |
t j
||| ||d�} | �| j|� |
t j
|||dd�} | �| j|� dS ) z0match() -- 'reuse' and 'last_counter' parametersr r� r r )Z last_counterr r r N)r� r� r| r� r� r� r� rL r r r r r ZUsedTokenErrorr* ) r0 r� r� r� Ztdatar� r� r� r r r. �errr r r �test_match_w_reuse= s2
���
�zTotpTest.test_match_w_reusec C sl t d�}|j}d}| �|d|�� | �|d|�� | �tj|d|� | �tj|d|� | �tj|d|� dS ) z match() -- token normalization�otxl2f5cctbprpzx��6Tz 3 32-136 s 332136Z12345Z12345XZ0123456N)r r r- r. r r� )r0 r� r r� r r r � test_match_w_token_normalizationf s z)TotpTest.test_match_w_token_normalizationc C sh | � � D ]Z\}}}}}|j}|||�}| �|� | j|j||j |d� | jtj|||d dd� qdS )z match() -- reference vectorsr� �d r r N) r� r r- r* r� r� r. r r )r0 r� r� r� r� r r rZ r r r � test_match_w_reference_vectors| s
z'TotpTest.test_match_w_reference_vectorsc s� ddl m} d� |j� fdd�d�}tddd d
�}|�d |�}| j|� d � tddd d
�}| �tj|jd
|� tddd�}| �t |jd
|� d}|�d |�}| j|� d � d}|�d |�}| j|� d � dS )zverify()r r� r$ c s � S r� r r r� r r r� � r` z&TotpTest.test_verify.<locals>.<lambda>r� r r r# �rA �typer[ Z332136r� Z332155�rA r) z3{"v": 1, "type": "totp", "key": "otxl2f5cctbprpzx"}z,otpauth://totp/Label?secret=otxl2f5cctbprpzxN)
r� r r� rL Zverifyr r. r r r )r0 r r� Zsource1r Z source1jsonZ
source1urir r� r � test_verify� s zTotpTest.test_verifyc C s ddl m} |j}|td��}| �|jt� |d�}| �|jt� |tddtd��}| �|jt� |td��}| �|jt� |d �}| �|jt� | � ||�|� t
� }|j |d
��|�}| � ||� | �|�
� |�
� � |j |d
��|�}| � ||� | �t|td �� | �t|d � d
S )z
from_source()r r� �Notpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&issuer=ExamplesN otpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&issuer=Exampler r r( z3{"v": 1, "type": "totp", "key": "JBSWY3DPEHPK3PXP"}s3 {"v": 1, "type": "totp", "key": "JBSWY3DPEHPK3PXP"})r1 Zfoos fooN)r� r � from_sourcer r* r[ �KEY4_RAWrL �KEY4r� r
r� r� �to_dictr. r )r0 r r- r� Zwallet1r� r� r r r �test_from_source� s* zTotpTest.test_from_sourcec C s ddl m} |j}|d�}| �||� | �|jt� | �|jd� | �|jd� | �|j d� | �|j
d� | �|j d� |d �}| �|jt� | � t
|d
� | � t|d � |d �}| �|jd
� | �|jd� |d�}| �|jd� | �|jd� |d�}| �|jd� | �|jd� | � t
|jd� |d�}| �|j d� | � t
|d� |d�}| �|j d� | � t
|d� | � t
|d� | � t
|d� |d �}| �|j
d � | � t
|d � | � t
|d � | �ttjd d!�g�� |d"�}W d# � n1 �s�0 Y | �|jt� | �|j
d � d#S )$z
from_uri()r r� r, �alice@google.com�Exampler� r� r^ zNotpauth://totp/Example:alice@google.com?secret=jbswy3dpehpk3pxp&issuer=Examplez0otpauth://totp/Example:alice@google.com?digits=6z?otpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHP@3PXPzOotpauth://totp/Provider1:Alice%20Smith?secret=JBSWY3DPEHPK3PXP&issuer=Provider1� Alice Smith� Provider1zMotpauth://totp/Big%20Corporation%3A%20alice@bigco.com?secret=JBSWY3DPEHPK3PXPzalice@bigco.comzBig CorporationzOotpauth://totp/alice@bigco.com?secret=JBSWY3DPEHPK3PXP&issuer=Big%20CorporationzGotpauth://totp/Provider1:alice?secret=JBSWY3DPEHPK3PXP&issuer=Provider2zPotpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&algorithm=SHA256r� zPotpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&algorithm=SHA333zHotpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&digits=8r zHotpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&digits=AzJotpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&digits=%20zIotpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&digits=15zIotpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&period=63r� zHotpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&period=0zIotpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&period=-1z!unexpected parameters encounteredr� zQotpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&foo=bar&period=63N)r� r �from_urir� r* r[ r. r� r� r� r� r~ r. r r� r� rL r ZPasslibRuntimeWarningr� r/ )r0 r r6 r� r r r �
test_from_uri� sX
� �(zTotpTest.test_from_uric C s� t tdddd�}| �|�dd�d� | �t|jdd� | �|�d�d � d|_| �|�� d � d|_| �|�� d� | �t|jd
� | �t|jdd � | �t td d
��d�d� | �t tdd��d�d� | �t tdd��d�d� dS )zto_uri()r� r^ r� �r� r~ r� r2 � Example OrgzZotpauth://totp/Example%20Org:alice@google.com?secret=JBSWY3DPEHPK3PXP&issuer=Example%20OrgNz7otpauth://totp/alice@google.com?secret=JBSWY3DPEHPK3PXPzlabel:with:semicolonszissuer:with:semicolonsr� r� zHotpauth://totp/alice@google.com?secret=JBSWY3DPEHPK3PXP&algorithm=SHA256r r� z@otpauth://totp/alice@google.com?secret=JBSWY3DPEHPK3PXP&digits=8r� r� zAotpauth://totp/alice@google.com?secret=JBSWY3DPEHPK3PXP&period=63)r r/ r* Zto_urir. r r� r� r� r r r � test_to_uriW s6 � �
�
����zTotpTest.test_to_uric
C sr ddl m} |j}|tddtddd��}| �||� | �|jt� | �|j d� | �|j
d� | �|j d� | �|j d � | �|j
d
� | �t|tdtd �� | �t|tddtd �� | �t|td
dtd �� | �t|tdtd�� |tddt�� ddd��}| �|jt� | �t|tddd�� | �t|tdddd �� |tddtddd��}| �|j d� | �|j
d� |tddtdd��}| �|j d� | �t|tddtdd�� |tddtdd��}| �|j
d� | �t|tddtdd�� | �t|tddtdd�� |tddtdd��}| �|j d� | �t|tddtdd�� | �t|tddtd d�� | �t|tddtd d �� d S ) z from_dict()r r� r r r2 r3 �rA r) r[ r� r� r� r� r^ )r) r[ r( rI )rA r[ r* zJBSWY3DPEHP@3PXPr4 r5 r� �rA r) r[ r� Zsha333r �rA r) r[ r~ r7 � r� �rA r) r[ r� r� r' )rA r) r[ �INVALIDN)r� r � from_dictrL r/ r� r* r[ r. r� r� r� r� r~ r. r r� r� r/ )r0 r rA r� r r r �test_from_dict� sF �zTotpTest.test_from_dictc
C s| t tdddd�}| �|�� tddtd�� t tddddd d
�}| �|�� tddtdd d �� t tddddd �}| �|�� tddtdd
�� t tdddd d�}| �|�� tddtd d�� t jd d�}|t�}| �|�� tddtd�� |td d�}| �|�� tddtd�� | �t tdd��� tddtdd�� | �t tdd��� tddtdd�� | �t tdd��� tddtdd�� dS )z to_dict()r� r^ r� r8 r r r( r2 r9 )r� r~ r� r� r� r; )r� r~ r� r� )rA r) r[ r� )r� r~ r� r� )rA r) r[ r� r� r� r� r< r r� r= r� r� r? N)r r/ r* r0 rL r� )r0 r� r� r r r � test_to_dict� sP
�
��
�
��
�
�� ���zTotpTest.test_to_dict)N)N)r r� r� r) )N)N)4rj rk rl �__doc__rm rt r| r� r� rL r� �RFC_KEY_BYTES_20�RFC_KEY_BYTES_32�RFC_KEY_BYTES_64r� 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+ r1 r7 r: rB rC �
__classcell__r r rx r rp � s�
�
�
�
�
�
��T
"
+
!
�
�
0)$/w=^rp )4rD r � functoolsr ZloggingZ getLoggerrj r� �sysr� r� Zpasslibr Zpasslib.utils.compatr r Zpasslib.tests.utilsr r r rv r� r r
r �__all__r/ r� r� � version_infoZbinasciir
rJ rK r� rN rO r
r/ r. �
float_info�radix�mant_digr rz r r! �encoderE rF rG rp r r r r �<module> sJ �
/