o
    úLfû  ã                   @   s8   d dl mZ d dlmZ d dlmZ dd„ Zdd„ ZdS )	é    )Úapp_settings)ÚAuthenticator)ÚRecoveryCodesc                 C   sÆ   t  | ¡}| ¡ }tt|ƒƒtjksJ ‚ttjƒD ]	}| |¡r"J ‚qd}| 	|| ¡s.J ‚ttjƒD ]}| |¡||kks@J ‚q3| 	|| ¡rJJ ‚| 
¡ }|| |vsVJ ‚t|ƒtjd ksaJ ‚d S )Né   é   )r   ÚactivateÚgenerate_codesÚlenÚsetr   ÚRECOVERY_CODE_COUNTÚrangeÚ_is_code_usedÚvalidate_codeÚget_unused_codes)ÚuserÚrcÚcodesÚiÚidxÚunused_codes© r   úf/var/www/html/humari/django-venv/lib/python3.10/site-packages/allauth/mfa/tests/test_recovery_codes.pyÚ	test_flow   s   
r   c                 C   s¶   t |dddgid}t|ƒ}| ¡ ddgksJ ‚| ¡ ddgks"J ‚| d¡r)J ‚| d¡s0J ‚| ¡  t|ƒ}| ¡ dgksAJ ‚| ¡ dgksJJ ‚| d¡ |jjd g ksYJ ‚d S )NÚmigrated_codesÚabcÚdef)r   ÚdataÚbad)r   r   r   r   r   Úrefresh_from_dbÚinstancer   )Údbr   Úauthr   r   r   r   Útest_migrated_codes   s   
r"   N)Úallauth.mfar   Úallauth.mfa.modelsr   Úallauth.mfa.recovery_codesr   r   r"   r   r   r   r   Ú<module>   s
    