관리-도구
편집 파일: test_totp.cpython-39.opt-1.pyc
a f�Wc� � @ sD 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'dd� Z(e(� Z)dd� Z*G dd� de�Z+d�,d �Z-e-d dd � Z.e-d dd � Z/G d!d"� d"e�Z0dS )#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!ޭ��c 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# r2 r3 )r r2 r3 )r r& z1-2_3.4z-abczab*$Zbbbr% N)r r( r) r r, r- r )r. r0 r/ 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# r2 r2 s twor& )r# r2 �Ar5 r r# )� default_tags one�BN)r r( r6 Z get_secretr, �KeyError)r. r/ 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)r. �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$ �rC s �D6.F7�Q�3�� )r? N)r �PASS1�PASS2�dictr< r � decrypt_keyr( �KEY1_RAW�KEY2_RAW�copy�updater, r8 r )r. r/ 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 )r@ r# rE N) r< r rH rI rJ r* rK rN rO r+ )r. r/ r0 rP 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)r? rC r@ rA rB r? r rC r@ rA rB N)r( �setrS �lenr � salt_sizerK )r. �resultr/ �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 rR rD rA rB � r" r$ )rZ �@ � N) r rH r< r � encrypt_keyrL r[ �assertNotEqualrI rW rM r, r )r. r/ rX �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 )Zmaxtimer r � �?)�deltaN)r< r rS r r r_ rL �assertAlmostEqual)r. r/ re �_Zdelta2r r r �test_encrypt_cost_timingj s z&AppWalletTest.test_encrypt_cost_timing)N)r# F) �__name__� __module__� __qualname__�descriptionPrefixr1 r4 r9 r< rQ rT r[ rb rh r r r r r o s 2% & � )r Z12345678901234567890�asciirc � 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) �superro �setUpZpasslib.crypto.digestrp � clear_cacheZ patchAttr� totp_module� getRandom)r. rp �� __class__r r rs � s zTotpTest.setUpc C s | � � �� t S )zY helper to generate random epoch time :returns float: epoch time )rv Zrandom� max_time_t�r. 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. rY T�new�digitsr\ � �alg)�sha1�sha256�sha512�period�x )rv � setdefault�randint�choicer )r. �cls�kwdsrq 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� r` rY �ranger} Zfailr )r. �otp1�otp2rg r r r � test_randotp� s zTotpTest.test_randotpZbase32r� � r )�formatr r� r} ZACDEFGHJKL234567r\ )rY 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)rY 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� �) )rY r} r� )i�ѡTZ 662331049�< )rY r� r� )r� Z19360094)rY 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_defaultsrN rO rV �log�debugr ) r. 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)rY r| �r| N)r, r- r r` rY )r. 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� rn r� r] r~ �r| �size� � )r| r� r r� z-.*for security purposes, secret key must be.*��categoryZ message_reZ0A0A0A0A0A0A0A0A0A�hexN) r( rV r rY r, r �assertWarningListrJ r ZPasslibSecurityWarning)r. rg 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 �KEY1rY rL �lowerr, �Base32DecodeError�Base16DecodeErrorrz 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 rz 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} rz 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 rz 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 rz 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 rz 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 rz 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� rf �normalize_time�intr r� r( r, r- ZassertRaisesRegex�AssertionError)r. 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�r. 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� rz 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~ rd 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- )r. � TotpFactoryr� rg 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) rv r rL r( rY Zhex_key� base32_keyr� Z pretty_keyr� )r. rq r� rg 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� rc r ���c S s dS )Ng �Tn'�Ar r r r r r� T r^ z*TotpTest.test_totp_token.<locals>.<lambda>rd c S s dS )NiU��Tr r r r r r� X r^ i7��TiW��TN)r� r r� �generate�assertIsInstancer( r� r� � expire_timerV r, � IndexError� __getitem__r+ r� Z remainingZvalidr* � assertIsNotr` )r. r r� r� rX 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� )rY r� N)r� r r{ r� r� r� r r� r( r` r r r� r� r� r� r, r )r. r r� rX 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� )r. r� r� r� r� r� rX 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 rc r� )r� r r� r r r( r� r� r� Zexpected_counter�skippedZ cache_secondsr� Z cache_timerV r, r� r� r+ ) r. �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 �r. r� r� rX 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 )r. r r� r� rX 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 )r. 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) r. Zexpect_skippedr� r� r� �gen_timer� r� rX 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 )r. 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 rc r� ���N)r� r� r{ r� r� rJ r r r r r r r r ) r. 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{ rJ r r r r� r� r r ) r. 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 rc r N)r� r� r{ r� r� r� r� rJ r r r r r ZUsedTokenErrorr( ) r. 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� )r. 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 )r. r� r� r� r� r� r rX 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" �r? �typerY Z332136r� Z332155�r? r( z3{"v": 1, "type": "totp", "key": "otxl2f5cctbprpzx"}z,otpauth://totp/Label?secret=otxl2f5cctbprpzxN) r� r r� rJ Zverifyr r, r r r )r. 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"})r/ Zfoos fooN)r� r � from_sourcer r( rY �KEY4_RAWrJ �KEY4r� r r� r� �to_dictr, r )r. 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( rY r- r� r� r r� r} r, r r� r� rJ r ZPasslibRuntimeWarningr� r. )r. r r5 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� r1 � 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 r1 r2 �r? r( rY r� r� r� r� r\ )r( rY r' rG )r? rY r) zJBSWY3DPEHP@3PXPr3 r4 r� �r? r( rY r Zsha333r �r? r( rY r} r5 � r� �r? r( rY r� r� r% )r? r( rY �INVALIDN)r� r � from_dictrJ r. r� r( rY r- r� r� r r� r} r, r r� r� r- )r. r r@ 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� r7 r r r' r1 r8 )r r} r� r� r� r: )r r} r� r� )r? r( rY r� )r r} r� r� )r? r( rY r� r� r� r� r; r r� r< r� r� r> N)r r. r( r/ rJ r� )r. r� r� r r r � test_to_dict� sP � �� � �� � �� ���zTotpTest.test_to_dict)N)N)r r� r� r' )N)N)4ri rj rk �__doc__rl rs r{ r� r� rJ 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* r0 r6 r9 rA rB � __classcell__r r rw r ro � s� � � � � � ��T " + ! � � 0)$/w=^ro )1rC r � functoolsr ZloggingZ getLoggerri r� �sysr� r� Zpasslibr Zpasslib.utils.compatr r Zpasslib.tests.utilsr r r ru r� r r r �__all__r- r� r� � version_infoZbinasciir rH rI r� rL rM r r. r- r ry r r �encoderD rE rF ro r r r r �<module> sF � /