�
�v,c� � � � d dl Z d dlZd dlZd dlZ d dlmZ n# e$ r d dlmZ Y nw xY wd dlm Z G d� de j
� � Z dS )� N)�urlopen)�basec �8 � e Zd ZdZd� Zd� Zd� Zd d�Zd� Zd� Z dS )
�TestWsgiScripts)�
pbr_test_wsgi�pbr_test_wsgi_with_classc � � t j � d| j z � � r d| j z }nrt j � d| j z � � r d| j z }n@t j � d| j z � � r
d| j z S t d� � �d|�dt
j d �d t
j d
�d �S ) Nz%s/lib64z%s/libz%s/site-packagesz.:%s/site-packagesz!Could not determine path for testz.:z/pythonr �.� z/site-packages)�os�path�isdir�temp_dir� Exception�sys� version_info)�selfr
s �5/usr/lib/python3/dist-packages/pbr/tests/test_wsgi.py� _get_pathzTestWsgiScripts._get_path! s� � �
�7�=�=��d�m�3� 4� 4� A� ��
�-�D�D�
�W�]�]�8�d�m�3�
4�
4� A��d�m�+�D�D�
�W�]�]�-��
�=�
>�
>� A�'�$�-�7� 7�� ?�@�@� @�� �D�D� � �Q� � � � � �Q� � � �!� !� c � � t j dk r| � d� � | � dd| j z � � \ }}}| � |� � dS )z5Test that we install a non-pkg-resources wsgi script.�nt� Windows support is passthrough�install� --prefix=%sN)r �name�skipTest� run_setupr �_check_wsgi_install_content�r �stdout�_� return_codes r �test_wsgi_script_installz(TestWsgiScripts.test_wsgi_script_install/ sb � �
�7�d�?�?� �M�M�:� ;� ;� ;�!%� � � �}�t�}�4�"6� "6� ���;�
�(�(��0�0�0�0�0r c �( � | � d� � t j dk r| � d� � | � dd| j z � � \ }}}| � |� � | j D ]}| � |d� � �dS )z�Test that we install a runnable wsgi script.
This test actually attempts to start and interact with the
wsgi script in question to demonstrate that it's a working
wsgi script using simple server.
�Not working on Debianr r r r s Hello WorldN)r r r r r r � cmd_names�
_test_wsgi)r r! r" r# �cmd_names r �test_wsgi_script_runz$TestWsgiScripts.test_wsgi_script_run9 s� � �
�
�
�-�.�.�.�
�7�d�?�?� �M�M�:� ;� ;� ;�!%� � � �}�t�}�4�"6� "6� ���;�
�(�(��0�0�0� �� 6� 6�H� �O�O�H�n� 5� 5� 5� 5� 6� 6r Nc � � t j � | j d|� � }t d|z � � |ddddg}|r|� |� � d| � � � i}t j |t j t j | j |�� � }| �
|j � � |j �
� � }|j �
� � }| � d |� � |j �
� � }t |� � t j d
|� � } | � | d |z � � |j �
� � }| � d |� � |j �
� � }t% | � d
� � � d� � � � }
| � ||
� � � � � t% | � d
� � � d� � � � |j �
� � }dt1 |� � z } | � | � d� � |� � d S )N�binz Running %s -p 0 -b 127.0.0.1z-p�0z-bz 127.0.0.1�
PYTHONPATH)r! �stderr�cwd�envs) STARTING test server pbr_testpackage.wsgis (http://[^:]+:\d+)/zRegex failed to match on %ss2 DANGER! For testing only, do not use in productionr zutf-8z"GET / HTTP/1.1" 200 %d)r r
�joinr �print�extendr �
subprocess�Popen�PIPE�
addCleanup�killr! �readline�assertIn�re�search�assertIsNotNoner �group�decode� assertEqual�readr/ �len�encode) r r) �output�
extra_args�cmd� popen_cmdr1 �p�
stdoutdata�m�f�statuss r r( zTestWsgiScripts._test_wsgiO s; � ��g�l�l�4�=�%��:�:��
�,�s�2�3�3�3��$��T�;�7� � � )� � � �Z� (� (� (��T�^�^� -� -�.�� � �Y�z��$.�O�� �!$�
&�
&�
&��
����� � � ��X�&�&�(�(�
��X�&�&�(�(�
� �
�
� 8� � � � � �X�&�&�(�(�
�
�j���� �I�-�z� :� :�� � � �Q� =�
� J�K�K�K��X�&�&�(�(�
� �
�
� A� � � � � �X�&�&�(�(�
� �A�G�G�A�J�J�%�%�g�.�.� /� /�� ���������*�*�*� �����
�
�!�!�'�*�*�+�+�+��X�&�&�(�(�
� +�S��[�[�8�� �
�
�f�m�m�G�,�,�j�9�9�9�9�9r c �� � | j D ]�}d|�d| j ��}| � ||� � t j � | j d|� � }t
|d� � � � � }| � d|� � d}|dk rd}nd }d
|z }d } | � ||� � | � ||� � | � | |� � ��d S ) Nz Installing z script to r, �r�
pkg_resourceszwif __name__ == "__main__":
import argparse
import socket
import sys
import wsgiref.simple_server as wssr �mainzWSGI.appz,STARTING test server pbr_testpackage.wsgi.%sz else:
application = None) r' r r; r r
r2 �openrB � assertNotIn)
r �install_stdoutr) � install_txt� cmd_filename�
script_txt�
main_block�app_name�starting_block�
else_blocks
r r z+TestWsgiScripts._check_wsgi_install_content} s
� � �� 2� 2�H�H�:B�(�(�:>�-�-�I�K� �M�M�+�~� 6� 6� 6� �7�<�<��
�u�h�G�G�L� �l�C�0�0�5�5�7�7�J� � � �_�j� 9� 9� 9�+�J� �?�*�*�!���%�� #�%-� .�N��J�
�M�M�*�j� 1� 1� 1� �M�M�.�*� 5� 5� 5� �M�M�*�j� 1� 1� 1� 1�; 2� 2r c �� � | � d� � t j dk r| � d� � | � dd| j z � � \ }}}| � ddg d�� � d S ) Nr& r r r r r s Foo Bar)z--z-czFoo Bar)r r r r r r( r s r �test_with_argumentz"TestWsgiScripts.test_with_argument� s~ � � �
�
�-�.�.�.�
�7�d�?�?� �M�M�:� ;� ;� ;�!%� � � �}�t�}�4�"6� "6� ���;�
�����5L�5L�5L�M�M�M�M�Mr )N)
�__name__�
__module__� __qualname__r' r r$ r* r( r r] � r r r r s � � � � � �=�I� !� !� !� 1� 1� 1�6� 6� 6�,,:� ,:� ,:� ,:�\ 2� 2� 2�@N� N� N� N� Nr r ) r r< r5 r �urllib2r � ImportError�urllib.request� pbr.testsr � BaseTestCaser ra r r �<module>rg s� ��
� � � � � � � � � � � �
�
�
�
�'� � � � � � � ��� '� '� '�&�&�&�&�&�&�&�&�'���� � � � � � �HN� HN� HN� HN� HN�d�'� HN� HN� HN� HN� HNs � � '�' |