File: //usr/lib/python3/dist-packages/samba/tests/__pycache__/__init__.cpython-310.pyc
o
�hb�[ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddlmZ ddl m
Z
ddlm Z ddl Z ddl
Z
ddlZddlZddlmZmZ ddlZddlZddlmZ dd lmZ dd
lmZ ddlZzdd lm Z W n
e y} d d
� Z Y nw ddl Zddl Zddl Zzddlm!Z! W n e y� G dd� de"�Z!Y nw ej#�$ej#�%ej#�&e'�d��Z(e)dd� e*d�D ��Z+dd� e,e��-� D �Z.dd� Z/dd� Z0G dd � d ej1�Z1G d d � d e1�Z2G d d � d e1�Z3d!d"� Z4d@d$d%�Z5dZ6G d&d'� d'e1�Z7G d(d)� d)e1�Z8G d*d+� d+e"�Z9G d,d-� d-e3�Z: .dAd/d0�Z; #dBd1d2�Z<dCd3d4�Z=d5d6� Z>d7d8� Z?eG d9d:� d:e��Z@ ;dDd<d=�ZA dEd>d?�ZBdS )FzSamba Python tests.� N)�param)� credentials)� Credentials)�gensec)�IntEnum�unique)�randint)� SystemRandom)�contextmanager)�SamDBc O s d S �N� )�args�kwargsr
r
�6/usr/lib/python3/dist-packages/samba/tests/__init__.pyr . � r )�SkipTestc @ � e Zd ZdZdS )r z
Test skipped.N��__name__�
__module__� __qualname__�__doc__r
r
r
r r 8 � r z../../../../binc C s4 g | ]}t tt|���d kr|dk r|ntd��qS )� � �.)�len�repr�chr�ord��.0�xr
r
r �
<listcomp>> s 4 r$ � c C s i | ] \}}|� d �r||�qS ��ERR_)�
startswith�r" �k�vr
r
r �
<dictcomp>@ s r, c
C st t | tj�r | jd } | tv rt| S zdd�dd� | D ��� d�W S ty9 } z t|� W Y d }~| S d }~ww )Nr �[�, c s s � | ] }t �||�V qd S r )� LDB_ERR_LUT�getr! r
r
r � <genexpr>J s � zldb_err.<locals>.<genexpr>�])�
isinstance�ldb�LdbErrorr r/ �join� TypeError�print)r+ �er
r
r �ldb_errB s
��r: c C s | � � | S r )�setUpDynamicTestCases)�clsr
r
r �DynamicTestCaseP s r= c sz e Zd ZdZedd�dd��Zedd� �Z� fdd �Zd
d � Zd d
� Z dd� Z
dd� Z ddd�Z ddd�Z
dd� Z� ZS )�TestCasezA Samba test case.N)�docc sJ � �fdd�}||_ d�|f }t| |�r td|� d���t| ||� dS )a
fnname is something like "test_dynamic_sum"
suffix is something like "1plus2"
argstr could be (1, 2)
This would generate a test case called
"test_dynamic_sum_1plus2(self)" that
calls
self._test_dynamic_sum_with_args(1, 2)
c s t | d� �� � d S )Nz
_%s_with_args)�getattr��self�r �fnnamer
r �fnc s z*TestCase.generate_dynamic_test.<locals>.fnz%s_%sz
Dynamic test z already exists!N)r �hasattr� RuntimeError�setattr)r< rD �suffixr? r rE �attrr
rC r �generate_dynamic_testW s
z TestCase.generate_dynamic_testc C s d| }t |��)z�This can be implemented in order to call cls.generate_dynamic_test()
In order to implement autogenerated testcase permutations.
z=%s needs setUpDynamicTestCases() if @DynamicTestCase is used!)� Exception)r< �msgr
r
r r; k s z TestCase.setUpDynamicTestCasesc sR t t| ��� t�d�}|d ur't|�}t�� | _t� |� | �
tj |� d S d S )N�TEST_DEBUG_LEVEL) �superr> �setUp�os�getenv�int�samba�get_debug_level�_old_debug_level�set_debug_level�
addCleanup)rB �test_debug_level�� __class__r
r rP s s
�zTestCase.setUpc C s t � S r )� env_loadparmrA r
r
r � get_loadparm| s zTestCase.get_loadparmc C s t S r )�cmdline_credentialsrA r
r
r �get_credentials r zTestCase.get_credentialsc C s" | � � }|�| �� �}|�� }|S r )r_ �get_named_ccacher] �get_name)rB �creds�ccache� ccache_namer
r
r �get_creds_ccache_name� s z TestCase.get_creds_ccache_namec C s� d}d}t |t�}|ru|d d� }|dd� }|dd � }|r>d�dd� |D ��}d�dd� |D ��}|�t�}|�t�}n$d�d d� |D ��}d�d
d� |D ��}|�t��d �}|�t��d �}|d |d
|d
|||f 7 }|d7 }|s |S )Nr � � � � c S � g | ]}d t |� �qS �z%02X�r r! r
r
r r$ � � z$TestCase.hexdump.<locals>.<listcomp>c S rj rk rl r! r
r
r r$ � rm c S � g | ]}d | �qS rk r
r! r
r
r r$ � � c S rn rk r
r! r
r
r r$ � ro �utf8z[%04X] %-*s %-*s %s %s
� )r3 �strr6 � translate�HEXDUMP_FILTER�decode) rB �src�N�result� is_string�ll�lr�hl�hrr
r
r �hexdump� s(
�zTestCase.hexdumpc C s� |d u rt d��|d ur|d u rt d��|d u r&|d u s J �|�� }|�� }|�� }|d u r2|�� }t� }|�|� |�|� |�|� � � |�
|� � � |� |�
� � |�|�� tjB � |�|� |rj|�|� |S )Nz)you need to supply a Credentials templatez8you cannot set creds username without setting a password)�
ValueError� get_username� get_password� get_bind_dn�get_kerberos_stater � set_username� set_password�
set_domain�
get_domain� set_realm� get_realm�set_workstation�get_workstation�set_gensec_features�get_gensec_featuresr � FEATURE_SEAL�set_kerberos_state� set_bind_dn)rB �template�username�userpass�kerberos_state�simple_bind_dn�cr
r
r � insta_creds� s4 �
�
zTestCase.insta_credsFc C s� |r
|� � }|� � }||krFtj�d|rdndt|�t|�f � ddlm} ||�d�|�d�dd�}|D ]}tj�|� q6| �|� d S d S )
z�Assert equality between two strings and highlight any differences.
If strip is true, leading and trailing whitespace is ignored.z8The strings differ %s(lengths %d vs %d); a diff follows
zwhen stripped rf r )� unified_diffT�a�bN) �strip�sys�stderr�writer �difflibr� �
splitlines�fail)rB r� r� rM r� r� �diff�liner
r
r �assertStringsEqual� s$
��
��zTestCase.assertStringsEqualc sr |du r|j � d|� d|� d�}z ||i |�� W no tjy� } zb|j\}}t|tjj�r3||v } n||k} | s}dd� tt�� � D �� t|tjj�rXd�
� fdd �|D ��}
n� � |�}
| � |� d
|
� d |� d � � |�� d
|� d|� � � W Y d}~dS W Y d}~dS d}~ww dd� tt�� � D �� t|tjj�r�d�
� fdd �|D ��}
n� � |�}
| � d||
|f � dS )z/Assert a function raises a particular LdbError.Nz(*z, **�)c S �* i | ]\}}|� d �rt|t�r||�qS r& �r( r3 rS r) r
r
r r, � �
� �z1TestCase.assertRaisesLdbError.<locals>.<dictcomp>ri c 3 � � | ]}� � |�V qd S r �r0 r! ��lutr
r r1 � � � z0TestCase.assertRaisesLdbError.<locals>.<genexpr>z, expected LdbError r. z got z (z) c S r� r& r� r) r
r
r r, � r� c 3 r� r r� r! r� r
r r1 � r� z1%s, expected LdbError %s, (%s) but we got success)
r r4 r5 r r3 � collections�abc� Container�vars�itemsr6 r0 r� ) rB �errcode�message�fr r r9 �numrM �found� errcode_namer
r� r �assertRaisesLdbError� sL
��������
�
�z TestCase.assertRaisesLdbError)NNNN)NF)r r r r � classmethodrK r; rP r] r_ re r~ r� r� r� �
__classcell__r
r
rZ r r> T s
"r> c s, e Zd ZdZ� fdd�Zg fdd�Z� ZS )� LdbTestCasez2Trivial test case for running tests against a LDB.c s8 t t| ��� tjdd�| _| jj| _t�| j�| _ d S )NF)�delete)
rO r� rP �tempfile�NamedTemporaryFile�name�filenamerT �Ldbr4 rA rZ r
r rP s
zLdbTestCase.setUpc C sD t �� }t �| j d�|_d�|�|d< | j �|� t�| j�| _ dS )z Change the modules for this Ldb.z@MODULES�,z@LISTN) r4 �Message�Dn�dnr6 �addrT r� r� )rB �modules�mr
r
r � set_modules s
zLdbTestCase.set_modules)r r r r rP r� r� r
r
rZ r r� � s r� c s0 e Zd Z� fdd�Zdd� Zedd� �Z� ZS )�TestCaseInTempDirc s( t t| ��� t�� | _| �| j� d S r )rO r� rP r� �mkdtemp�tempdirrX �_remove_tempdirrA rZ r
r rP s
zTestCaseInTempDir.setUpc C s* | � g t�| j�� t�| j� d | _d S r )� assertEqualrQ �listdirr� �rmdirrA r
r
r r� s
z!TestCaseInTempDir._remove_tempdirc c s� � z9t j| jd�\}}|V W z
t�|� t�|� W dS ttfy: } ztd| t j
d� W Y d}~dS d}~ww z t�|� t�|� W w ttfyd } ztd| t j
d� W Y d}~w d}~ww )z*Yield a temporary filename in the tempdir.)�dirz#could not remove temporary file: %s)�fileN) r� �mkstempr� rQ �close�unlink�OSError�IOErrorr8 r� r� )rB �fdrE r9 r
r
r �mktemp s, �
����
���zTestCaseInTempDir.mktemp)r r r rP r� r
r� r� r
r
rZ r r� s
r� c C s6 t �� } z | �tjd � W | S ty td��w )N�
SMB_CONF_PATHzSMB_CONF_PATH not set)r �LoadParm�loadrQ �environ�KeyError)�lpr
r
r r\ + s ��r\ Fc C s: |r | t j�� vr dS | t j�� v sJ d| ��t j| S )z�Returns value for variable in os.environ
Function throws AssertionError if variable is defined.
Unit-test based python tests require certain input params
to be set in environment, otherwise they can't be run
Nz Please supply %s in environment)rQ r� �keys)�var_name�
allow_missingr
r
r �env_get_var_value4 s
r� c @ r )�RpcInterfaceTestCasezDCE/RPC Test case.Nr r
r
r
r r� E r r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ValidNetbiosNameTestsc C � | � t�d�� d S )N�FOO)�
assertTruerT �valid_netbios_namerA r
r
r �
test_validK � z ValidNetbiosNameTests.test_validc C r� )N� FOOFOOFOOFOOFOOFOOFOOFOOFOOFOO�� assertFalserT r� rA r
r
r �
test_too_longN r� z#ValidNetbiosNameTests.test_too_longc C r� )Nz*BLAr� rA r
r
r �test_invalid_charactersQ r� z-ValidNetbiosNameTests.test_invalid_charactersN)r r r r� r� r� r
r
r
r r� I s r� c @ s" e Zd ZdZddd�Zdd� ZdS )�BlackboxProcessErrorz�This is raised when check_output() process returns a non-zero exit status
Exception instance should contain the exact exit code (S.returncode),
command line (S.cmd), process output (S.stdout) and process error stream
(S.stderr)
Nc C sF || _ t|t�rd�|�| _d| _n|| _d| _|| _|| _|| _d S )Nri FT) �
returncoder3 �listr6 �cmd�shell�stdoutr� rM )rB r� r� r� r� rM r
r
r �__init__] s
z BlackboxProcessError.__init__c C s8 d| j | j| j| j| jf }| jd urd|| jf }|S )NzBCommand '%s'; shell %s; exit status %d; stdout: '%s'; stderr: '%s'z%s; message: %s)r� r� r� r� r� rM )rB �sr
r
r �__str__i s �
z BlackboxProcessError.__str__r )r r r r r� r� r
r
r
r r� U s
r� c @ sF e Zd ZdZdd� Zddd�Zddd�Zd d
� Zd d � Zddd�Z dS )�BlackboxTestCasez"Base test case for blackbox tests.c C s� t |t�r|}n|�dd�}|d }tj�t|�}g d�}tj�|�r'||d< ||v r:t�dd�r:|� dtj
d � t |t�sDd�|�}|S )zQExpand the called script into a fully resolved path in the bin
directory.ri � r )z
samba-tool�samba_dnsupdate�samba_upgradednszscript/traffic_replayzscript/traffic_learner�PYTHONN) r3 r� �splitrQ �pathr6 �BINDIR�existsrR �insertr� )rB r� �partsr� �exe� python_cmdsr
r
r �
_make_cmdlinev s
z BlackboxTestCase._make_cmdlineNc C s | j |d|d� d S )Nr )rM )�check_exit_code)rB r� rM r
r
r � check_run� r� zBlackboxTestCase.check_runc C sn | � |�}t|t� }tj|tjtj|d�}|�� \}}|j}||kr5|d u r-d||f }t|||||��|S )N�r� r� r� z expected return code %s; got %s) r r3 r� �
subprocess�Popen�PIPE� communicater� r� ) rB r� �expectedrM � use_shell�p�
stdoutdata�
stderrdata�retcoder
r
r r � s&
� �z BlackboxTestCase.check_exit_codec C sV t |t� }| �|�}tj|tjtj|dd�}|�� \}}|j}|r)t||||��|S )NT)r� r� r� � close_fds) r3 r� r r r
r r r� r� �rB r� r r r r r r
r
r � check_output� s
� z BlackboxTestCase.check_outputc C sT | � |�}t|t� }tj|tjtj|d�}|�� \}}|j}||�d�|�d�fS )Nr �UTF8) r r3 r� r r
r r r� ru r r
r
r � run_command� s
� z BlackboxTestCase.run_command� c C sV t � �tj�}|t � �tj�7 }|t � �tj�7 }|d�dd� t|d �D ��7 }|S )Nrf c s s* � | ]}t � �tjtj tj �V qd S r )r �choice�string�ascii_uppercase�ascii_lowercase�digitsr! r
r
r r1 � s �
���z3BlackboxTestCase.random_password.<locals>.<genexpr>r )r r r r r r r6 �range)rB �count�passwordr
r
r �random_password� s
�z BlackboxTestCase.random_passwordr )r )
r r r r r r
r r r r$ r
r
r
r r� s s
r� Tc C s� d| vr|st j�| �rd| } nd| } | �d�r dg}n|r&td| ��|du r-t� }|du r7tj�|�}|du r=t }t
| ||||||d�S ) au Create SamDB instance and connects to samdb_url database.
:param samdb_url: Url for database to connect to.
:param lp: Optional loadparm object
:param session_info: Optional session information
:param credentials: Optional credentials, defaults to anonymous.
:param flags: Optional LDB flags
:param ldap_only: If set, only remote LDAP connection will be created.
:param global_schema: Whether to use global schema.
Added value for tests is that we have a shorthand function
to make proper URL for ldb.connect() while using default
parameters for connection based on test environment
z://ztdb://%sz ldap://%szldap://zmodules:paged_searchesz;Trying to connect to %s while remote connection is requiredN)�urlr� � session_infor �flags�options�
global_schema) rQ r �isfiler( �AssertionErrorr\ rT �auth�system_sessionr^ r )� samdb_urlr� r& r r' � ldb_options� ldap_onlyr) r
r
r �
connect_samdb� s0
� �r1 c C s6 t | ||||||�}|jddtjdgd�}||d fS )a� Connects to samdb_url database
:param samdb_url: Url for database to connect to.
:param lp: Optional loadparm object
:param session_info: Optional session information
:param credentials: Optional credentials, defaults to anonymous.
:param flags: Optional LDB flags
:param ldap_only: If set, only remote LDAP connection will be created.
:return: (sam_db_connection, rootDse_record) tuple
rf �*)�base�
expression�scope�attrsr )r1 �searchr4 �
SCOPE_BASE) r. r� r& r r' r/ r0 �sam_db�resr
r
r �connect_samdb_ex� s
� � r; c C sT t | �}t�� }|du rt�� }|�|� |�t |�� |�t |�� t|||d�S )a Connect to SamDB by getting URL and Credentials from environment
:param env_url: Environment variable name to get lsb url from
:param env_username: Username environment variable
:param env_password: Password environment variable
:return: sam_db_connection
N)r r� ) r� r r r r� �guessr� r� r1 )�env_url� env_username� env_passwordr� r. rb r
r
r �connect_samdb_env s
r@ c
K sb z | j |fi |�� W d S tjy0 } z|j\}}|tjks%J d| ��W Y d }~d S d }~ww )Nzldb.delete() failed: %s)r� r4 r5 r �ERR_NO_SUCH_OBJECT)�samdbr� r �errorr� �errstrr
r
r � delete_force# s
$��rE c C s8 t dd�}t�| d||| �� f �}| �|dd�� |S )z Creates a unique OU for the testr� i��� z
OU=%s%d,%s�organizationalUnit)r� � objectclass)r r4 r� �get_default_basednr� )rB r� �randr� r
r
r �create_test_ou+ s
rJ c @ s e Zd ZdZdZdZdZdS )�OptStater r� � r N)r r r �NOOPT�HYPHEN1�HYPHEN2�NAMEr
r
r
r rK 8 s
rK �
c
C s� |d u rg }nd }| � d�D ]3}|�� }|dkrq|d u r%||kr$g }nqt|�t|�| k r0q|d dkr;|�|� ||krA nq|d u rHd S t�d�j}|D ]{}tj} d }
d} |D ]a} | tjkrp| dkrm| � � rmtj
} | } q[| tj
kr�| � � r�d| }
tj } n| dkr�tj
} q[| tj
kr�| � � r�d| }
tj } ntj} | } q[| tj kr�|| �r�|
| 7 }
q[|�|
g ��|� tj} | } q[| tj kr�|�|
g ��|� qPd S )N�
rf r �-z^[\w-]$ri z--)r �lstripr �append�re�compile�matchrK rM �isspacerN �isalnumrP rO �
setdefault)
�out�
options_start� options_end�optmap�max_leading_spaces� opt_lines�raw_liner� �is_longname_char�stater� �prevr� r
r
r �parse_help_consistency@ sp
�
�
��rf c C sf i }t | |||� g }t|�� �D ]\}}t|�dkr)|D ] }|�d||f � q q|r1d�|�S dS )a� Ensure that options are not repeated and redefined in --help
output.
Returns None if everything is OK, otherwise a string indicating
the problems.
If options_start and/or options_end are provided, only the bit in
the output between these two lines is considered. For example,
with samba-tool,
options_start='Options:', options_end='Available subcommands:'
will prevent the test looking at the preamble which may contain
examples using options.
r� z%s: %srR N)rf �sortedr� r rU r6 )r\ r] r^ r_ �errorsr* �valuesr+ r
r
r �check_help_consistency� s � �
�rj )F)NNNr NFT)NNNr NFr )NNNrQ )NN)Cr rQ r� �warningsr� r4 rT r r �samba.credentialsr r r r� �unittestrV �enumr r �
samba.auth�samba.dcerpc.base�randomr r �
contextlibr
r � samba.samdbr � ImportError� samba.ndr�samba.dcerpc.dcerpc�samba.dcerpc.epmapperr rL r �abspathr6 �dirname�__file__r � bytearrayr! rt r� r� r/ r: r= r> r� r� r\ r� r^ r� r� r� r� r1 r; r@ rE rJ rK rf rj r
r
r
r �<module> s� � �� *
[
�-
�
�K�