Current Dir: /snap/core22/2437/usr/share/subiquity/console_conf/models/tests/__pycache__/
[FILE] test_systems.cpython-310.pyc [ edit | delete | rename ]
[FILE] __init__.cpython-310.pyc [ edit | delete | rename ]
Viewing: /snap/core22/2437/usr/share/subiquity/console_conf/models/tests/__pycache__/test_systems.cpython-310.pyc
o
K��^�$ � @ sZ d dl Z d dlZd dlZd dlmZ d dlmZmZmZm Z m
Z
m Z G dd� de j �Z
dS )� N)�StringIO)�RecoverySystemsModel�RecoverySystem�Brand� SystemModel� SystemAction�SelectedSystemActionc @ s� e Zd Zdddddddd�d dd
d �d d
d�ddd�gd�dddddd�dddd �d d
d�gd�giZdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Z d%d&� Z
d'd(� Z d)d*� Z d+S ),�RecoverySystemsModelTests�systemsT�1234�brand-id�brand-username�this is my brand�verified��id�username� display-name�
validation� core20-amd64�Core 20 AMD64 system��modelr r � reinstall�install��title�mode�recover��current�label�brandr �actions�other�other-brand-id� other-brand�my brand�unproven� my-brand-box� Funky box)r! r" r r# c C s� t �| j�}t�t|��}ttddtdddd�tdddd d
�t d d d
�t ddd
�gd�tddtdddd�tddddd
�t d d d
�gd�g�}| �
|j |j � | �
|j |j d � d S )NTr r r r )r �brand_id� display_namer
r r )�IDr r, r r r r r )r r! r r" r# Fr$ r) r% r* r& r' r( r )
�json�dumps� referencer �from_systems_streamr r r r r � assertEqualr
r )�self�rawr
�exp� r6 �>/usr/share/subiquity/console_conf/models/tests/test_systems.py� test_from_systems_stream_happyJ sP ��
���� �� z8RecoverySystemsModelTests.test_from_systems_stream_happyc C s@ | � tj�� t�td�� W d � d S 1 sw Y d S )Nz{})� assertRaises�
jsonschema�ValidationErrorr r1 r )r3 r6 r6 r7 �&test_from_systems_stream_invalid_emptyo s "�z@RecoverySystemsModelTests.test_from_systems_stream_invalid_emptyc C sz t �dddddd�dddd �d
d d �d
d
d �gd�gi�}| �tj�� t�t|�� W d � d S 1 s6w Y d S )Nr
r r
r r r r r r r r r r )r" r r# �r. r/ r9 r: r; r r1 r �r3 r4 r6 r6 r7 �5test_from_systems_stream_invalid_missing_system_labels s( ������"�zORecoverySystemsModelTests.test_from_systems_stream_invalid_missing_system_labelc C sp t �dddddd�ddd �d
d
d �gd �gi�}| �tj�� t�t|�� W d � d S 1 s1w Y d S ) Nr
r r r r r r r r r )r! r r# r= r> r6 r6 r7 �.test_from_systems_stream_invalid_missing_brand� s �����"�zHRecoverySystemsModelTests.test_from_systems_stream_invalid_missing_brandc C sr t �ddddddd�dd d
�d d d
�gd �gi�}| �tj�� t�t|�� W d � d S 1 s2w Y d S )
Nr
r r r
r r r r r r r )r! r" r# r= r> r6 r6 r7 �.test_from_systems_stream_invalid_missing_model� s" �����"�zHRecoverySystemsModelTests.test_from_systems_stream_invalid_missing_modelc
C s> t �dddddd�dddd d
�g d �gi�}t�t|�� d S ) Nr
r r r r r r
r r r )r! r r" r# )r. r/ r r1 r r> r6 r6 r7 �)test_from_systems_stream_valid_no_actions� s" �����zCRecoverySystemsModelTests.test_from_systems_stream_valid_no_actionsc C sx t �| j�}t�t|��}|�|jd |jd jd � | � |j
t |jd |jd jd d�� |� � | �
|j
� d S )N� r )�system�action)r. r/ r0 r r1 r �selectr
r# r2 � selectionr �unselect� assertIsNone)r3 r4 r r6 r6 r7 �test_selection� s ��z(RecoverySystemsModelTests.test_selectionc C sv t �| j�}t�t|��}|�|jd |jd jd � t� }t� |j
|� t � |� � �}| �
|dddd�d�� d S )NrC r r$ r r )r r )r! rE )r. r/ r0 r r1 r rF r
r# �to_response_streamrG �loads�getvaluer2 )r3 r4 r �stream�fromjsonr6 r6 r7 �test_to_response_stream� s � �z1RecoverySystemsModelTests.test_to_response_streamc C s| dddddddd�d dd
d �d d
d�ddd�gd�gi}t �|d �}| �t|j�d� | �|jd jd� | �|j� d S )Nr
Fr r r
r r r r r r r r r r r rC r )r � from_systemsr2 �lenr
r! rI r )r3 r0 r
r6 r6 r7 �test_no_current� s, ������z)RecoverySystemsModelTests.test_no_currentN)
�__name__�
__module__� __qualname__r0 r8 r< r? r@ rA rB rJ rP rS r6 r6 r6 r7 r sR ����������*% r )�unittestr. r: �ior �console_conf.models.systemsr r r r r r �TestCaser r6 r6 r6 r7 �<module> s