o

�/a2�@srdd
lm
Z
mZ
ddlZddlZddlmZ
ddlmZ
ddl	m
Z
mZmZ
ddl
mZ
	dZGdd	�d	e
�ZdS)
�)�TestCase�delete_forceN)
�system_session)
�Credentials)�UF_WORKSTATION_TRUST_ACCOUNT�UF_PASSWD_NOTREQD�UF_NORMAL_ACCOUNT)
�SamDB�
krb5credstestcsHeZ
dZ�f
d
d�Z�f
dd�Zdd�Zdd�Zd	d
�Zdd�Z�Z	S)
�PyKrb5CredentialsTestsc

sxtt
|���
tjd
|_tjd|_tjd|_|��|_	|�
�|_t�|_
td|j|j
|j|j	d�|_|��
dS)N�SERVER�DOMAIN�	SERVER_IPz	ldap://%s)�url�session_info�credentials�lp)�superr�setUp�os�environ�server�domain�host�get_loadparmr�get_credentialsrr�sessionr	�ldb�create_machine_account�
�self�
�	__class__��>/usr/lib/python3/dist-packages/samba/tests/krb5_credentials.pyr-s










�zPyKrb5CredentialsTests.setUpc

s tt
|���
t|j|j�
dS�
N)rr�tearDownrr�
machine_dnrr!r#r$r&?�

zPyKrb5CredentialsTests.tearDownc
Cs(d
}
|j�
|j|
�}|�|��|
�
dS)NzMEMORY:py_creds_machine)�
machine_creds�get_named_ccacher�assertEqual�get_name)r �name�ccacher#r#r$�test_get_named_ccacheCs




�z,PyKrb5CredentialsTests.test_get_named_ccachec
Cs |j�
|j�
}
|�|
���

dSr%)r)r*r�assertIsNotNoner,)r r.r#r#r$�test_get_unnamed_ccacheIr(z.PyKrb5CredentialsTests.test_get_unnamed_ccachec
CsF|j�
|j�
}
t�}|�|
���

|�
|j�
}|�|
��|���
dSr%)r)r*rr�set_named_ccacher,r+)r r.�creds�ccache2r#r#r$�test_set_named_ccacheMs



z,PyKrb5CredentialsTests.test_set_named_ccachec
Cs�t�
d
d
�|_t|_d|j|j��f|_t|j|j�
|j�	d�
�
d�
}
d�|
�
�	d�
}|j�|jdd|jt
ttB�
|d��

t�|_|j�|���

|j�|j�

|j�|jd	�

|j�|j�

dS)
N� zcn=%s,%szutf-8z"{0}"z	utf-16-le�computerz%s$)�dn�objectclass�sAMAccountName�userAccountControl�
unicodePwd�
$)�samba�generate_random_password�machine_pass�MACHINE_NAME�machine_namer�	domain_dnr'r�encode�decode�format�add�strrrrr)�guessr�set_password�set_username�set_workstation)r �pass_unicode�utf16pwr#r#r$rXs$









�



z-PyKrb5CredentialsTests.create_machine_account)
�__name__�
__module__�__qualname__rr&r/r1r5r�
__classcell__r#r#r!r$r+sr)�samba.testsrrrr>�
samba.authr�samba.credentialsr�
samba.dsdbrrr�samba.samdbr	rArr#r#r#r$�<module>
s