1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
|
"""
Unit tests for nose2 decorators.
"""
from nose2.tests._common import TestCase
from nose2.tools.decorators import with_setup, with_teardown
class WithSetupDecoratorTests(TestCase):
def fake_setup(self):
pass
class fake_test(object):
setup = None
def test_setup_injection(self):
sut = self.fake_test()
expected = with_setup(self.fake_setup)(sut).setup
self.assertEquals(expected, self.fake_setup)
class WithTeardownDecoratorTests(TestCase):
def fake_teardown(self):
pass
class fake_test(object):
teardown = None
def test_teardown_injection(self):
sut = self.fake_test()
expected = with_teardown(self.fake_teardown)(sut).tearDownFunc
self.assertEquals(expected, self.fake_teardown)
|