a
    I_P                     @   s4   d dl Z d dlZd dl mZ G dd de jjZdS )    N)netbiosc                       sH   e Zd Z fddZ fddZdd Zdd Zd	d
 Zdd Z  Z	S )NetBiosTestsc                    s4   t t|   t | _tjd | _tjd | _	d S )NZ	SERVER_IPZDC_NETBIOSNAME)
superr   setUpr   ZNodenosenvironifcdcself	__class__ 5/usr/lib/python3/dist-packages/samba/tests/netbios.pyr      s    
zNetBiosTests.setUpc                    s   t t|   d S )N)r   r   tearDownr   r   r   r   r      s    zNetBiosTests.tearDownc                 C   sR   | j j| j| jdd\}}}|| jks*J |d | jks<J |d | jksNJ d S N   timeoutr   )r   Z
query_namer
   r	   )r   
reply_fromnamesZ	addressesr   r   r   test_query_name!   s    zNetBiosTests.test_query_namec                 C   sP   | j j| j| jdd\}}}|| jks*J |d | jks<J t|dksLJ d S r   )r   Zname_statusr
   r	   len)r   r   nameZ	name_listr   r   r   test_name_status'   s    zNetBiosTests.test_name_statusc                 C   sf   d}| j j| jdf|| jddd\}}}}|| jks8J |d | jksJJ ||ksVJ |dksbJ d S )N	127.0.0.3    Fr   )Zmulti_homedr   r      )r   Zregister_namer
   r	   )r   addressr   r   Zreply_addresscoder   r   r   test_register_name-   s    &zNetBiosTests.test_register_namec                 C   s$   d}| j j| jdf|| jdd}d S )Nr   r   
   r   )r   Zrefresh_namer
   r	   )r   r   resr   r   r   disabled_test_refresh5   s    z"NetBiosTests.disabled_test_refresh)
__name__
__module____qualname__r   r   r   r   r!   r$   __classcell__r   r   r   r   r      s   r   )Zsambar   r   ZtestsZTestCaser   r   r   r   r   <module>   s   