def testExplicitGeneratorConvenienceFunctionUsage(self): """Check normal operation with explicit generators using """ c = Controller() x = c.mock() c.generator(x.g(8, 9), [10, 11]) c.replay() self.failUnless([k for k in x.g(8, 9)] == [10, 11])
def testExplicitGeneratorConvenienceFunctionExceptionUsage(self):
"""Check explicit generators using compact specification"""
c = Controller()
x = c.mock()
c.generator(x.g(8, 9), [10], Exception("bogus"))
c.replay()
g = x.g(8, 9)
self.failUnless(g.next() == 10)
self.failUnlessRaises(Exception, g.next)
def testExplicitGeneratorUsage(self): """Check operation of explicit generators using discrete settings""" c = Controller() x = c.mock() x.g(8, 9) c.generator() c.setReturn(10) c.setReturn(11) c.replay() self.failUnless([k for k in x.g(8, 9)] == [10, 11])
def testExplicitGeneratorExecptionUsage(self):
"""Check exception raising with explicit generators using discrete settings"""
c = Controller()
x = c.mock()
x.g(8, 9)
c.generator()
c.setReturn(10)
c.setException(Exception("bogus"))
c.replay()
g = x.g(8, 9)
self.failUnless(g.next() == 10)
self.failUnlessRaises(Exception, g.next)
