o

�/a(�@stdZd
dl
mZ
d
dlZd
dlmZmZ
d
dlZd
dl	Z	Gdd�de
�ZGdd�dejj
�ZGd	d
�d
ejj
�ZdS)zOTests for string and unicode handling in PIDL generated bindings
samba.dcerpc.*�)
�drsblobsN)�
ndr_unpack�ndr_packc
@seZ
dZd
S)�
TestExceptionN)�__name__�
__module__�__qualname__�r	r	�A/usr/lib/python3/dist-packages/samba/tests/dcerpc/string_tests.pyrs
rcsPeZ
dZ�f
d
d�Z�f
dd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	�Z
S)�StringTestsc

s$tt
|���
t��
t��|_dS�
N)�superr�setUp�talloc�enable_null_tracking�total_blocks�startup_blocks�
�self�
�	__class__r	r
r"s


zStringTests.setUpc

s>tt
|���
t��
t��|jkrt��
|�	d
�

dSdS)Nz it appears we are leaking memory)
r
r�tearDown�gc�collectrrr�report_full�failrrr	r
r's




�zStringTests.tearDownc
Cs�t�
�}
d
|
_d|
_t��
|�|
�

|�|
jd
�
|�|
jd�
d|
_d|
_|�|
jd�
|�|
jd�
d|
_|�|
jd�
|�|
j�

dS)Nzhello.example.comzgoodbye.example.com�)	r�repsFromTo2OtherInfo�	dns_name1�	dns_name2rr�assertIsNotNone�assertEqual�assertIsNone�r�infor	r	r
�test_string_from_python.s










z#StringTests.test_string_from_pythonc

Cs$z|��
t
��
t
y


YdSwr)r%rrr	r	r
�test_string_with_exceptionBs




�z&StringTests.test_string_with_exceptionc
Cs@d
d�}
|
�}t�
�
|�|�

|�|jd�
|�|jd�
dS)Nc
Sst�
�}d
|_d|_|S�N�
1.example.com�
2.example.com)rrrr)
r$r	r	r
�get_infoJs



z>StringTests.test_string_from_python_function.<locals>.get_infor(r))rrr r!rr)rr*r$r	r	r
� test_string_from_python_functionIs





z,StringTests.test_string_from_python_functionc
Cs�t�
�}
d
|
_d|
_t��
|�|
�

|�|
jd
�
|�|
jd�
|
jd7_|
jd;_|�|
jd�
|�|
jd�
~
dS)Nr(z%s.example.comz.co.nz�z1.example.com.co.nzr))rrrrrrr r!r#r	r	r
�test_string_modify_in_placeVs












z'StringTests.test_string_modify_in_placec
Csdt�
�
t��}
d
|
_d|
_d|
_z|
`Wn	ty


Yn
w|�|
�

|�|
j�

|�|
j�

dSr')	rrrrrr�AttributeErrorr r"r#r	r	r
�test_string_deleteds








�


zStringTests.test_string_delete)rrrrrr%r&r+r-r/�
__classcell__r	r	rr
r s
rc@seZ
dZd
Zdd�ZdS)�StringTestsWithoutLeakCheckz�We know that the ndr unpacking test leaves stuff in the
    autofree_context, and we don't want to worry about that. So for
    this test we don't make meory leak assertions.c
CsZt�
�}
d
|
_d|
_t|
�
}t��
ttj
|�}|�|�

|�	|jd
�
|�	|jd�
dSr')
rrrrrrrrr r!)rr$�packed�
info_unpackedr	r	r
�test_string_from_ndrys







z0StringTestsWithoutLeakCheck.test_string_from_ndrN)rrr�__doc__r4r	r	r	r
r1ts
r1)r5�samba.dcerpcr�samba.tests�samba�	samba.ndrrrrr�	Exceptionr�tests�TestCaserr1r	r	r	r
�<module>
s



T