b3416e44da
generally useful non-testing facilities. Since extras has been around for a couple of testtools releases now, we're making this into a hard dependency of testtools. (Robert Collins) * Testtools now uses setuptools rather than distutils so that we can document the extras dependency. (Robert Collins)
101 lines
2.9 KiB
Python
101 lines
2.9 KiB
Python
# Copyright (c) 2010-2011 Testtools authors. See LICENSE for details.
|
|
|
|
"""Tests for the distutils test command logic."""
|
|
|
|
from distutils.dist import Distribution
|
|
|
|
from extras import try_import
|
|
|
|
from testtools.compat import (
|
|
_b,
|
|
_u,
|
|
BytesIO,
|
|
)
|
|
fixtures = try_import('fixtures')
|
|
|
|
import testtools
|
|
from testtools import TestCase
|
|
from testtools.distutilscmd import TestCommand
|
|
from testtools.matchers import MatchesRegex
|
|
|
|
|
|
if fixtures:
|
|
class SampleTestFixture(fixtures.Fixture):
|
|
"""Creates testtools.runexample temporarily."""
|
|
|
|
def __init__(self):
|
|
self.package = fixtures.PythonPackage(
|
|
'runexample', [('__init__.py', _b("""
|
|
from testtools import TestCase
|
|
|
|
class TestFoo(TestCase):
|
|
def test_bar(self):
|
|
pass
|
|
def test_quux(self):
|
|
pass
|
|
def test_suite():
|
|
from unittest import TestLoader
|
|
return TestLoader().loadTestsFromName(__name__)
|
|
"""))])
|
|
|
|
def setUp(self):
|
|
super(SampleTestFixture, self).setUp()
|
|
self.useFixture(self.package)
|
|
testtools.__path__.append(self.package.base)
|
|
self.addCleanup(testtools.__path__.remove, self.package.base)
|
|
|
|
|
|
class TestCommandTest(TestCase):
|
|
|
|
def setUp(self):
|
|
super(TestCommandTest, self).setUp()
|
|
if fixtures is None:
|
|
self.skipTest("Need fixtures")
|
|
|
|
def test_test_module(self):
|
|
self.useFixture(SampleTestFixture())
|
|
stdout = self.useFixture(fixtures.StringStream('stdout'))
|
|
dist = Distribution()
|
|
dist.script_name = 'setup.py'
|
|
dist.script_args = ['test']
|
|
dist.cmdclass = {'test': TestCommand}
|
|
dist.command_options = {
|
|
'test': {'test_module': ('command line', 'testtools.runexample')}}
|
|
cmd = dist.reinitialize_command('test')
|
|
with fixtures.MonkeyPatch('sys.stdout', stdout.stream):
|
|
dist.run_command('test')
|
|
self.assertThat(
|
|
stdout.getDetails()['stdout'].as_text(),
|
|
MatchesRegex(_u("""Tests running...
|
|
|
|
Ran 2 tests in \\d.\\d\\d\\ds
|
|
OK
|
|
""")))
|
|
|
|
def test_test_suite(self):
|
|
self.useFixture(SampleTestFixture())
|
|
stdout = self.useFixture(fixtures.StringStream('stdout'))
|
|
dist = Distribution()
|
|
dist.script_name = 'setup.py'
|
|
dist.script_args = ['test']
|
|
dist.cmdclass = {'test': TestCommand}
|
|
dist.command_options = {
|
|
'test': {
|
|
'test_suite': (
|
|
'command line', 'testtools.runexample.test_suite')}}
|
|
cmd = dist.reinitialize_command('test')
|
|
with fixtures.MonkeyPatch('sys.stdout', stdout.stream):
|
|
dist.run_command('test')
|
|
self.assertThat(
|
|
stdout.getDetails()['stdout'].as_text(),
|
|
MatchesRegex(_u("""Tests running...
|
|
|
|
Ran 2 tests in \\d.\\d\\d\\ds
|
|
OK
|
|
""")))
|
|
|
|
|
|
def test_suite():
|
|
from unittest import TestLoader
|
|
return TestLoader().loadTestsFromName(__name__)
|