
    b	                     p    d dl mZ d dlZd dlZd dlmZ d dlmZ d dlmZm	Z	 d dl
Z
	  G d de          ZdS )    )TestCaseN)Credentials)netlogon)NTSTATUSErrorntstatusc                   4     e Zd Z fdZ fdZd Zd Z xZS )NetlogonServiceTestsc                    t          t          |                                            t          j        d         | _        |                                 | _        t                      | _	        dt          j        v rcdt          j        v rU| j	        
                    t          j        d                    | j	                            t          j        d                    nT| j	        
                    t          j        d                    | j	                            t          j        d                    | j	                            | j                   d S )NSERVERDC_USERNAMEDC_PASSWORDUSERNAMEPASSWORD)superr	   setUposenvironserverget_loadparmlpr   credsset_usernameset_passwordguessself	__class__s    9/usr/lib/python3/dist-packages/samba/tests/netlogonsvc.pyr   zNetlogonServiceTests.setUp"   s    "D))//111:h/,,.. ]]
 BJ&&=BJ+F+FJ##BJ}$=>>>J##BJ}$=>>>>J##BJz$:;;;J##BJz$:;;;
!!!!!    c                 V    t          t          |                                            d S )N)r   r	   tearDownr   s    r   r!   zNetlogonServiceTests.tearDown3   s&    "D))2244444r   c                     	 |                                  }|                     |           d S # t          $ rV}t          j        |j        d                   j        }|t          j        k    r| 	                    d           n Y d }~d S d }~ww xY w)Nr   znetlogon service is not running)
get_netlogon_connectionassertIsNotNoner   ctypesc_uint32argsvaluer   NT_STATUS_OBJECT_NAME_NOT_FOUNDfail)r   ceenums       r   test_have_netlogon_connectionz2NetlogonServiceTests.test_have_netlogon_connection6   s    
	,,..A  ##### 	 	 	 ?16!9--3Dx???		;<<<< =<<<<<	s   )- 
BABBc                 R    t          j         d| j        z  | j        | j                  S )Nzncacn_np:%s[seal])r   r   r   r   )r   s    r   r#   z,NetlogonServiceTests.get_netlogon_connectionD   s*     !4t{!BDG!%- - 	-r   )__name__
__module____qualname__r   r!   r.   r#   __classcell__)r   s   @r   r	   r	       so        " " " " ""5 5 5 5 5  - - - - - - -r   r	   )samba.testsr   r   sambasamba.credentialsr   samba.dcerpcr   r   r   r%   r	    r   r   <module>r9      s   $ !           				  ) ) ) ) ) ) ! ! ! ! ! ! ) ) ) ) ) ) ) ) 
&- &- &- &- &-8 &- &- &- &- &-r   