← Back
Editing: obj_test.cpython-38.pyc
U �].& � @ sL d Z ddlZddlZG dd� dej�ZG dd� dej�ZedkrHe�� dS )z-Test the helper objects in certbot_nginx.obj.� Nc @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )�AddrTestzTest the Addr class.c C sp ddl m} |�d�| _|�d�| _|�d�| _|�d�| _|�d�| _|�d�| _|�d �| _ |�d �| _ d S ) Nr ��Addr� 192.168.1.1�192.168.1.1:* ssl�192.168.1.1:80�*:80 default_server ssl�myhostz80 default_server spdyzunix:/var/run/nginx.sockz*:80 default ssl) �certbot_nginx.objr � fromstring�addr1�addr2�addr3�addr4�addr5�addr6�addr7�addr8)�selfr � r �>/usr/lib/python3/dist-packages/certbot_nginx/tests/obj_test.py�setUp s zAddrTest.setUpc C s� | � | j�� d� | � | j�� d� | �| jj� | �| jj� | � | j�� d� | � | j�� d� | �| jj� | �| jj� | � | j �� d� | � | j �� d� | �| j j� | �| j j� | � | j �� d� | � | j �� d� | �| j j� | �| j j� | � | j �� d� | � | j �� d� | �| j j� | �| j j� | � | j �� d� | � | j �� d� | �| j j� | �| j j� | �| j j� | � d | j� d S )Nr � �*�80r )� assertEqualr Zget_addrZget_port� assertFalseZssl�defaultr � assertTruer r r r r r �r r r r �test_fromstring s4 zAddrTest.test_fromstringc C s� | � t| j�d� | � t| j�d� | � t| j�d� | � t| j�d� | � t| j�d� | � t| j�d� | � t| j�d� d S )Nr r r r r �80 default_server) r �strr r r r r r r r r r r �test_str6 s zAddrTest.test_strc C s� | � | j�� d� | � | j�� d� | � | j�� d� | � | j�� d� | � | jjdd�d� | � | j�� d� | � | j�� d � | � | jjdd�d � d S ) Nr r r r F)Zinclude_defaultz*:80 sslr r! r )r r Z to_stringr r r r r r r r r �test_to_string? s zAddrTest.test_to_stringc C sH ddl m} |�d�}| �| j|� | �| j| j� | �| jdk� d S )Nr r z192.168.1.1 spdyi )r r r r r �assertNotEqualr r )r r Z new_addr1r r r �test_eqI s zAddrTest.test_eqc C s� ddl m} d}t�|d�D ] \}}| �|�|�|�|�� q | �|�|d �|�d�� |D ] }| �|�d�|�|�� q^d S )Nr r )z 0.0.0.0:80 default_server sslz80 default_server sslr z80 default ssl� z 0.0.0.0:443 default_server sslz!192.168.1.2:80 default_server ssl)r r � itertools� combinationsr r r% )r r Z any_addresses�first�secondZ any_addressr r r � test_equivalent_any_addressesP s ��z&AddrTest.test_equivalent_any_addressesc C sL ddl m} t| j| jg�}|�d�}|�d�}t||g�}| �||� d S )Nr r r r )r r �setr r r r )r r Zset_aZaddr1bZaddr2bZset_br r r �test_set_inclusionc s zAddrTest.test_set_inclusionN) �__name__� __module__� __qualname__�__doc__r r r# r$ r& r, r. r r r r r s # r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �VirtualHostTestzTest the VirtualHost class.c C sj ddl m} ddl m} ddgdddd gd d d gggd dgg}|dt|�d�g�ddtdg�|g �| _ddgdddd gd d d gggg}|dt|�d�g�ddtdg�|g �| _ddgddddgg}|dt|�d�g�ddtdg�|g �| _ddgddgg}|dt|�d�g�ddtdg�|g �| _ddgddgdddd d gg}|dt|�d�g�ddtdg�|g �| _ d S ) Nr �� VirtualHostr �listenz69.50.225.155:9000�if�($scheme�!=z "https") �return�301�https://$host$request_uri�#� managed by Certbot�filep� localhostFZrewritez^(.*)$z$scheme://www.domain.com$1Z permanent� server_namez return.com�filpZ add_header�alwaysr- �Strict-Transport-Securityz"max-age=31536000") r r5 r r- r �vhost1Zvhost2Zvhost3Zvhost4�vhost_has_hsts)r r5 r Zraw1Zraw2Zraw3Zraw4Z raw_has_hstsr r r r o s� �� � �� � � �� � � �zVirtualHostTest.setUpc C st ddl m} ddl m} |dt|�d�g�ddtdg�g g �}| �|| j� | �t|�t| j�� | �|dk� d S ) Nr r r4 r? zlocalhost blahFr@ i� ) r r r5 r- r r rE r" r )r r r5 Zvhost1br r r r& � s �zVirtualHostTest.test_eqc C s* d� dddddg�}| �|t| j�� d S )N� z file: filepzaddrs: localhostznames: ['localhost']z ssl: Falsezenabled: False)�joinr r" rE )r Z stringifiedr r r r# � s �zVirtualHostTest.test_strc C sL | � | j�d�� | �| j�d�� | �| j�d�� | �| j�d�� d S )NrD z Bogus-Header)r rF Z has_headerr rE r r r r �test_has_header� s z VirtualHostTest.test_has_headerc C s` ddl m} ddl m} ddlm} dddddd gd gg}||�}d d gd dgddgddgd dgddgddgddgddgddgddd gddgg g }|dt|�d�g�d d tdg�|g �}d d gd dgddgddgd dgddgddgddgddgddgd d d d gdd!d gggddgg g }|dt|�d�g�d d tdg�|g �} | �|�|�� | � | �|�� d S )"Nr r4 r )�_test_block_from_blockz r: � r; r< rG r6 r �rootz /var/www/html�indexz,index.html index.htm index.nginx-debian.htmlrA ztwo.functorkitten.xyzz443 sslr= r> Zssl_certificatez9/etc/letsencrypt/live/two.functorkitten.xyz/fullchain.pemZssl_certificate_keyz7/etc/letsencrypt/live/two.functorkitten.xyz/privkey.pemrB r@ Fr7 r8 r9 z"https")Z302) r r5 r Zcertbot_nginx.configuratorrJ r- r r Z contains_listr ) r r5 r rJ Z test_blockZ test_needleZ test_haystackZvhost_haystackZtest_bad_haystackZvhost_bad_haystackr r r �test_contains_list� sn � � � � � �z"VirtualHostTest.test_contains_listN) r/ r0 r1 r2 r r& r# rI rN r r r r r3 m s 2 r3 �__main__)r2 Zunittestr( ZTestCaser r3 r/ �mainr r r r �<module> s gv
Save File
Cancel