File: //snap/core20/current/usr/share/subiquity/subiquitycore/tests/__pycache__/__init__.cpython-38.pyc
U
K��^^ � @ sH d dl Z d dlZd dlZd dlZd dlmZ G dd� de�Zdd� ZdS )� N)�TestCasec @ s e Zd Zddd�Zddd�ZdS ) � SubiTestCaseNTc C sB |d kr t jd| jj d�}n t j|d�}| �t�tj|�� |S )Nz
subiquity-%s.)�prefix)�dir) �tempfileZmkdtemp� __class__�__name__Z
addCleanup� functools�partial�shutilZrmtree)�selfr ZcleanupZtmpd� r
�4/usr/share/subiquity/subiquitycore/tests/__init__.py�tmp_dir
s
� zSubiTestCase.tmp_dirc C s. |d kr| � � }tj�tj�tj�||���S )N)r �os�path�normpath�abspath�join)r r r r
r
r �tmp_path s zSubiTestCase.tmp_path)NT)N)r �
__module__� __qualname__r r r
r
r
r r s
r c
C s� t j�| �st �| � g }|�� D ]�\}}t jj�| |g�}t j�t j�|��sbt �t j�|�� t |d��4}t
|t �r�|� |� n|� |�
d�� |�� W 5 Q R X |�|� q"|S )N�wbzutf-8)r r �exists�makedirs�items�sepr �isdir�dirname�open�
isinstance�bytes�write�encode�close�append)r �filesZret�nameZcontent�p�fpr
r
r � populate_dir s
r* )r r r r Zunittestr r r* r
r
r
r �<module> s