/usr/lib/python3.9/site-packages/certbot/_internal/tests/compat/__pycache__
a }|�g�j � @ s4 d Z ddlZddlZddlZddlZddlmZ ddlZddlmZ ddl m Z ddl m Z ddl m Z ddlm mZ ddlmZ z ddlZddlZddlZd ZW n ey� d ZY n0 d Zd Zd Zejedd�G dd� de��ZG dd� de�Z G dd� de�Z ejedd�G dd� de��Z G dd� dej�Z ejedd�G dd � d ej��Z G d d � d ej�Z!G d d � d ej�Z"G d!d"� d"ej�Z#G d#d$� d$ej�Z$G d%d&� d&ej�Z%G d'd(� d(ej�Z&G d)d*� d*ej'�Z(ej)d+d,� �Z*d-d.� Z+d/d0� Z,d1d2� Z-d8d4d5�Z.e/d6k�r0e�0e�1ej2d7d� e3g �� dS )9z#Tests for certbot.compat.filesystem� N)�mock)�util)�lock)� filesystem)�os)�TempDirTestCaseFTzS-1-1-0zS-1-5-18z S-1-5-32-544z"Tests specific to Windows security��reasonc sp e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Z� ZS )�WindowsChmodTestsz:Unit tests for Windows chmod function in filesystem modulec s t � �� t| j�| _d S �N��super�setUp� _create_probe�tempdir� probe_path��self�� __class__� �R/usr/lib/python3.9/site-packages/certbot/_internal/tests/compat/filesystem_test.pyr # s zWindowsChmodTests.setUpc C sb t j�| jd�}t �| j|� t| j��� }t|��� }t� |d� t| j��� }t|��� }d S )N�link� ) r �path�joinr �symlinkr �_get_security_dacl�GetSecurityDescriptorDaclr �chmod)r � link_path�ref_dacl_probeZ ref_dacl_link�cur_dacl_probeZ cur_dacl_linkr r r �test_symlink_resolution' s z)WindowsChmodTests.test_symlink_resolutionc sF t �t��t�| jd� t| j��� � t�| jd� t| j��� � d S )Nr i� )� win32security�ConvertStringSidToSid� EVERYBODY_SIDr r r r r r r ��daclZ everybodyr �test_world_permission6 s z'WindowsChmodTests.test_world_permissionc C s< t �| jd� t| j��� }t �| jd� t| j��� }d S )Nr i� )r r r r r )r r! r" r r r �test_group_permissions_noopE s z-WindowsChmodTests.test_group_permissions_noopc sp t �t��t �t�� t�| jd� t| j��� ���fdd�t d�� � �D �}� �fdd�t d�� � �D �}d S )N� c s( g | ] }� � |�d �kr� � |��qS �� �ZGetAce��.0�index)r( �systemr r � <listcomp>U s �z<WindowsChmodTests.test_admin_permissions.<locals>.<listcomp>r c s( g | ] }�� |�d � kr�� |��qS r, r. r/ )�adminsr( r r r3 W s �) r$ r% � SYSTEM_SID� ADMINS_SIDr r r r r �range� GetAceCount)r Z system_acesZ admin_acesr )r4 r( r2 r �test_admin_permissionsN s z(WindowsChmodTests.test_admin_permissionsc C s | � dtj� d S )N� )� _test_flag� ntsecuritycon�FILE_GENERIC_READr r r r �test_read_flag` s z WindowsChmodTests.test_read_flagc C s | � dtj� d S �N� )r; r< �FILE_GENERIC_EXECUTEr r r r �test_execute_flagc s z#WindowsChmodTests.test_execute_flagc C s | � dtjtjA tjA � d S )Nr- )r; r< �FILE_ALL_ACCESSr= rA r r r r �test_write_flagf s ��z!WindowsChmodTests.test_write_flagc C s | � dtj� d S )N� )r; r< rC r r r r �test_full_flagk s z WindowsChmodTests.test_full_flagc sT t �| jd|B � t| j��� � t�t��� �fdd�td� � � �D �}|d }d S )Nr c s( g | ] }� � |�d �kr� � |��qS r, r. r/ r' r r r3 v s �z0WindowsChmodTests._test_flag.<locals>.<listcomp>r ) r r r r r r$ r% r& r7 r8 )r Z everyone_modeZ windows_flagZacls_everybodyr r' r r; n s z WindowsChmodTests._test_flagc C s� t �dt�� �\}}}t| j�}t| j||� t�| jd� t | j�}t � t �}t| j�}t| j||� t�| jd� t | j�}d S )N� r ) r$ ZLookupAccountName�win32apiZ GetUserName�_get_security_ownerr � _set_ownerr r r r% r6 )r Zauthenticated_user�_�security_ownerZ security_daclZ admin_userr r r � test_user_admin_dacl_consistency s z2WindowsChmodTests.test_user_admin_dacl_consistency)�__name__� __module__� __qualname__�__doc__r r# r) r* r9 r> rB rD rF r; rM � __classcell__r r r r r s r c @ s* e Zd Zdd� Zdd� Zed dd��ZdS ) � UmaskTestc C s� t �d�}zjtj�| jd�}t �|� t �d� tj�| jd�}t �|� tj�| jd�}t j|dd� W t �|� n t �|� 0 d S )N� �probe1�? �dir2�dir3� )�mode)r �umaskr r r r �mkdir)r �previous_umaskZdir1rW rX r r r �test_umask_on_dir� s zUmaskTest.test_umask_on_dirc C s� t �d�}zftj�| jd�}t�|� t �d� tj�| jd�}t�|� tj�| jd�}t�|� W t �|� n t �|� 0 d S )NrT rU rV Zprobe2Zprobe3)r r[ r r r r rS � _create_file)r r] Zfile1Zfile2Zfile3r r r �test_umask_on_file� s z UmaskTest.test_umask_on_filerY c C s> d }z$t j| tj|d�}W |r:t�|� n|r8t�|� 0 d S )N)�flagsrZ )r �openr �O_CREAT�close)r rZ Z file_descr r r r_ � s �zUmaskTest._create_fileN)rY )rN rO rP r^ r` � staticmethodr_ r r r r rS � s rS c s$ e Zd Z� fdd�Zdd� Z� ZS )�ComputePrivateKeyModeTestc s t � �� t| j�| _d S r r r r r r r � s z ComputePrivateKeyModeTest.setUpc C s& t �| jd� t �| jd�}tr"n d S )NrY i� )r r r Zcompute_private_key_mode� POSIX_MODE)r Znew_moder r r � test_compute_private_key_mode� s z7ComputePrivateKeyModeTest.test_compute_private_key_mode)rN rO rP r rh rR r r r r rf � s rf c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �WindowsOpenTestc sP t j�| jd�}t�|t jt jB t jB d�}t � |� t |�� � � t � t��d S )N�filer )r r r r r rb rc �O_EXCL�O_RDWRrd r r r$ r% r& �r r Zdescr r' r �!test_new_file_correct_permissions� s z1WindowsOpenTest.test_new_file_correct_permissionsc sX t j�| jd�}t|d��� t�|t jt jB d�}t �|� t |�� � � t � t ��d S )Nrj �wr )r r r r rb rd r rk rl r r r$ r% r& rm r r' r �&test_existing_file_correct_permissions� s z6WindowsOpenTest.test_existing_file_correct_permissionsc C s< | j ddtjtjB d� t�t��(}| j ddtjtjB d� W d � n1 sP0 Y | j ddtjd� | j ddtjd� tj�| j d�}t |d �� � t � |�}zHt�t��"}| j d dtjd� W d � n1 s�0 Y W |�� n |�� 0 t�t��" | j d dtjd� W d � n1 �s.0 Y d S ) Nr@ F)� file_existra r- T� r: �5ro � � )�_test_one_creationr rc rk �pytest�raises�OSErrorr r r rb rd r ZLockFile�release�O_RDONLY)r �exc_infor Zfilelockr r r �test_create_file_on_open� s 6 0 z(WindowsOpenTest.test_create_file_on_openc C s� t j�| jt|��}|rNt j�|�sNt|d�� W d � n1 sD0 Y d }z t�||�}W |r�t �|� n|r�t �|� 0 d S )Nro ) r r r r �str�existsrb r rd )r Znumrq ra Zone_fileZhandlerr r r rv s �z"WindowsOpenTest._test_one_creationN)rN rO rP rn rp r} rv r r r r ri � s ri c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �TempUmaskTestsz8Tests for using the TempUmask class in `with` statementsc C s t �d�}t �|� |S )Nr )r r[ )r Z old_umaskr r r � _check_umask& s zTempUmaskTests._check_umaskc C s8 t �d� t �d�� W d � n1 s*0 Y d S �NrT rV )r r[ � temp_umaskr r r r �test_works_normally+ s z"TempUmaskTests.test_works_normallyc C sP t �d� z4t �d�� t� �W d � n1 s20 Y W n Y n0 d S r� )r r[ r� � Exceptionr r r r �!test_resets_umask_after_exception2 s (z0TempUmaskTests.test_resets_umask_after_exceptionN)rN rO rP rQ r� r� r� r r r r r� $ s r� �!Test specific to Windows securityc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �WindowsMkdirTestszFUnit tests for Windows mkdir + makedirs functions in filesystem modulec s6 t j�| jd�}t�|d� t�t��t |�� � � d S )N�dirr ) r r r r r r\ r$ r% r& r r )r r r r' r � test_mkdir_correct_permissions@ s z0WindowsMkdirTests.test_mkdir_correct_permissionsc sD t j�| jd�}t j�|d�}t�|d� t�t��t |�� � � d S )Nr� �subpathr ) r r r r r �makedirsr$ r% r& r r )r r r� r r' r �!test_makedirs_correct_permissionsK s z3WindowsMkdirTests.test_makedirs_correct_permissionsc C sN t j�| jd�}dd l }|j}t�|� zt�|� W n tyH Y n0 d S )Nr� r )r r r r r\ r r� ry )r r �std_osZoriginal_mkdirr r r � test_makedirs_switch_os_mkdirW s z/WindowsMkdirTests.test_makedirs_switch_os_mkdirN)rN rO rP rQ r� r� r� r r r r r� = s r� c @ s e Zd ZdZdd� ZdS )� MakedirsTestsz5Unit tests for makedirs function in filesystem modulec C sT t j�| jd�}t j�|d�}t�d�}zt�|d� W t�|� n t�|� 0 d S )Nr� r� rT r )r r r r r r[ r� )r r r� r] r r r r� h s z/MakedirsTests.test_makedirs_correct_permissionsN)rN rO rP rQ r� r r r r r� f s r� c sj e Zd ZdZ� fdd�Zejedd�dd� �Zej edd�d d � �Z d d � Z ejedd�d d� �Z � Z S )�CopyOwnershipAndModeTestzYTests about copy_ownership_and_apply_mode, copy_ownership_and_mode and has_same_ownershipc s t � �� t| j�| _d S r r r r r r r y s z CopyOwnershipAndModeTest.setUpr� r c s� t �t�}t �� j}|�|d� t�d��V}t�d��,}||_t j d| j dddd� W d � n1 sh0 Y W d � n1 s�0 Y |j d }|d d }|j d }|d d }|� � � t �t��d S ) NF� win32security.GetFileSecurityz win32security.SetFileSecurity�dummyr T�Z copy_userZ copy_groupr r- r@ )r$ r% r5 �SECURITY_ATTRIBUTES�SECURITY_DESCRIPTOR�SetSecurityDescriptorOwnerr �patch� return_valuer � copy_ownership_and_apply_moder Zcall_args_listr r& )r r2 �security�mock_getZmock_setZ first_callZ second_callr r' r �*test_copy_ownership_and_apply_mode_windows} s �B zCCopyOwnershipAndModeTest.test_copy_ownership_and_apply_mode_windows� Test specific to Linux securityc C s� t �d���}t �d��`}t �d��6}d|j_d|j_tjd| jdddd � W d � n1 s^0 Y W d � n1 s|0 Y W d � n1 s�0 Y |�| jdd� |�| jd� d S ) Nzos.chownzos.chmodzos.stat�2 �3 r� r Tr� ) r r� r� �st_uid�st_gidr r� r �assert_called_once_with)r Z mock_chownZ mock_chmodZ mock_statr r r �(test_copy_ownership_and_apply_mode_linux� s �`zACopyOwnershipAndModeTest.test_copy_ownership_and_apply_mode_linuxc C sD t j�| jd�}t j�| jd�}t�|d��� t�|d��� d S )NZtest1Ztest2ro )r r r r r � safe_openrd )r Zpath1Zpath2r r r �test_has_same_ownership� s z0CopyOwnershipAndModeTest.test_has_same_ownershipc C sf | j }t| jdd�}t�|d� t�d�� }t�||� W d � n1 sL0 Y |�||� d S )N�dst)�namer z-certbot.compat.filesystem._copy_win_ownership) r r r r r r r� Zcopy_ownership_and_moder� )r �srcr� Zmock_copy_ownerr r r �$test_copy_ownership_and_mode_windows� s * z=CopyOwnershipAndModeTest.test_copy_ownership_and_mode_windows)rN rO rP rQ r �unittest�skipIfrg r� � skipUnlessr� r� r� rR r r r r r� w s r� c sl e Zd ZdZ� fdd�Zdd� Zejedd�dd � �Z ej ed d�d d � �Z d d� Z dd� Z dd� Z� ZS )�CheckPermissionsTestz-Tests relative to functions that check modes.c s t � �� t| j�| _d S r r r r r r r � s zCheckPermissionsTest.setUpc C s t �| jd� d S )Nr �r r r r r r r �test_check_mode� s z$CheckPermissionsTest.test_check_moder� r c C sT t �t�}t �� j}|�|d� t�d��}||_W d � n1 sF0 Y d S )NFr� ) r$ r% r5 r� r� r� r r� r� )r r2 r� r� r r r �test_check_owner_windows� s z-CheckPermissionsTest.test_check_owner_windowsr� c C sH dd l }|�� }t�d��}|d |_W d � n1 s:0 Y d S )Nr z os.getuidr@ )r �getuidr r� r� )r r� �uidZmock_uidr r r �test_check_owner_linux� s z+CheckPermissionsTest.test_check_owner_linuxc C sd t �d��}d|_W d � n1 s&0 Y t �d��}d|_W d � n1 sV0 Y d S )Nz$certbot.compat.filesystem.check_modeFz%certbot.compat.filesystem.check_owner)r r� r� )r Z mock_modeZ mock_ownerr r r �test_check_permissions� s z+CheckPermissionsTest.test_check_permissionsc C s. t �| jd� t �| jd� t �| jd� d S )N�� r i� r� r r r r �test_check_min_permissions� s z/CheckPermissionsTest.test_check_min_permissionsc C s t �| jd� t �| jd� d S )Nr� r r� r r r r �test_is_world_reachable� s z,CheckPermissionsTest.test_is_world_reachable)rN rO rP rQ r r� r� r� rg r� r� r� r� r� r� rR r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )� OsReplaceTestz3Test to ensure consistent behavior of rename methodc C sL t j�| jd�}t j�| jd�}t|d��� t|d��� t�||� dS )zKEnsure that replace will effectively rename src into dst for all platforms.r� r� ro N)r r r r rb rd r �replace)r r� r� r r r � test_os_replace_to_existing_file s z.OsReplaceTest.test_os_replace_to_existing_fileN)rN rO rP rQ r� r r r r r� s r� c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )� RealpathTestzTests for realpath methodc s t � �� t| j�| _d S r r r r r r r s zRealpathTest.setUpc C s� t �| j�| _tj�| jd�}t�| j|� t�� }tj�| jd�}tj� | j�}z.t� tj� | j�� t�||� W t� |� n t� |� 0 d S )NZlink_absZlink_rel) r �realpathr r r r r r �getcwd�basename�chdir�dirname)r r �curdirZ probe_namer r r r# s z$RealpathTest.test_symlink_resolutionc C s� t j�| jd�}t j�| jd�}t j�| jd�}t �||� t �||� t �||� tjtdd��}t� |� W d � n1 s�0 Y d S )NZlink1Zlink2Zlink3zlink1 is a loop!)�match) r r r r r rw rx � RuntimeErrorr r� )r Z link1_pathZ link2_pathZ link3_path�errorr r r � test_symlink_loop_mitigation0 s z)RealpathTest.test_symlink_loop_mitigation)rN rO rP rQ r r# r� rR r r r r r� s r� c @ sl e Zd ZdZdd� Ze�d�e�d�dd� ��Ze�d�e�d�dd � ��Ze�d�e�d�d d � ��Z d S ) �IsExecutableTestz Tests for is_executable methodc sx t j�| jd�}ddlm� d � fdd� }tjd|d��, t �t � |t j t j B d�� W d � n1 sj0 Y d S ) NZfoor ��_generate_daclc s. � | ||�}t d|�� �D ]}|�d� q|S r? )r7 r8 Z DeleteAce)Zuser_sidrZ �maskr( rK r� r r � _execute_mockK s z;IsExecutableTest.test_not_executable.<locals>._execute_mockz(certbot.compat.filesystem._generate_dacl)Z side_effecti� )N) r r r r Zcertbot.compat.filesystemr� r r� rd r rb rc �O_WRONLY)r Z file_pathr� r r� r �test_not_executable> s :z$IsExecutableTest.test_not_executablez(certbot.compat.filesystem.os.path.isfilez#certbot.compat.filesystem.os.accessc C s6 t � � d|_d|_W d � n1 s(0 Y d S �NT��_fix_windows_runtimer� �r Z mock_accessZ mock_isfiler r r �test_full_pathW s z IsExecutableTest.test_full_pathc C s6 t � � d|_d|_W d � n1 s(0 Y d S r� r� r� r r r � test_rel_path_ s z IsExecutableTest.test_rel_pathc C s6 t � � d|_d|_W d � n1 s(0 Y d S )NTFr� r� r r r �test_not_foundg s z IsExecutableTest.test_not_foundN) rN rO rP rQ r� r r� r� r� r� r r r r r� < s r� c @ sl e Zd Zejedd�e�d�dd� ��Zej edd�e�d�dd� ��Z ej edd�e�d�d d � ��Z d S ) � ReadlinkTestzTests specific to Linuxr z%certbot.compat.filesystem.os.readlinkc C s d|_ d S )Nz /normal/path�r� �r Z mock_readlinkr r r �test_path_posixq s z ReadlinkTest.test_path_posixzTests specific to Windowsc C s d|_ d S )Nz\\?\C:\short\pathr� r� r r r �test_normal_path_windowsw s z%ReadlinkTest.test_normal_path_windowsc C sF ddd |_ t�t�� t�d� W d � n1 s80 Y d S )Nz \\?\C:\longi� z\pathr� )r� rw rx � ValueErrorr �readlinkr� r r r �test_extended_path_windows~ s z'ReadlinkTest.test_extended_path_windowsN) rN rO rP r� r� rg r r� r� r� r� r� r r r r r� p s r� c c s^ t jdkrd V nHt�d��.} | jj}|jj}tj|_d V W d � n1 sP0 Y d S )N�ntr� ) r r� r r� r� r ZGetEffectiveRightsFromAclr< rA )r� Z dacl_mockZ mode_mockr r r r� � s r� c C s t �| t j�S r )r$ �GetFileSecurityZDACL_SECURITY_INFORMATION��targetr r r r � s r c C s t �| t j�S r )r$ r� �OWNER_SECURITY_INFORMATIONr� r r r rI � s rI c C s |� |d� t�| tj|� d S )NF)r� r$ ZSetFileSecurityr� )r� rL �userr r r rJ � s �rJ �probec C s2 t �| d� tj�| |�}tj|ddd��� |S )Nr� ro )r )r r r r r r r� rd )r r� r r r r r � s r �__main__r@ )r� )4rQ � contextlib�errno�sysr� r rw Zcertbotr Zcertbot._internalr Zcertbot.compatr r Zcertbot.tests.utilZtestsZ test_utilr r< rH r$ rg � ImportErrorr& r5 r6 r� r rS rf ri r� r� r� r� r� r� r� r� ZTestCaser� �contextmanagerr� r rI rJ r rN �exit�main�argv�__file__r r r r �<module> s\ w3 F (IC)4
.
Edit
..
Edit
__init__.cpython-39.opt-1.pyc
Edit
__init__.cpython-39.pyc
Edit
filesystem_test.cpython-39.opt-1.pyc
Edit
filesystem_test.cpython-39.pyc
Edit
misc_test.cpython-39.opt-1.pyc
Edit
misc_test.cpython-39.pyc
Edit
os_test.cpython-39.opt-1.pyc
Edit
os_test.cpython-39.pyc
Edit