c65dbba2b9
Some WSGI applications require a lot of prior setup before they can be use. Since we want most of that setup to happen in fixtures, wsgi-interception is delayed until after all extant fixtures have been started. This is done with a built in default fixture that is run as the most deeply nested of the context managers. The intecept callable is passed on along to the tests so that suite can be made aware of and use it when using the intercept context manager. If the callable is a function, when it is passed into the class builder, it can become bound to the TestCase. This binding is unbound in the caller.
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
# Copyright 2014, 2015 Red Hat
|
|
#
|
|
# Authors: Chris Dent <chdent@redhat.com>
|
|
#
|
|
# 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.
|
|
|
|
from unittest import case
|
|
from unittest import suite
|
|
|
|
from . import fixture
|
|
|
|
|
|
class GabbiSuite(suite.TestSuite):
|
|
"""A TestSuite with fixtures.
|
|
|
|
The suite wraps the tests with a set of nested context managers that
|
|
operate as fixtures.
|
|
|
|
If a fixture raises unittest.case.SkipTest during setup, all the
|
|
tests in this suite will be skipped.
|
|
"""
|
|
|
|
def run(self, result, debug=False):
|
|
"""Override TestSuite run to start suite-level fixtures.
|
|
|
|
To avoid exception confusion, use a null Fixture when there
|
|
are no fixtures.
|
|
"""
|
|
|
|
# If there are fixtures, nest in their context.
|
|
fixtures = [fixture.GabbiFixture]
|
|
intercept = None
|
|
|
|
try:
|
|
first_test = self._tests[0]
|
|
fixtures = first_test.fixtures
|
|
host = first_test.host
|
|
port = first_test.port
|
|
intercept = first_test.intercept
|
|
|
|
# Unbind a passed in WSGI application. During the
|
|
# metaclass building process intercept becomes bound.
|
|
try:
|
|
intercept = intercept.__func__
|
|
except AttributeError:
|
|
pass
|
|
except AttributeError:
|
|
pass
|
|
|
|
try:
|
|
with fixture.nest([fix() for fix in fixtures]):
|
|
if intercept:
|
|
with fixture.InterceptFixture(host, port, intercept):
|
|
result = super(GabbiSuite, self).run(result, debug)
|
|
else:
|
|
result = super(GabbiSuite, self).run(result, debug)
|
|
except case.SkipTest as exc:
|
|
[result.addSkip(test, str(exc)) for test in self._tests]
|
|
|
|
return result
|