File: //lib/python3/dist-packages/samba/tests/samba_tool/__pycache__/base.cpython-310.pyc
o
�/aR � @ sr d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl m
Z
d dl Z d dd�Z
G d d
� d
e jj�ZdS ) � N)�system_session)�SamDB)�StringIO)�
cmd_sambatool�d c C s0 t | �|d k r
| S | d |� dt | �| S )N� z[%d more characters])�len)�s�cutoff� r �=/usr/lib/python3/dist-packages/samba/tests/samba_tool/base.py�truncate_string" s r
c @ sn e Zd Zdd� Zdd� Zdd� Zdd� Zdd
d �Zdd d
�Zddd�Z d dd�Z
dd� Z ddd�Z dd� Z
dS ) �SambaToolCmdTestc G s� t jd jd }|�d�\}}|�t|��\}}|dd� }t|j�}|jD ]}|jD ] } | j dur6|| j = q+q&|�
|� |� dd�}
|� dd�} |� dd�} | � � }
| j
|
dd �}t|
t� ||
d
�}|S ) zFa convenience function to get a samdb instance so that we can query it�user� setexpiry� N�H� sambaopts�credoptsT)�fallback_machine)�url� session_info� credentials�lp)r � subcommands�_create_parser�
parse_args�list�dict�__dict__�
option_groups� option_list�dest�update�get� get_loadparm�get_credentialsr r )�self�argv�cmd�parser� optiongroups�opts�args�kwargs� option_group�optionr r r r �creds�samdbr r r �getSamDB* s*
��
�zSambaToolCmdTest.getSamDBc G sF t j| }t� |_t� |_|jd| g|�R � }||j�� |j�� fS )zrun a single level command�
samba-tool %s�r r r �outf�errf�_run�getvalue)r' �namer- r) �resultr r r �runcmdH s
zSambaToolCmdTest.runcmdc G sP t j| j| }t� |_t� |_|jd||f g|�R � }||j�� |j�� fS )z run a command with sub commandszsamba-tool %s %sr5 )r' r: �subr- r) r; r r r � runsubcmdP s
zSambaToolCmdTest.runsubcmdc G sj t j| }d| }|D ]
}|j| }|d| 7 }q t� |_t� |_|j|g|�R � }||j�� |j�� fS )z3run a command with any number of sub command levelsr4 z %sr5 )r' r: � sublevelsr- r) �cmd_strr= r; r r r �runsublevelcmd[ s
z SambaToolCmdTest.runsublevelcmd� c C s* d||||f }| j ||�dd�d� d S )Nz"exit[%s] stdout[%s] stderr[%s]: %sz
]
z
]
)�msg)� assertIsNone�replace)r' �exit�out�errrC r r r �assertCmdSuccessl s z!SambaToolCmdTest.assertCmdSuccessc C s | � ||� d S �N)�assertIsNotNone)r' �valrC r r r �
assertCmdFails s z SambaToolCmdTest.assertCmdFailNc C s. |d u rdt |�t |�f }| �|||� d S )Nz%r is not in %r)r
�assertIn)r' �base�stringrC r r r � assertMatchv s
�z SambaToolCmdTest.assertMatch� c C s0 t �tj�}|d�dd� t|d �D ��7 }|S )zPCreate a random name, cap letters and numbers, and always starting with a letterrB c s s( � | ]}t �tjtj tj �V qd S rJ )�random�choicerP �ascii_uppercase�ascii_lowercase�digits)�.0�xr r r � <genexpr>� s �& z.SambaToolCmdTest.randomName.<locals>.<genexpr>r )rS rT rP rU �join�range)r' �countr: r r r �
randomName} s zSambaToolCmdTest.randomNamec C sr z| j W n ty g | _ Y nw d}|dk r5t�dd�}|| j vr-| j |g7 _ |S |d7 }|dk sJ d��)Nr �2 iX�G i:I r FzNo Xid are available)� used_xids�AttributeErrorrS �randint)r' �failed�xidr r r � randomXid� s
�
�zSambaToolCmdTest.randomXidc C s$ | � || |ko
|| |k |� dS )zFAssert that val1 is within delta of val2, useful for time computationsN)�
assertTrue)r' �val1�val2�deltarC r r r � assertWithin� s $z SambaToolCmdTest.assertWithinc C sL | � ddd| dtjd tjd f dtjd �\}}}| �|||� d S ) N�domain�demotez --remove-other-dead-server=%s z-U%s%%%s�USERNAME�PASSWORDz --server=%s�SERVER)r> �os�environrI )r' � netbios_namer; rG rH r r r � cleanup_join� s ��z SambaToolCmdTest.cleanup_join)rB rJ )rR )�__name__�
__module__� __qualname__r3 r< r> rA rI rM rQ r^ re rj rs r r r r r ( s
r )r )rp rS rP �
samba.authr � samba.samdbr �ior �samba.netcmd.mainr � samba.tests�sambar
�tests�BlackboxTestCaser r r r r �<module> s