mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
from nose2 import events, loader, session, util
|
|
from nose2.plugins.loader import generators, testcases
|
|
from nose2.tests._common import TestCase
|
|
|
|
|
|
class TestGeneratorUnpack(TestCase):
|
|
tags = ['unit']
|
|
|
|
def setUp(self):
|
|
self.session = session.Session()
|
|
self.loader = loader.PluggableTestLoader(self.session)
|
|
self.expect = [(0, ('call', (0, 1))),
|
|
(1, ('call', (1, 2))),
|
|
(2, ('call', (2, 3))), ]
|
|
self.plugin = generators.Generators(session=self.session)
|
|
# need testcase loader to make the initial response to load from module
|
|
self.tcl = testcases.TestCaseLoader(session=self.session)
|
|
|
|
def test_unpack_handles_nose_style_generators(self):
|
|
def gen():
|
|
for i in range(0, 3):
|
|
yield 'call', i, i + 1
|
|
out = list(self.plugin.unpack(gen()))
|
|
self.assertEqual(out, self.expect)
|
|
|
|
def test_unpack_handles_unittest2_style_generators(self):
|
|
def gen():
|
|
for i in range(0, 3):
|
|
yield 'call', (i, i + 1)
|
|
out = list(self.plugin.unpack(gen()))
|
|
self.assertEqual(out, self.expect)
|
|
|
|
def test_ignores_ordinary_functions(self):
|
|
class Mod(object):
|
|
pass
|
|
|
|
def test():
|
|
pass
|
|
m = Mod()
|
|
m.test = test
|
|
event = events.LoadFromModuleEvent(self.loader, m)
|
|
self.session.hooks.loadTestsFromModule(event)
|
|
self.assertEqual(len(event.extraTests), 0)
|
|
|
|
def test_can_load_tests_from_generator_functions(self):
|
|
class Mod(object):
|
|
__name__ = 'themod'
|
|
|
|
def check(x):
|
|
assert x == 1
|
|
|
|
def test():
|
|
yield check, 1
|
|
yield check, 2
|
|
m = Mod()
|
|
m.test = test
|
|
test.__module__ = m.__name__
|
|
event = events.LoadFromModuleEvent(self.loader, m)
|
|
self.session.hooks.loadTestsFromModule(event)
|
|
self.assertEqual(len(event.extraTests), 2)
|
|
# check that test names are sensible
|
|
self.assertEqual(util.test_name(event.extraTests[0]),
|
|
'themod.test:1')
|
|
self.assertEqual(util.test_name(event.extraTests[1]),
|
|
'themod.test:2')
|
|
|
|
def test_can_load_tests_from_generator_methods(self):
|
|
class Mod(object):
|
|
pass
|
|
|
|
def check(x):
|
|
return x == 1
|
|
|
|
class Test(TestCase):
|
|
|
|
def test(self):
|
|
yield check, 1
|
|
yield check, 2
|
|
m = Mod()
|
|
m.Test = Test
|
|
event = events.LoadFromModuleEvent(self.loader, m)
|
|
self.session.hooks.loadTestsFromModule(event)
|
|
self.assertEqual(len(event.extraTests), 1)
|
|
self.assertEqual(len(event.extraTests[0]._tests), 2)
|