
    b                     N    d dl Z d dlmZmZ d dlmZ dZdZ G d de          ZdS )    N)BlackboxTestCaseBlackboxProcessError)	Messagingzbin/smbcontrolpingc                   6     e Zd Z fdZd Zd Zd Zd Z xZS )SmbcontrolProcessBlockboxTestsc                     t          t          |                                            |                                 }t	          |          | _        d S )N)lp_ctx)superr   setUpget_loadparmr   msg_ctx)selfr
   	__class__s     I/usr/lib/python3/dist-packages/samba/tests/blackbox/smbcontrol_process.pyr   z$SmbcontrolProcessBlockboxTests.setUp    sG    ,d3399;;;""$$ ///    c                     | j                                         }g }|D ]-}|j        D ]#}|                    |j        |j        f           $.|S N)r   irpc_all_serversidsappendnamepid)r   services	processesserviceids        r   get_process_dataz/SmbcontrolProcessBlockboxTests.get_process_data%   se    <0022	 	9 	9Gk 9 9  ',!788889r   c                 R    |                                  }|D ]\  }}||k    r|c S d S r   )r   )r   r   r   pnamer   s        r   get_processz*SmbcontrolProcessBlockboxTests.get_process.   sB    ))++	# 	 	JE3u}}


 tr   c           	      \   d}d}|                      d          }	 |                     t          d|dt          d           n,# t          $ r}|                     d           Y d }~nd }~ww xY w	 |                     t          d|d|d|d           n;# t          $ r.}t          |           |                     d	           Y d }~nd }~ww xY w	 |                     t          d|dt          d           |                     d
           d S # t          $ r}Y d }~d S d }~ww xY w)Ninjectsegv
rpc_server trying to ping rpc_servermsg!Unable to ping rpc_server processzinjecting fault into rpc_serverz4Unable to inject a fault into the rpc_server processzCould ping rpc_server process)r!   	check_runCOMMANDPINGr   failprint)r   INJECTFAULTr   es        r   test_inject_faultz0SmbcontrolProcessBlockboxTests.test_inject_fault5   s   
 |,,
	;NN###tt<:  < < < <# 	; 	; 	;II9::::::::	;	NNNGGGSSS&&&%%H@  B B B B# 	N 	N 	N!HHHIILMMMMMMMM	N	NN###tt<:  < < <II566666# 	 	 	DDDDD	sA   )A 
A.A))A.2'B 
C$$CC>D 
D+&D+c           	         d}d}d}|                      d          }	 |                     t          d|dt          d           n,# t          $ r}|                     d           Y d }~nd }~ww xY wt          j                    }	 |                     t          d|d|d|d	|z             n># t          $ r1}t          |           |                     d
|z             Y d }~nd }~ww xY w	 |                     t          d|dt          d           t          j                    }||z
  }|                     ||z   |           d S # t          $ r }|                     d           Y d }~d S d }~ww xY w)Nsleep      ldap_serverr&   r'   r(   r*   z"putting rpc_server to sleep for %dz(Failed to get rpc_server to sleep for %d)	r!   r+   r,   r-   r   r.   timer/   assertGreater)	r   SLEEPDURATIONDELTAr   r2   startenddurations	            r   
test_sleepz)SmbcontrolProcessBlockboxTests.test_sleep[   s    }--	;NN###tt<:  < < < <# 	; 	; 	;II9::::::::	; 		MNNGGGSSS%%%JChN  P P P P# 	M 	M 	M!HHHII@8KLLLLLLLL	M	;NN###tt<:  < < <)++CU{Hx%/:::::# 	; 	; 	;II9:::::::::	;sH   )A 
A0A++A0*B2 2
C-<'C((C-1AE 
E7E22E7)	__name__
__module____qualname__r   r   r!   r3   rA   __classcell__)r   s   @r   r   r      su        0 0 0 0 0
    $ $ $L(; (; (; (; (; (; (;r   r   )	r9   samba.testsr   r   samba.messagingr   r,   r-   r    r   r   <module>rI      s   ,  > > > > > > > > % % % % % %
e; e; e; e; e;%5 e; e; e; e; e;r   