Gecko [ 167.86.127.34 ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
__init__.py	985 B	-rw-r--r--
base.py	8.57 KB	-rw-r--r--
test_commands.py	3.6 KB	-rw-r--r--
test_core.py	5.29 KB	-rw-r--r--
test_files.py	2.52 KB	-rw-r--r--
test_hooks.py	3.94 KB	-rw-r--r--
test_integration.py	10.64 KB	-rw-r--r--
test_packaging.py	30.35 KB	-rw-r--r--
test_pbr_json.py	1.13 KB	-rw-r--r--
test_setup.py	22.44 KB	-rw-r--r--
test_util.py	2.73 KB	-rw-r--r--
test_version.py	13.28 KB	-rw-r--r--
test_wsgi.py	5.67 KB	-rw-r--r--
util.py	2.61 KB	-rw-r--r--

Code Editor : test_wsgi.py

# Copyright (c) 2015 Hewlett-Packard Development Company, L.P. (HP)
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

import os
import re
import subprocess
import sys
try:
    # python 2
    from urllib2 import urlopen
except ImportError:
    # python 3
    from urllib.request import urlopen

from pbr.tests import base

class TestWsgiScripts(base.BaseTestCase):

cmd_names = ('pbr_test_wsgi', 'pbr_test_wsgi_with_class')

def _get_path(self):
        if os.path.isdir("%s/lib64" % self.temp_dir):
            path = "%s/lib64" % self.temp_dir
        elif os.path.isdir("%s/lib" % self.temp_dir):
            path = "%s/lib" % self.temp_dir
        elif os.path.isdir("%s/site-packages" % self.temp_dir):
            return ".:%s/site-packages" % self.temp_dir
        else:
            raise Exception("Could not determine path for test")
        return ".:%s/python%s.%s/site-packages" % (
            path,
            sys.version_info[0],
            sys.version_info[1])

def test_wsgi_script_install(self):
        """Test that we install a non-pkg-resources wsgi script."""
        if os.name == 'nt':
            self.skipTest('Windows support is passthrough')

stdout, _, return_code = self.run_setup(
            'install', '--prefix=%s' % self.temp_dir)

self._check_wsgi_install_content(stdout)

def test_wsgi_script_run(self):
        """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.

"""
        self.skipTest('Not working on Debian')
        if os.name == 'nt':
            self.skipTest('Windows support is passthrough')

stdout, _, return_code = self.run_setup(
            'install', '--prefix=%s' % self.temp_dir)

self._check_wsgi_install_content(stdout)

# Live test run the scripts and see that they respond to wsgi
        # requests.
        for cmd_name in self.cmd_names:
            self._test_wsgi(cmd_name, b'Hello World')

def _test_wsgi(self, cmd_name, output, extra_args=None):
        cmd = os.path.join(self.temp_dir, 'bin', cmd_name)
        print("Running %s -p 0" % cmd)
        popen_cmd = [cmd, '-p', '0']
        if extra_args:
            popen_cmd.extend(extra_args)

env = {'PYTHONPATH': self._get_path()}

p = subprocess.Popen(popen_cmd, stdout=subprocess.PIPE,
                             stderr=subprocess.PIPE, cwd=self.temp_dir,
                             env=env)
        self.addCleanup(p.kill)

stdoutdata = p.stdout.readline()  # ****...

stdoutdata = p.stdout.readline()  # STARTING test server...
        self.assertIn(
            b"STARTING test server pbr_testpackage.wsgi",
            stdoutdata)

stdoutdata = p.stdout.readline()  # Available at ...
        print(stdoutdata)
        m = re.search(b'(http://[^:]+:\d+)/', stdoutdata)
        self.assertIsNotNone(m, "Regex failed to match on %s" % stdoutdata)

stdoutdata = p.stdout.readline()  # DANGER! ...
        self.assertIn(
            b"DANGER! For testing only, do not use in production",
            stdoutdata)

stdoutdata = p.stdout.readline()  # ***...

f = urlopen(m.group(1).decode('utf-8'))
        self.assertEqual(output, f.read())

# Request again so that the application can force stderr.flush(),
        # otherwise the log is buffered and the next readline() will hang.
        urlopen(m.group(1).decode('utf-8'))

stdoutdata = p.stderr.readline()
        # we should have logged an HTTP request, return code 200, that
        # returned the right amount of bytes
        status = '"GET / HTTP/1.1" 200 %d' % len(output)
        self.assertIn(status.encode('utf-8'), stdoutdata)

def _check_wsgi_install_content(self, install_stdout):
        for cmd_name in self.cmd_names:
            install_txt = 'Installing %s script to %s' % (cmd_name,
                                                          self.temp_dir)
            self.assertIn(install_txt, install_stdout)

cmd_filename = os.path.join(self.temp_dir, 'bin', cmd_name)

script_txt = open(cmd_filename, 'r').read()
            self.assertNotIn('pkg_resources', script_txt)

main_block = """if __name__ == "__main__":
    import argparse
    import socket
    import sys
    import wsgiref.simple_server as wss"""

if cmd_name == 'pbr_test_wsgi':
                app_name = "main"
            else:
                app_name = "WSGI.app"

starting_block = ("STARTING test server pbr_testpackage.wsgi."
                              "%s" % app_name)

else_block = """else:
    application = None"""

self.assertIn(main_block, script_txt)
            self.assertIn(starting_block, script_txt)
            self.assertIn(else_block, script_txt)

def test_with_argument(self):
        self.skipTest('Not working on Debian')
        if os.name == 'nt':
            self.skipTest('Windows support is passthrough')

stdout, _, return_code = self.run_setup(
            'install', '--prefix=%s' % self.temp_dir)

self._test_wsgi('pbr_test_wsgi', b'Foo Bar', ["--", "-c", "Foo Bar"])

${this.title}

Code Editor : test_wsgi.py