summaryrefslogtreecommitdiff
path: root/functional_tests/test_generator_fixtures.py
blob: 3240141162f1d3d839e36e749b607bcb9b145efd (plain)
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
from nose.tools import eq_
called = []

def outer_setup():
    called.append('outer_setup')

def outer_teardown():
    called.append('outer_teardown')

def inner_setup():
    called.append('inner_setup')

def inner_teardown():
    called.append('inner_teardown')

def test_gen():
    called[:] = []
    for i in range(0, 5):
        yield check, i
        
def check(i):
    expect = ['outer_setup']
    for x in range(0, i):
        expect.append('inner_setup')
        expect.append('inner_teardown')
    expect.append('inner_setup')
    eq_(called, expect)

    
test_gen.setup = outer_setup
test_gen.teardown = outer_teardown
check.setup = inner_setup
check.teardown = inner_teardown


class TestClass(object):
    def setup(self):
        print "setup called in", self
        self.called = ['setup']

    def teardown(self):
        print "teardown called in", self
        eq_(self.called, ['setup'])
        self.called.append('teardown')

    def test(self):
        print "test called in", self
        for i in range(0, 5):
            yield self.check, i

    def check(self, i):
        print "check called in", self
        expect = ['setup']
        #for x in range(0, i):
        #    expect.append('setup')
        #    expect.append('teardown')
        #expect.append('setup')
        eq_(self.called, expect)