← Back
Editing: parser_test.cpython-38.pyc
U �]�Y � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m Z ddl m Z ddl m Z ddl mZ dd lmZ G d d � d ej�Zed kr�e�� dS ) z Tests for certbot_nginx.parser.� N)�List)�errors)�os)� nginxparser)�obj)�parser)�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&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0S )1�NginxParserTestzNginx Parser Test.c C s( t �| j� t �| j� t �| j� d S �N)�shutilZrmtree�temp_dirZ config_dirZwork_dir��self� r �A/usr/lib/python3/dist-packages/certbot_nginx/tests/parser_test.py�tearDown s zNginxParserTest.tearDownc C s. t j�| jd�}t�|�}| �|j| j� d S )Nz*etc_nginx/////ubuntu_nginx/../../etc_nginx) r �path�joinr r � NginxParser� assertEqual�root� config_path)r r �nparserr r r �test_root_normalized s z$NginxParserTest.test_root_normalizedc C sN t �� }z4t �| j� t�t j�| j��}| � |j | j� W 5 t �|� X d S r ) r �getcwd�chdirr r r r �relpathr r r )r Zcurr_dirr r r r �test_root_absolute s z"NginxParserTest.test_root_absolutec C s( t �| jtjj �}| �|j| j� d S r )r r r r r �sepr r �r r r r r �test_root_no_trailing_slash, s z+NginxParserTest.test_root_no_trailing_slashc s� t �| j�� � �� | �t� fdd�dD ��t� j�� �� | �ddddgg� j� �d� � | �d gd d gd d gdd gddgggg� j� �d� � dS )z+Test recursive conf file parsing. c s g | ]}� � |��qS r )�abs_path��.0�x�r r r � <listcomp>6 s z-NginxParserTest.test_load.<locals>.<listcomp>) �foo.conf� nginx.conf� server.conf�sites-enabled/default�sites-enabled/example.comzsites-enabled/headers.comzsites-enabled/migration.comzsites-enabled/sslon.com�sites-enabled/globalssl.comzsites-enabled/ipv6.comzsites-enabled/ipv6ssl.com� server_name�somename�alias� another.aliasr) �server�listen�69.50.225.155:9000� 127.0.0.1� .example.com� example.*r+ N) r r r �loadr �set�parsed�keysr! r r r% r � test_load0 s& � � ����zNginxParserTest.test_loadc C s| t �| j�}tjdkrH| �d|�d�� | �tj�| jd�|�d�� n0| �d|�d�� | �tj�| jd�|�d�� d S )N�ntz /etc/nginx/*zfoo/barzC:\etc\nginx\*zfoo\bar) r r r r �namer r! r r r r r r � test_abs_pathJ s ��z NginxParserTest.test_abs_pathc C s� t �| j�}|jddd� |�|�d��}| �dtt�|�d���� | �dtt�|�d���� | �d gd d gd d gd dgd dgggg|d � d S )NZtestF)Zlazyz sites-enabled/example.com.test� z*.test� zsites-enabled/*.testr1 r2 r3 r4 r- r5 r6 r ) r r r �filedumpZ _parse_filesr! r �len�glob)r r r9 r r r � test_filedumpV s � ���z NginxParserTest.test_filedumpc C sz dgdgdggdgdggfdgdgdggdggfdgdgdggdggfg g fdg fdgdgdggdgdgdgggddgddgddgddggfdgdgdggdgdgdgggddgddgddggfdgdgdggdgdgdgggddgddggfdgdgdggdgdgdgggddggfdgdgdggdgdgdgggg fg }|D ]4\}}g }t �|dd� |fd d�� | �||� �q@d S ) N� r? r � � � c S s" t | t�o t| �dko | d dkS )NrG r rE )� isinstance�listrB )r$ r r r �<lambda>v s �z7NginxParserTest.test__do_for_subarray.<locals>.<lambda>c S s |� |�S r )�append)r$ �yZptsr r r rK y � )r Z_do_for_subarrayr )r ZmylistsZmylist�result�pathsr r r �test__do_for_subarraye s( �4.("� �z%NginxParserTest.test__do_for_subarrayc C sl t �| j�}|�� }t�|�d�t�dddddd�gddtdg�g dg�}dd � |D �d }| � ||� d S ) Nr, z4.8.2.6Z57TF� globalssl.comr c S s g | ]}d |j kr|�qS )rR ��filepr" r r r r& � s z>NginxParserTest.test_get_vhosts_global_ssl.<locals>.<listcomp>) r r r � get_vhostsr � VirtualHostr! �Addrr8 r )r r �vhostsZvhostZ globalssl_comr r r �test_get_vhosts_global_ssl| s � �z*NginxParserTest.test_get_vhosts_global_sslc C s t �| j�}|�� }t�|�d�t�dddddd�gddtddg�g dd d g�}t�|�d�t�d ddddd�t�dd dddd�gddtd d dg�g dd dg�}t�|�d�t�dddddd�t�dddddd�gddtddg�g dg�}t�|�d�t�dddddd�t�dddddd�gddtdg�g dg�}t�|�d�t�d d dddd�gddtd d g�g d d dg�}| � d!t |�� d"d#� |D �d }| � ||� d$d#� |D �d } | � || � d%d#� |D �d } | � || � d&d#� |D �d } | � || � d'd#� |D �d } | � || � d S )(Nr( � Z8080FT� localhost�~^(www\.)?(example|bar)\.� rG � r. Z8000r0 r/ � r+ z 69.50.225.155Z9000r4 r5 r6 r r* ZmyhostZ otherhost�www.example.orgr' �*�80� *.www.foo.com�*.www.example.comrE � c S s g | ]}d |j kr|�qS )� example.comrS r" r r r r& � s z3NginxParserTest.test_get_vhosts.<locals>.<listcomp>c S s g | ]}d |j kr|�qS ��defaultrS r" r r r r& � s c S s g | ]}d |j kr|�qS )r' rS r" r r r r& � s c S s g | ]}d |j kr|�qS )r[ ��namesr" r r r r& � s c S s g | ]}d |j kr|�qS )r. ri r" r r r r& � s ) r r r rU r rV r! rW r8 r rB ) r r rX Zvhost1Zvhost2Zvhost3Zvhost4Zvhost5� example_comrh Zfooconfr[ r. r r r �test_get_vhosts� s� � � � � �� � � �� � � �� � � � � z NginxParserTest.test_get_vhostsc C s� t �| j�}t�d d d d d ddgddgddgddgd d ggggd �}| �|�|�� dd d d gdddgddgg|_| �|�|�� ddgdddgg|_| �|�|�� ddgd dgdddgg|_| �|�|�� d S )Nr2 zmyhost default_serverr- r` �location�/r Zhtml�indexzindex.html index.htmz*:80Zdefault_server�sslrc rd z/home/ubuntu/sites/foo/z80 sslrb �on) r r r r rV � assertFalseZhas_ssl_on_directive�raw� assertTrue)r r � mock_vhostr r r �test_has_ssl_on_directive� s. �� ���z)NginxParserTest.test_has_ssl_on_directivec C s� t �| j�}t�|�d�d d d tddg�d dddg�}|�d�}tdd g�}||_||_d g|_ |� |d d gd dgg� |� |d � |� |d � | � |j | dgddgddgddgdd gg ggg� d S )Nr( r[ r\ r] rG r^ r+ r5 r6 r �foo�bar�ssl_certificate�/etc/ssl/cert2.pemr1 r2 r3 r4 r- )r r r r rV r! r8 rT rj r �add_server_directivesZremove_server_directivesr r9 )r r ru rk rj r r r � test_remove_server_directives� s> � � �� ��z-NginxParserTest.test_remove_server_directivesc C s� t �| j�}t�|�d�d d d tddg�d dddg�}|�|ddgd d d d gg� t� d �}t � |j |�d� �}| � dtt�||��� |�d�}tddg�}||_||_dg|_|�|ddgd dgg� |�|ddgg� ddlm} | � |j | dgddgddgddgddgddgd|gd dgd|gg g g gg� |�d�}tdd d g�}||_||_g |_| �tj|j|ddgd dgg� d S ) Nr( r[ r\ r] rG r^ rw rx z ry � z/etc/ssl/cert.pemz&\n\s+ssl_certificate /etc/ssl/cert.pemr+ r5 r6 r rz ��COMMENTr1 r2 r3 r4 r- �#r) r/ r0 r. )r r r r rV r! r8 r{ �re�compiler �dumpsr9 r rB �findallrT rj r �certbot_nginx.parserr Z assertRaisesr ZMisconfigurationError) r r ru Zssl_re�dumprk rj r Z server_confr r r �test_add_server_directives� sp � � �� �� �� ��z*NginxParserTest.test_add_server_directivesc C s� t �| j�}|�d�}t�|d d d tddg�d dg�}|�|ddddgg� |�|dd d|�d �gg� dd lm } | � |j | d gd dgd dgddgddgdddgg d |�d �gd|gg g gg� d S )Nr+ r5 r6 r z r� r} zwhat a nice commentZincludezcomment_in_file.confr~ r1 r2 r3 r4 r- ) r r r r! r rV r8 r{ r� r r r9 )r r rk ru r r r r �test_comment_is_repeatable s> � ��� ��z*NginxParserTest.test_comment_is_repeatablec C s� t �| j�}tddg�}|�d�}t�|d d d |d dg�}|�|ddgg� ddlm } | � |j | dgd d gd d gddgd |gddgg ggg� tddg�|_ |�|d dgg� | � |j | dgd d gd d gddgd |gddgg d dgd |gg g gg� d S )Nr5 r6 r+ r r- z foobar.comr~ r1 r2 r3 r4 r� ry zcert.pem) r r r r8 r! r rV Z update_or_add_server_directivesr� r r r9 rj )r r �targetrT ru r r r r � test_replace_server_directives0 sJ � �� � ��z.NginxParserTest.test_replace_server_directivesc C s� d}t dddg�t dddg�t ddg�t d dg�t d d d g�t d ddg�t ddg�t dddg�t ddd g�t d dg�t dg�t g �t d g�g }ddddddddddd ddg }t|�D ] \}}| �|t�||| �� q�d S ) N� www.eff.orgz irrelevant.long.name.eff.org�*.orgzeff.orgz ww2.eff.orgztest.www.eff.org� *.eff.org� .www.eff.org�.eff.orgzwww.eff.� www.eff.*z *.www.eff.orgrf �~^(www\.)?(eff.+)z*.eff.*ra �www.*z .test.eff.org)�exactr� )NN)r� r� )�wildcard_startr� )� wildcard_endr� )Zregexr� )r� ra )r� r� )r� r� )r� r� )r8 � enumerater r Zget_best_match)r Z target_namerj Zwinners�i�winnerr r r �test_get_best_matchK sB � ��z#NginxParserTest.test_get_best_matchc C s� t �dddddgdddgddddgg�}d d lm}m} ||d � ||d � | �|jdddddg|ddddg|ddddgg� d S ) N� �ar} �b�c�d�e�fr )�comment_directive� COMMENT_BLOCKrG )r Z UnspacedListr� r� r� r �spaced)r �blockr� r� r r r �test_comment_directivel s � �z&NginxParserTest.test_comment_directivec C s� t �d�}|d d }ddlm} ||dd� ||dd� ||d d � | �|jd d d dgd dd dgd dd dgddd dgd ddgd ddgdddgdg� d S )Na4 server { listen 80; root /var/www/html; index star.html; server_name *.functorkitten.xyz; ssl_session_timeout 1440m; ssl_protocols TLSv1 TLSv1.1 TLSv1.2; ssl_prefer_server_ciphers on; }r rG )�_comment_out_directiverF Zblah1rH Zblah2� Zblah3z r2 r} rb r z /var/www/htmlro z star.htmlz r- z*.functorkitten.xyzr� z1 ssl_session_timeout 1440m; # duplicated in blah1z; ssl_protocols TLSv1 TLSv1.1 TLSv1.2; # duplicated in blah2z4 ssl_prefer_server_ciphers on; # duplicated in blah3z )r �loadsr� r� r r� )r Z server_blockr� r� r r r �test_comment_out_directive} s �z*NginxParserTest.test_comment_out_directivec C s� t �ddgg�}| �|d � t �dddgg�}| �|d � t �ddgddgg�}| �|d � t �ddgddgg�}| �|d � d S )Nr2 �443rp Zoffrq )r �_parse_server_rawrr rt �r r1 r r r �test_parse_server_raw_ssl� s$ �� � �z)NginxParserTest.test_parse_server_raw_sslc C s( t �ddgg�}| �t|d �d� d S )Nr2 zunix:/var/run/nginx.sock�addrsr )r r� r rB r� r r r �test_parse_server_raw_unix� s �z*NginxParserTest.test_parse_server_raw_unixc C s. t �| j�}|�ddgg�}| �|d � d S )Nr2 r� rp )r r r Z parse_serverrt )r r r1 r r r �$test_parse_server_global_ssl_applied� s �z4NginxParserTest.test_parse_server_global_ssl_appliedc C s t �| j�}|�� }dd� |D �d }|j|dd�}|jdd� | �tt|j ��j � | � |j |j � t �| j�}|�� }dd� |D �}| � t|�d � |d }| �tt|j ��j � | � tt|j��tt|j��� | � t|j�t|j�� | �tt|j ���tt|j ���� d S ) Nc S s g | ]}d |j kr|�qS rg rS r" r r r r& � s z8NginxParserTest.test_duplicate_vhost.<locals>.<listcomp>r T�Z remove_singleton_listen_paramsrZ �Zextc S s g | ]}d |j kr|�qS rg rS r" r r r r& � s rE rG )r r r rU �duplicate_vhostrA rr �next�iterr� rh ZassertNotEqualr r rB rj rs rt Zsuper_eq) r r rX rh � new_vhostZ new_nparserZ new_vhostsZ new_defaultsZnew_vhost_parsedr r r �test_duplicate_vhost� s z$NginxParserTest.test_duplicate_vhostc C s� t �| j�}|�� }dd� |D �d }|j|dd�}|jdd� |jD ]}| �|j� qF|j|dd�}|jdd� d}|jD ]}|j r|| � |j� d}q|| � |� d S ) Nc S s g | ]}d |j kr|�qS )�ipv6sslrS r" r r r r& � s zHNginxParserTest.test_duplicate_vhost_remove_ipv6only.<locals>.<listcomp>r Tr� rZ r� F) r r r rU r� rA r� rr Zipv6onlyZipv6rt )r r rX r� r� ZaddrZidentical_vhostZcalledr r r �$test_duplicate_vhost_remove_ipv6only� s z4NginxParserTest.test_duplicate_vhost_remove_ipv6onlyN)�__name__� __module__� __qualname__�__doc__r r r r r; r> rD rQ rY rl rv r| r� r� r� r� r� r� r� r� r� r� r� r r r r r s0 4.!r �__main__)r� rC r� r ZunittestZacme.magic_typingr Zcertbotr Zcertbot.compatr Z certbot_nginxr r r Zcertbot_nginx.testsr Z NginxTestr r� �mainr r r r �<module> s" Y
Save File
Cancel