← Back
Editing: nginxparser_test.cpython-38.pyc
U �]AA � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZm Z m Z m Z m Z m Z ddlmZ e�d�ZG dd� dej�ZG dd � d ej�Zed kr�e�� dS ) z#Test for certbot_nginx.nginxparser.� N)�ParseException)�RawNginxParser�loads�load�dumps�dump� UnspacedList)�utilc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&S )'�TestRawNginxParserz+Test the raw low-level Nginx config parser.c C sP t j�d��� }| �|dddg� t j�d��� }| �|dddgdddg� d S )Nz root /test;�root� z/testzroot /test;foo bar;�foo�bar)r Z assignment� parseString�asList� assertEqual��self�parsed� r �F/usr/lib/python3/dist-packages/certbot_nginx/tests/nginxparser_test.py�test_assignments s z#TestRawNginxParser.test_assignmentsc C s� t j�d��� }| �|ddgg g� t j�d��� }| �|dddgg g� t j�d��� }| �|ddgdddddgdgg� d S ) Nzfoo {}r r zlocation /foo{}�locationz/foozfoo { bar foo ; }r )r �blockr r r r r r r � test_blocks s z TestRawNginxParser.test_blocksc C sF t j�d��� }|\}}| �t|�dddgg g� | �t|�d� d S )Nzfoo { bar {} }r r r )r r r r r �FIRST)r r r Zcontentr r r �test_nested_blocks" s z%TestRawNginxParser.test_nested_blocksc C s� t tdddgdddgddddgdd dd gdd dd gd ddddgdddgdddgddddgggggggg��}| �|�d�d�d�� d S )N�userr �www-data� �server� �listenZ80� server_namezfoo.comr �/home/ubuntu/sites/foo/z r �/statusz � check_status� z �typesz � image/jpeg�jpgz�user www-data; server { listen 80; server_name foo.com; root /home/ubuntu/sites/foo/; location /status { check_status; types { image/jpeg jpg;}}})r r r �split)r Zdumpedr r r �test_dump_as_string( s& ���� ��z&TestRawNginxParser.test_dump_as_stringc C s� t t�d���}t�t|��}W 5 Q R X | �|ddgdgdgdddd gd d d gd dgddgdgddgggggdddgddgd dgggdddgg gddd gg gdd d gg gggggg� d S ) N�foo.confr r �httpr r" z*:80Zdefault_server�sslr# z *.www.foo.comz*.www.example.comr r$ r r% r( r) r* �~zcase_sensitive\.php$�indexz index.phpz /var/root�~*zcase_insensitive\.php$�=zexact_match\.php$z^~zignore_regex\.php$��openr �get_data_filenameZfilter_commentsr r �r �handler r r r �test_parse_from_fileC s0 �� ����z'TestRawNginxParser.test_parse_from_filec C s� t t�d���}t�t|��}W 5 Q R X | �|dgddgggdgddgdddgd d d d gd ddgggggggdgddgdd dgd ddgd ddddgddgddd gggggg� d S ) Nzedge_cases.confr r# Zsimplezwith.ifr r0 z^/services/.+$�ifz($request_filenamer2 z\.(ttf|woff)$)� add_headerzAccess-Control-Allow-Originz"*"zwith.complicated.headersz\.(?:gif|jpe?g|png)$ZPragmaZpublic� Cache-Controlz+'public, must-revalidate, proxy-revalidate'z "test,;{}"r Zblahz "hello;world"� try_filesz$uriz @rewritesr4 r7 r r r �test_parse_from_file2[ s@ ���� �������z(TestRawNginxParser.test_parse_from_file2c C s^ t t�d���}t�t|��}W 5 Q R X | �|dgdgddgddgdd gggggggg� d S ) Nzmultiline_quotes.confr. r r" z*:443r �/Zbody_filter_by_luaz�'ngx.ctx.buffered = (ngx.ctx.buffered or "") .. string.sub(ngx.arg[1], 1, 1000) if ngx.arg[2] then ngx.var.resp_body = ngx.ctx.buffered end'r4 r7 r r r �test_parse_from_file3p s ������z(TestRawNginxParser.test_parse_from_file3c C s, t t�d���}| �tt|� W 5 Q R X d S )Nz broken.conf)r5 r r6 � assertRaisesr r )r r8 r r r �test_abort_on_parse_failure� s z.TestRawNginxParser.test_abort_on_parse_failurec C s� t t�d���}t|�}W 5 Q R X |d d �tdgdddddgddd gd dd gd dd gdddgdddgdddgdddgdddgdddddggggg�� tjdd ��"}t||� |� d � t|�}W 5 Q R X | � ||� d S ) Nz nginx.conf���r r" r Z443r/ r# Z localhostZssl_certificatezcert.pemZssl_certificate_keyzcert.keyZssl_session_cachez shared:SSL:1mZssl_session_timeoutZ5mZ ssl_cipherszHIGH:!aNULL:!MD5r r? r Zhtmlr1 z index.htmlz index.htm�w+t��moder ) r5 r r6 r �appendr �tempfile� TemporaryFiler �seekr �r r8 r �fZ parsed_newr r r �test_dump_as_file� s, ���� z$TestRawNginxParser.test_dump_as_filec C s� t t�d���}t|�}W 5 Q R X tjdd��"}t||� |�d� t|�}W 5 Q R X | �||� | �|ddgddgdd gd gd gdd gdd gdd gddgddgggg� d S )Nzminimalistic_comments.confrD rE r �#z$ Use bar.conf when it's a full moon!Zincluder- z Kilroy was herer& r r' z' Don't forget to open up your firewall!r" Z1234z listen 80;) r5 r r6 r rH rI r rJ r rK r r r � test_comments� s* ���z TestRawNginxParser.test_commentsc C s. t d�}| �|ddddgdddgggg� d S ) Nz1if ($http_accept ~* "webp") { set $webp "true"; }r: z ($http_acceptr2 z"webp")�setz$webpz"true"�r r r r r r �test_issue_518� s ��z!TestRawNginxParser.test_issue_518c C s& t d�}| �|dgddgggg� d S )Nz&http { # server{ }r. rN z server{rQ r r r r �test_comment_in_block� s ��z(TestRawNginxParser.test_comment_in_blockc C s t d�}| �|dddgg� d S )NzYaccess_log syslog:server=unix:/dev/log,facility=auth,tag=nginx_post,severity=info custom;Z access_logzFsyslog:server=unix:/dev/log,facility=auth,tag=nginx_post,severity=infoZcustomrQ r r r r �test_access_log� s ��z"TestRawNginxParser.test_access_logc C s t d�}| �|dddgg� d S )NzEadd_header Cache-Control no-cache,no-store,must-revalidate,max-age=0;r; r< z+no-cache,no-store,must-revalidate,max-age=0rQ r r r r �test_add_header� s �z"TestRawNginxParser.test_add_headerc C sL d}t |�}| �|dgdddgddgdd gd d gd d gggd gggg� d S )Nz�http { map $http_upgrade $connection_upgrade { default upgrade; '' close; "~Opera Mini" 1; *.example.com 1; } one; }r. �mapz $http_upgradez$connection_upgrade�defaultZupgradez''�closez "~Opera Mini"�1z *.example.comZonerQ )r Ztest_strr r r r �!test_map_then_assignment_in_block� s ���z4TestRawNginxParser.test_map_then_assignment_in_blockc C s t d�}| �|dddgg� d S )NzQtry_files /typo3temp/tx_ncstaticfilecache/$host${request_uri}index.html @nocache;r= z=/typo3temp/tx_ncstaticfilecache/$host${request_uri}index.htmlz@nocacherQ r r r r �test_variable_name� s ��z%TestRawNginxParser.test_variable_namec C s� d}t |�}| �|ddddgdddd gggdd d d gd ddgggddddgddgggddgddgggddddgdddgggddd gd d gggd d d!gg� d S )"Na� if ($http_user_agent ~ MSIE) { rewrite ^(.*)$ /msie/$1 break; } if ($http_cookie ~* "id=([^;]+)(?:;|$)") { set $id $1; } if ($request_method = POST) { return 405; } if ($request_method) { return 403; } if ($args ~ post=140){ rewrite ^ http://example.com/; } location ~ ^/users/(.+\.(?:gif|jpe?g|png))$ { alias /data/w3/images/$1; } proxy_set_header X-Origin-URI ${scheme}://${http_host}/$request_uri; r: z($http_user_agentr0 zMSIE)Zrewritez^(.*)$z/msie/$1�break� ($http_cookier2 �"id=([^;]+)(?:;|$)")rP z$idz$1z($request_methodr3 zPOST)�returnZ405z($request_method)Z403z($argsz post=140)�^zhttp://example.com/r z ^/users/(.+\.(?:gif|jpe?g|png))$�aliasz/data/w3/images/$1Zproxy_set_headerz X-Origin-URIz%${scheme}://${http_host}/$request_urirQ )r �testr r r r �test_weird_blocks� s$ �� ���z$TestRawNginxParser.test_weird_blocksc C s" t d�}| �|dgddgg� t d�}| �|dddd gdd gggg� d }t |�}| �|d gd gdgdgdgdgdgdgdgg � | �tt d� | �tt d� d}t |�}| �|ddddd gddddd gg� t d �}| �|d d d d!d"gg� | �tt d#� | �t d$�d%gg� | �tt d&� d S )'Nz"hello\""; # blah "heh heh"z "hello\""rN z blah "heh heh"zAif ($http_cookie ~* "id=([^;]+)(?:;|$)") { # blah ) }r: r] r2 r^ z blah )z� one"test"; ("two"); "test")red; "test")"blue"; "test")"three; (one"test")one; one"; one"test; one"test"one; z one"test"z("two")z "test")redz "test")"blue"z "test")"threez(one"test")onezone"zone"testz one"test"onez "test"one;z"test;z� server_name foo.example.com bar.example.com baz.example.com qux.example.com; server_name foo.example.com bar.example.com baz.example.com qux.example.com; r# zfoo.example.comzbar.example.comzbaz.example.comzqux.example.comz directive $var ${var} $ ${};Z directivez$varz${var}�$z${}z server {server_name test.com};zblag${dfgdfg};z blag${dfgdfg}zblag${dfgdf{g};)r r rA r )r r rb r r r �test_edge_cases" sL � � � ��z"TestRawNginxParser.test_edge_casesN)�__name__� __module__� __qualname__�__doc__r r r r, r9 r> r@ rB rM rO rR rS rT rU rZ r[ rc re r r r r r s& )r c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dS )�TestUnspacedListz$Test the UnspacedList data structurec C sT ddddg| _ ddg| _| j d d � | _| jd d � | _t| j�| _t| j�| _d S )Nr! �thingsr �quirk�y)�a�b�l�l2r �ul�ul2�r r r r �setUpa s zTestUnspacedList.setUpc C s& | � | jddg� | � | jdg� d S �Nrk rl rm )r rr rs rt r r r �test_constructioni s z"TestUnspacedList.test_constructionc C sB t �| j�}|�d� | �|dddg� | �|j| jdg � d S )NZwiserk rl )�copy�deepcopyrr rG r �spacedrn �r �ul3r r r � test_appendm s z TestUnspacedList.test_appendc C s~ | j | j }| �|dddg� | �|j| j| j � | �| j j| j� | j | j }| �|dddg� | �|j| j| j � d S rv )rr rs r rz rn ro rq r{ r r r �test_adds s zTestUnspacedList.test_addc C sV t �| j�}|�| j� | �|dddg� | �|j| j| j � | �| jj| j� d S rv ) rx ry rr �extendrs r rz rn ro r{ r r r � test_extend| s z TestUnspacedList.test_extendc C s^ t �| j�}d|d< dddg}t|�|d< | �|dddgg� | �|j| jd dd|g� d S )NZzitherr z ZzatherZzest� r )rx ry rr r r rz rn )r r| rp r r r �test_set� s zTestUnspacedList.test_setc C s( | � t| jjd� | � t| jjd� d S )N� ���)rA � IndexErrorrs � __getitem__rt r r r �test_get� s zTestUnspacedList.test_getc C s� t ddddgddddgddddgdddd gd ddd dd gg�}|�d d� | �|ddgddgddgdd gdd d gdg� | �|jddddgddddgddddgdddd gd ddd dd gdg� d S )Nr! r" z z69.50.225.155:9000z 127.0.0.1r# r z .example.comz example.*r Z5001r/ � ZFROGZ)r �insertr rz )r �xr r r � test_insert� s8 �� �� ��z TestUnspacedList.test_insertc C sP t �| j�}|�dd� |�d� |�ddg� |d= | �|dddddg� d S )Nr ZsomeZwhyZdidZwhetherr� rk )rx ry rr r� rG r r r{ r r r � test_rawlists� s z TestUnspacedList.test_rawlistsc C s� | � d| j�� � tg �}|�| j� | � d| j�� � | � d|�� � tdgdddgg�}| � d|�� � d|d d< | � d|�� � d S )NFTr� r� � � r� )r rs Zis_dirtyr rG rr )r r| Zul4r r r � test_is_dirty� s z TestUnspacedList.test_is_dirtyN)rf rg rh ri ru rw r} r~ r� r� r� r� r� r� r r r r rj _ s rj �__main__)ri rx �operatorrH ZunittestZ pyparsingr Zcertbot_nginx.nginxparserr r r r r r Zcertbot_nginx.testsr � itemgetterr ZTestCaser rj rf �mainr r r r �<module> s PX
Save File
Cancel