← Back
Editing: parser_obj_test.cpython-38.pyc
U �]j+ � @ s� d Z ddlZddlZddlmZ ddlmZ G dd� dej�ZG dd� dej�ZG d d � d ej�Z G d d � d ej�Z G d d� dej�Z e dkr�e� � dS )z2 Tests for functions and classes in parser_obj.py � N)� parse_raw)� COMMENT_BLOCKc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�CommentHelpersTestc C sT ddl m} | �|tdg��� | �|tddg��� | �|tddddg��� d S ) Nr )� _is_comment�#z literally anything else�not�even�a�comment)�certbot_nginx.parser_objr � assertTruer � assertFalse)�selfr � r �E/usr/lib/python3/dist-packages/certbot_nginx/tests/parser_obj_test.py�test_is_comment s z"CommentHelpersTest.test_is_commentc C sj ddl m} | �|tt��� | �|tddg��� | �|tdddg��� | �|tddd d g��� d S ) Nr )�_is_certbot_commentr z not a certbot commentz managed by Certbotz also not a certbot commentr r r r )r r r r r r )r r r r r �test_is_certbot_comment s � � ��z*CommentHelpersTest.test_is_certbot_commentc C sp ddl m}m} |d �}| �||�� | �|�� t� | �|�d�dgt � | �|d d��d�dgt � d S )Nr )�_certbot_commentr Tz � � )r r r r � assertEqual�dumpr )r r r r r r r �test_certbot_comment s �z'CommentHelpersTest.test_certbot_commentN)�__name__� __module__� __qualname__r r r r r r r r s r c @ s> e Zd Zdd� Zdd� Zdd� Ze�d�dd � �Zd d � Z d S ) �ParsingHooksTestc C s~ ddl m} | �|�g �� | �|�dg�� | �|�dg�� | �|�ddg�� | �|�g g�� | �|�dg g�� d S )Nr ��Sentence� Zword�two�words)r r r � should_parser �r r r r r �test_is_sentence% s z!ParsingHooksTest.test_is_sentencec C s� ddl m} | �|�g �� | �|�dg�� | �|�ddg�� | �|�g gg g�� | �|�dgdg gg g�� | �|�dgdg�� | �|�dgdg gg�� | �|�d gg g�� | �|�dgd gd gd gg�� | �|�d ddgdgg�� d S )Nr ��Blockr r! r" Z block_name�hi�lol�helloZmany� statements�here�if� z (whatever))r r' r r# r �r r' r r r � test_is_block. s z ParsingHooksTest.test_is_blockc C s` t �� }dd� |_t �� }dd� |_tg � |�� |�� |�� tg � |�� |�� d S )Nc S s dS �NTr ��xr r r �<lambda>= � z1ParsingHooksTest.test_parse_raw.<locals>.<lambda>c S s dS �NFr r2 r r r r4 ? r5 )�mock�Mockr# r Z called_onceZ not_calledZ reset_mock)r � fake_parser1Z fake_parser2r r r �test_parse_raw; s z ParsingHooksTest.test_parse_rawz/certbot_nginx.parser_obj.Parsable.parsing_hooksc C sR ddl m} t�� }dd� |_|f|_| �|jtg � t � |_| �|jtg � d S )Nr ��errorsc S s dS r6 r r2 r r r r4 N r5 z:ParsingHooksTest.test_parse_raw_no_match.<locals>.<lambda>) �certbotr< r7 r8 r# � return_value� assertRaises�MisconfigurationErrorr �tuple)r Z parsing_hooksr< r9 r r r �test_parse_raw_no_matchJ s z(ParsingHooksTest.test_parse_raw_no_matchc C sH t �� }dd� |_tg � |j�d g� tg dd� |j�d dg� d S )Nc S s dS r1 r r2 r r r r4 V r5 zCParsingHooksTest.test_parse_raw_passes_add_spaces.<locals>.<lambda>T�Z add_spaces)r7 r8 r# r �parseZ called_with)r r9 r r r � test_parse_raw_passes_add_spacesT s z1ParsingHooksTest.test_parse_raw_passes_add_spacesN) r r r r% r0 r: r7 ZpatchrB rE r r r r r $ s r c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� SentenceTestc C s ddl m} |d �| _d S )Nr r )r r �sentencer$ r r r �setUp] s zSentenceTest.setUpc C sP ddl m} | �|j| jjd� | �|j| jjg g� | �|j| jjdg� d S )Nr r; r) � )r= r< r? r@ rG rD �r r< r r r �$test_parse_bad_sentence_raises_errora s z1SentenceTest.test_parse_bad_sentence_raises_errorc C sf ddddddddg}| j �|� | �| j jdddg� | �| j �� dddg� | �| j �d�|� d S )Nz r* r. z r) ZspacesT)rG rD r r" r )r Z og_sentencer r r �&test_parse_sentence_words_hides_spacesg s z3SentenceTest.test_parse_sentence_words_hides_spacesc C sh | j jddgdd� | �| j �d�dddg� | j jddddgdd� | �| j �d�dddddg� d S ) Nr( ZthereTrC r. �oneZspaceZnone)rG rD r r �r r r r �#test_parse_sentence_with_add_spacesn s z0SentenceTest.test_parse_sentence_with_add_spacesc C sR dddgg}| j �dddddg� t| j �� �D ] \}}| �|�� || � q0d S )N�1�2�3r. )rG rD � enumerate�iterater r )r �expected�irG r r r � test_iteratet s zSentenceTest.test_iteratec C sN | j jddgdd� | j �� | �| j �d�d d� | j jddgdd� d S )N�tabsZplsTrC r � )rG rD �set_tabsr r rN r r r � test_set_tabsz s zSentenceTest.test_set_tabsc C s� | j �ddg� | �| j �� d� | j �ddg� | �| j �� d� | j �ddg� | �| j �� d� | j �ddg� | �| j �� d� d S ) NZnorX r z r z z z )rG rD r �get_tabsrN r r r � test_get_tabs� s zSentenceTest.test_get_tabsN) r r r rH rK rL rO rW r[ r] r r r r rF \ s rF c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� BlockTestc C sP ddl m} |d �| _ddg| _ddgddgdd gg| _| j�| j| jg� d S ) Nr r& �server�name�thingrP rQ �anotherrM )r r' �blocr` �contentsrD r/ r r r rH � s zBlockTest.setUpc C s^ | � | jt| j�� �� | j�� g| j }t| jjdd��D ] \}}| � || |�� � q<d S )NT)�expanded)r rc �nextrT r rd rS �r rU rV �elemr r r rW � s zBlockTest.test_iteratec s� ddl m� m� ddgddgg}t| jjd�fdd�d ��D ] \}}| �|| |�� � q<| �| jt| jjd� fd d�d ��� d S ) Nr �r' r ra rP rQ Tc s t | � �od| jkS )Nra )� isinstancer" r2 r r r r4 � r5 z.BlockTest.test_iterate_match.<locals>.<lambda>)re �matchc s t | � �od| jkS )Nr_ )rj �namesr2 r& r r r4 � r5 ) r r' r rS rc rT r r rf rg r ri r �test_iterate_match� s � �z BlockTest.test_iterate_matchc C s| dd l }| jj|� | j�| jgdd� | �| j�� | j| jg� | �| j�d�ddddgdddgddd gd dd ggg� d S ) Nr TrC r_ r. r` ra rP rQ rb rM )�copyrc rD r` rd r r )r rn r r r � test_parse_with_added_spaces� s ��z&BlockTest.test_parse_with_added_spacesc C sZ ddl m} | �|j| jjg gg gg� | �|j| jjdg� | �|j| jjddg� d S )Nr r; r) ZfakeZnews)r= r< r? r@ rc rD rJ r r r �test_bad_parse_raises_error� s z%BlockTest.test_bad_parse_raises_errorc C sr | j �� | �| j j�d�d d� | j j�d�d d� D ]}| �|d d� q:| �| j j�d�d d d� d S )NTr rY ���z � )rc rZ r rl r rd )r rh r r r r[ � s zBlockTest.test_set_tabsc C s* | j �ddgg g� | �| j �� d� d S )Nz r) � )rc rD r r\ rN r r r r] � s zBlockTest.test_get_tabsN) r r r rH rW rm ro rp r[ r] r r r r r^ � s r^ c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�StatementsTestc C sX ddl m} |d �| _ddgddgddgg| _ddd dgddd dgddd dgd g| _d S ) Nr )� StatementsrG rM r! �andrb z r. z )r ru r+ �rawZ raw_spaced)r ru r r r rH � s � �zStatementsTest.setUpc C sB | j �| j� | j �� | j �� D ]}| �|�d�d d� q"d S )NTr rY )r+ rD rw rZ rT r r �r Z statementr r r r[ � s z StatementsTest.test_set_tabsc C sr | j �| j� t�� | j _d| j jj_| j �� | j � � D ]}| � |� d�d d� q:| � | j � d�d d� d S )Nz Tr rY rq z ) r+ rD rw r7 r8 �parentr\ r> rZ rT r r rx r r r �test_set_tabs_with_parent� s z(StatementsTest.test_set_tabs_with_parentc C sT | j d �dd� | j�| j � | �| j�� d� | j�g � | �| j�� d� d S )Nr z rs r )rw �insertr+ rD r r\ rN r r r r] � s z StatementsTest.test_get_tabsc C s4 | j j| jdd� | �| j �d�d dddg� d S )NTrC r rG r. rM )r+ rD rw r r rN r r r ro � s z+StatementsTest.test_parse_with_added_spacesc C s$ ddl m} | �|j| jjd� d S )Nr r; zlol not a list)r= r< r? r@ r+ rD rJ r r r � test_parse_bad_list_raises_error� s z/StatementsTest.test_parse_bad_list_raises_errorc C sd | j �| jdg � | �t| j �� d t�� | �| j �d�d �� � | �| j �d�d d� d S )Nz rq T) r+ rD rw r rj r �list�isspacer rN r r r �$test_parse_hides_trailing_whitespace� s z3StatementsTest.test_parse_hides_trailing_whitespacec C sV | j �| j� ddgddgg}t| j jdd� d��D ] \}}| �|| |�� � q4d S )NrG rM r! c S s d| kS )NrG r r2 r r r r4 � r5 z-StatementsTest.test_iterate.<locals>.<lambda>)rk )r+ rD rw rS rT r r rg r r r rW � s zStatementsTest.test_iterateN) r r r rH r[ rz r] ro r| r rW r r r r rt � s rt �__main__)�__doc__Zunittestr7 r r r ZTestCaser r rF r^ rt r �mainr r r r �<module> s 8.7;
Save File
Cancel