105 lines
3.2 KiB
Python
105 lines
3.2 KiB
Python
from unittest import TestCase
|
|
from nose.tools import assert_equal
|
|
from nose.plugins.skip import SkipTest
|
|
|
|
from .compat import PY3
|
|
from .parameterized import parameterized, param
|
|
|
|
def assert_contains(haystack, needle):
|
|
if needle not in haystack:
|
|
raise AssertionError("%r not in %r" %(needle, haystack))
|
|
|
|
missing_tests = set([
|
|
"test_naked_function(42, bar=None)",
|
|
"test_naked_function('foo0', bar=None)",
|
|
"test_naked_function('foo1', bar=None)",
|
|
"test_naked_function('foo2', bar=42)",
|
|
"test_instance_method(42, bar=None)",
|
|
"test_instance_method('foo0', bar=None)",
|
|
"test_instance_method('foo1', bar=None)",
|
|
"test_instance_method('foo2', bar=42)",
|
|
"test_on_TestCase(42, bar=None)",
|
|
"test_on_TestCase('foo0', bar=None)",
|
|
"test_on_TestCase('foo1', bar=None)",
|
|
"test_on_TestCase('foo2', bar=42)",
|
|
"test_on_old_style_class('foo')",
|
|
"test_on_old_style_class('bar')",
|
|
])
|
|
|
|
test_params = [
|
|
(42, ),
|
|
"foo0",
|
|
param("foo1"),
|
|
param("foo2", bar=42),
|
|
]
|
|
|
|
@parameterized(test_params)
|
|
def test_naked_function(foo, bar=None):
|
|
missing_tests.remove("test_naked_function(%r, bar=%r)" %(foo, bar))
|
|
|
|
|
|
class TestParameterized(object):
|
|
@parameterized(test_params)
|
|
def test_instance_method(self, foo, bar=None):
|
|
missing_tests.remove("test_instance_method(%r, bar=%r)" %(foo, bar))
|
|
|
|
|
|
class TestParamerizedOnTestCase(TestCase):
|
|
@parameterized.expand(test_params)
|
|
def test_on_TestCase(self, foo, bar=None):
|
|
missing_tests.remove("test_on_TestCase(%r, bar=%r)" %(foo, bar))
|
|
|
|
|
|
def test_warns_when_using_parameterized_with_TestCase():
|
|
try:
|
|
class TestTestCaseWarnsOnBadUseOfParameterized(TestCase):
|
|
@parameterized([42])
|
|
def test_in_subclass_of_TestCase(self, foo):
|
|
pass
|
|
except Exception as e:
|
|
assert_contains(str(e), "parameterized.expand")
|
|
else:
|
|
raise AssertionError("Expected exception not raised")
|
|
|
|
missing_tests.add("test_wrapped_iterable_input()")
|
|
@parameterized(lambda: iter(["foo"]))
|
|
def test_wrapped_iterable_input(foo):
|
|
missing_tests.remove("test_wrapped_iterable_input()")
|
|
|
|
def test_helpful_error_on_non_iterable_input():
|
|
try:
|
|
for _ in parameterized(lambda: 42)(lambda: None)():
|
|
pass
|
|
except Exception as e:
|
|
assert_contains(str(e), "expected iterable input")
|
|
else:
|
|
raise AssertionError("Expected exception not raised")
|
|
|
|
|
|
def teardown_module():
|
|
missing = sorted(list(missing_tests))
|
|
assert_equal(missing, [])
|
|
|
|
|
|
def test_old_style_classes():
|
|
if PY3:
|
|
raise SkipTest("Py3 doesn't have old-style classes")
|
|
class OldStyleClass:
|
|
@parameterized(["foo"])
|
|
def parameterized_method(self, param):
|
|
pass
|
|
try:
|
|
list(OldStyleClass().parameterized_method())
|
|
except TypeError as e:
|
|
assert_contains(str(e), "new-style")
|
|
assert_contains(str(e), "parameterized.expand")
|
|
assert_contains(str(e), "OldStyleClass")
|
|
else:
|
|
raise AssertionError("expected TypeError not raised by old-style class")
|
|
|
|
|
|
class TestOldStyleClass:
|
|
@parameterized.expand(["foo", "bar"])
|
|
def test_old_style_classes(self, param):
|
|
missing_tests.remove("test_on_old_style_class(%r)" %(param, ))
|