
    b                         d Z ddlZddlZddlZddlmZmZ ddlmZmZ  G d de          Z	 G d de          Z
 G d	 d
e          Z G d de          ZdS )zSamba Python tests.    N)arcfour_encryptstring_to_byte_array)TestCaseTestCaseInTempDirc                   2    e Zd Zd Zd Zd Zd Zd Zd ZdS )SubstituteVarTestCasec                 X    |                      dt          j        di                      d S )N assertEqualsambasubstitute_varselfs    2/usr/lib/python3/dist-packages/samba/tests/core.py
test_emptyz SubstituteVarTestCase.test_empty   s+    U1"b99:::::    c                 \    |                      dt          j        dddi                     d S )Nzfoo barbarblar   r   s    r   test_nothingz"SubstituteVarTestCase.test_nothing    s?    .y5%.II	K 	K 	K 	K 	Kr   c                 \    |                      dt          j        dddi                     d S )Nzfoo blaz
foo ${bar}r   r   r   r   s    r   test_replacez"SubstituteVarTestCase.test_replace$   s?    .|eU^LL	N 	N 	N 	N 	Nr   c                 \    |                      dt          j        dddi                     d S )Nzfoo ${bdkjfhsdkfh sdkfh r   r   r   r   s    r   test_brokenz!SubstituteVarTestCase.test_broken(   sB    3./IESX>ZZ	\ 	\ 	\ 	\ 	\r   c                 \    |                      dt          j        dddi                     d S )Nzfoo ${bla} gsffr   r   r   r   s    r   test_unknown_varz&SubstituteVarTestCase.test_unknown_var,   sA    *./@5%.QQ	S 	S 	S 	S 	Sr   c                 z    t          j        d           |                     t          t           j        d           d S )Nznothing to see herezNot subsituted: ${FOOBAR})r   check_all_substitutedassertRaises	Exceptionr   s    r   test_check_all_substitutedz0SubstituteVarTestCase.test_check_all_substituted0   s@    #$9:::)U%@5	7 	7 	7 	7 	7r   N)	__name__
__module____qualname__r   r   r   r   r   r"    r   r   r   r      sz        ; ; ;K K KN N N\ \ \S S S7 7 7 7 7r   r   c                       e Zd Zd ZdS )ArcfourTestCasec                 ^    d}d}d}t          ||          }|                     ||           d S )Ns   12345678s	   abcdefghis	   ڑZl׹ϙ)r   r   )r   keyplaincrypt_expectedcrypt_calculateds        r   test_arcfour_directz#ArcfourTestCase.test_arcfour_direct8   s>    :*366)9:::::r   N)r#   r$   r%   r.   r&   r   r   r(   r(   6   s#        ; ; ; ; ;r   r(   c                       e Zd Zd ZdS )StringToByteArrayTestCasec                 X    g d}t          d          }|                     ||           d S )N)	      Z      l               u   ÚZ°l×¹Ï)r   r   )r   expected
calculateds      r   test_byte_arrayz)StringToByteArrayTestCase.test_byte_arrayB   s7    ???)*JKK
:.....r   N)r#   r$   r%   r=   r&   r   r   r0   r0   @   s#        / / / / /r   r0   c                       e Zd Zd ZdS )LdbExtensionTestsc           	      N   | j         dz   }t          j        |          }	 |                    ddd           |                     d|                    t          j        |d          d                     ~t          j	        |           d S # ~t          j	        |           w xY w)Nz/searchone.ldbzfoo=dcr   )dnr   s   blar   )basedn	attribute)
tempdirr   Ldbaddr   	searchoneldbDnosunlink)r   pathls      r   test_searchonez LdbExtensionTests.test_searchoneJ   s    |..IdOO	EE%00111Vkk81D1DPUkVVX X X IdOOOOO IdOOOOs   AB B$N)r#   r$   r%   rN   r&   r   r   r?   r?   H   s#        	 	 	 	 	r   r?   )__doc__rH   rJ   r   r   r   samba.testsr   r   r   r(   r0   r?   r&   r   r   <module>rQ      s  $   



 				  7 7 7 7 7 7 7 7 3 3 3 3 3 3 3 37 7 7 7 7H 7 7 76; ; ; ; ;h ; ; ;/ / / / / / / /    )     r   