U
    òccö  ã                   @   sp   d dl Z d dlmZ d dlmZ d dlmZ zd dl mZ W n ek
rX   d dlZY nX G dd„ de jƒZdS )é    N)Úapi)Úapi_communicator)Úbase_api)Úmockc                   @   s,   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	S )
ÚTestLoginRouterOsApic                 C   s<   t  ¡ }t |¡}t jddgd|_t |¡}t 	|¡}|S )Ns   !dones   .tag=1)Zreturn_value)
r   Z	MagicMockr   Ú
ConnectionZMockZreceive_sentencer   ZApiCommunicatorr   ZRouterOsApi)ÚselfÚsocketÚbaseZcommunicatorÚrouteros_api© r   úX/var/www/html/project/venv/lib/python3.8/site-packages/tests/test_login_router_os_api.pyÚget_api   s    


zTestLoginRouterOsApi.get_apic                 C   s   |   ¡ }|jdddd d S )NÚadminÚpassword123F©Zplaintext_login©r   Úlogin©r   r   r   r   r   Ú
test_login   s    zTestLoginRouterOsApi.test_loginc                 C   s   |   ¡ }|jdddd d S )Nr   r   Tr   r   r   r   r   r   Útest_plain_text_login   s    z*TestLoginRouterOsApi.test_plain_text_loginc                 C   s   |   ¡ }|jdddd d S )Ns   admins   password123Tr   r   r   r   r   r   Ú test_plain_text_login_with_bytes   s    z5TestLoginRouterOsApi.test_plain_text_login_with_bytesN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r   r   r      s   r   )	Zunittestr   r   r   r   r   ÚImportErrorZTestCaser   r   r   r   r   Ú<module>   s   