관리-도구
편집 파일: test_totp.cpython-39.pyc
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 � /