def test392_chained_function_throws(self):
stub = SinonStub(C_func)
stub.withArgs(42).onFirstCall().throws(
Exception('A')).onSecondCall().throws(Exception('B'))
with self.assertRaisesRegexp(Exception, 'A'):
stub.g.C_func(42)
with self.assertRaisesRegexp(Exception, 'B'):
stub.g.C_func(42)
def test391_chained_module_throws(self):
stub = SinonStub(os, 'system')
stub.withArgs(42).onFirstCall().throws(
Exception('A')).onSecondCall().throws(Exception('B'))
with self.assertRaisesRegexp(Exception, 'A'):
os.system(42)
with self.assertRaisesRegexp(Exception, 'B'):
os.system(42)
def test403_correct_precedence_withArgs_kwargs(self):
stub = SinonStub()
stub.withArgs(a='A').onSecondCall().returns('second call!')
stub.withArgs(a='A').returns('Kwarg of A')
self.assertEqual(stub(), None)
self.assertEqual(stub(a='A'), 'Kwarg of A')
self.assertEqual(stub(a='A'), 'second call!')
self.assertEqual(stub(a='A'), 'Kwarg of A')
def test404_correct_precedence_withArgs_args_kwargs(self):
stub = SinonStub()
stub.withArgs('A', b='B').onSecondCall().returns('second call!')
stub.withArgs('A', b='B').returns('Arg of A, Kwarg of B')
self.assertEqual(stub(), None)
self.assertEqual(stub('A', b='B'), 'Arg of A, Kwarg of B')
self.assertEqual(stub('A', b='B'), 'second call!')
self.assertEqual(stub('A', b='B'), 'Arg of A, Kwarg of B')
def test442_withArgs_onCall_returns_does_not_reset_call_count(self):
stub = SinonStub()
self.assertEqual(stub(), None)
self.assertEqual(stub('A'), None)
stub.withArgs('A').onSecondCall().returns('second call!')
self.assertEqual(stub('A'), 'second call!')
self.assertEqual(stub('A'), None)
self.assertEqual(stub(), None)
def test390_chained_pure_throws(self):
stub = SinonStub()
stub.withArgs(42).onFirstCall().throws(
Exception('A')).onSecondCall().throws(Exception('B'))
with self.assertRaisesRegexp(Exception, 'A'):
stub(42)
with self.assertRaisesRegexp(Exception, 'B'):
stub(42)
def test400_correct_precedence(self):
stub = SinonStub()
stub.withArgs('A').returns('Arg of A')
stub.onFirstCall().returns('First call!')
stub.onSecondCall().returns('Second call!')
stub.returns('No args')
self.assertEqual(stub(), 'First call!')
self.assertEqual(stub('A'), 'Arg of A')
self.assertEqual(stub(), 'No args')
def test443_withArgs_onCall_throws_does_not_reset_call_count(self):
stub = SinonStub()
self.assertEqual(stub(), None)
self.assertEqual(stub('A'), None)
stub.withArgs('A').onSecondCall().throws(Exception('second call!'))
with self.assertRaisesRegexp(Exception, 'second call'):
stub('A')
self.assertEqual(stub('A'), None)
self.assertEqual(stub(), None)
def test393_chained_method_throws(self):
o = A_object()
stub = SinonStub(o, 'A_func')
stub.withArgs(42).onFirstCall().throws(
Exception('A')).onSecondCall().throws(Exception('B'))
with self.assertRaisesRegexp(Exception, 'A'):
o.A_func(42)
with self.assertRaisesRegexp(Exception, 'B'):
o.A_func(42)
def test408_chained_withArgs_onCall_returns(self):
stub = SinonStub()
stub.withArgs(42).onFirstCall().returns(1).onSecondCall().returns(2)
stub.returns(0)
self.assertEqual(stub(1), 0)
self.assertEqual(stub(42), 1)
self.assertEqual(stub(1), 0)
self.assertEqual(stub(42), 2)
self.assertEqual(stub(1), 0)
self.assertEqual(stub(42), 0)
def test406_correct_precedence_withArgs_kwargs_throws(self):
stub = SinonStub()
stub.withArgs(a='A').onSecondCall().throws(Exception('second call!'))
stub.withArgs(a='A').throws(Exception('Kwarg of A'))
self.assertEqual(stub(), None)
with self.assertRaisesRegexp(Exception, 'Kwarg of A'):
stub(a='A')
with self.assertRaisesRegexp(Exception, 'second call!'):
stub(a='A')
with self.assertRaisesRegexp(Exception, 'Kwarg of A'):
stub(a='A')
def test401_correct_precedence_throws(self):
stub = SinonStub()
stub.withArgs('A').throws(Exception('A'))
stub.onFirstCall().throws(Exception('First call!'))
stub.onSecondCall().throws(Exception('Second call!'))
stub.throws(Exception('No args'))
with self.assertRaisesRegexp(Exception, 'First call!'):
stub()
with self.assertRaisesRegexp(Exception, 'A'):
stub('A')
with self.assertRaisesRegexp(Exception, 'No args'):
stub()
def test224_onCall_withArgs(self):
fto = ForTestOnly()
self.assertEqual(fto.func1(), "func1")
stub = SinonStub(ForTestOnly, "func1")
stub.withArgs(1).onThirdCall().returns("oncall")
stub.withArgs(2).onSecondCall().returns("oncall")
self.assertEqual(fto.func1(1), None)
self.assertEqual(fto.func1(1), None)
self.assertEqual(fto.func1(1), "oncall")
self.assertEqual(fto.func1(1), None)
self.assertEqual(fto.func1(2), None)
self.assertEqual(fto.func1(2), "oncall")
self.assertEqual(fto.func1(2), None)
def test225_onCall_plus_withArgs(self):
fto = ForTestOnly()
self.assertEqual(fto.func1(), "func1")
stub = SinonStub(ForTestOnly, "func1")
stub.withArgs(1).returns("1")
stub.withArgs(1).onSecondCall().returns("oncall")
stub.onThirdCall().returns("##")
stub.onCall(4).returns("###") # on fifth call will return "###"
self.assertEqual(fto.func1(1), "1")
self.assertEqual(fto.func1(1), "oncall")
self.assertEqual(fto.func1(1),
"1") # the priority of onCall is lower than withArgs
self.assertEqual(fto.func1(), None)
self.assertEqual(fto.func1(), "###")
def test409_chained_withArgs_onCall_throws(self):
stub = SinonStub()
stub.withArgs(42).onFirstCall().throws(
Exception('A')).onSecondCall().throws(Exception('B'))
stub.throws(Exception('C'))
with self.assertRaisesRegexp(Exception, 'C'):
stub(1)
with self.assertRaisesRegexp(Exception, 'A'):
stub(42)
with self.assertRaisesRegexp(Exception, 'C'):
stub(1)
with self.assertRaisesRegexp(Exception, 'B'):
stub(42)
with self.assertRaisesRegexp(Exception, 'C'):
stub(1)
with self.assertRaisesRegexp(Exception, 'C'):
stub(42)
def test222_withArgs(self):
fto = ForTestOnly()
stub = SinonStub(ForTestOnly, "func1")
stub.withArgs(1).returns("#")
self.assertEqual(fto.func1(1), "#")
stub.withArgs(b=1).returns("##")
self.assertEqual(fto.func1(b=1), "##")
stub.withArgs(1, b=1).returns("###")
self.assertEqual(fto.func1(1, b=1), "###")
self.assertEqual(fto.func1(2), None)
def test430_withArgs_onCall_combine_correctly_returns(self):
stub = SinonStub()
stub.withArgs('A').onFirstCall().returns(1)
stub.withArgs('A').onSecondCall().returns(2)
stub.withArgs('B').onFirstCall().returns(10)
stub.withArgs('B').onSecondCall().returns(20)
self.assertEqual(stub(), None)
self.assertEqual(stub('A'), 1)
self.assertEqual(stub('A'), 2)
self.assertEqual(stub('A'), None)
self.assertEqual(stub('B'), 10)
self.assertEqual(stub('B'), 20)
self.assertEqual(stub('B'), None)
def test432_withArgs_onCall_combine_correctly_class(self):
fto = ForTestOnly()
stub = SinonStub(ForTestOnly, "func1")
stub.withArgs('A').onFirstCall().returns(1)
stub.withArgs('A').onSecondCall().returns(2)
stub.withArgs('B').onFirstCall().returns(10)
stub.withArgs('B').onSecondCall().returns(20)
self.assertEqual(stub(fto), None)
self.assertEqual(stub(fto, 'A'), 1)
self.assertEqual(stub(fto, 'A'), 2)
self.assertEqual(stub(fto, 'A'), None)
self.assertEqual(stub(fto, 'B'), 10)
self.assertEqual(stub(fto, 'B'), 20)
self.assertEqual(stub(fto, 'B'), None)
def test431_withArgs_onCall_combine_correctly_throws(self):
stub = SinonStub()
stub.withArgs('A').onFirstCall().throws(Exception('1'))
stub.withArgs('A').onSecondCall().throws(Exception('2'))
stub.withArgs('B').onFirstCall().throws(Exception('10'))
stub.withArgs('B').onSecondCall().throws(Exception('20'))
self.assertEqual(stub(), None)
with self.assertRaisesRegexp(Exception, '1'):
stub('A')
with self.assertRaisesRegexp(Exception, '2'):
stub('A')
self.assertEqual(stub('A'), None)
with self.assertRaisesRegexp(Exception, '10'):
stub('B')
with self.assertRaisesRegexp(Exception, '20'):
stub('B')
self.assertEqual(stub('B'), None)
def test383_chained_method_returns(self):
o = A_object()
stub = SinonStub(o, 'A_func')
stub.withArgs(42).onFirstCall().returns(1).onSecondCall().returns(2)
self.assertEqual(o.A_func(42), 1)
self.assertEqual(o.A_func(42), 2)
def test381_chained_module_returns(self):
stub = SinonStub(os, 'system')
stub.withArgs(42).onFirstCall().returns(1).onSecondCall().returns(2)
self.assertEqual(os.system(42), 1)
self.assertEqual(os.system(42), 2)
def test415_conditions_can_be_overwritten_withArgs(self):
stub = SinonStub()
stub.withArgs('A').withArgs('B').returns(3)
self.assertEqual(stub('A'), None)
self.assertEqual(stub('B'), 3)
def test410_conditions_do_not_persist(self):
stub = SinonStub()
stub.withArgs('A')
stub.onThirdCall()
stub.returns(5)
self.assertEqual(stub(), 5)
def test380_chained_pure_returns(self):
stub = SinonStub()
stub.withArgs(42).onFirstCall().returns(1).onSecondCall().returns(2)
self.assertEqual(stub(42), 1)
self.assertEqual(stub(42), 2)
def test350_withArgs_empty_stub(self):
stub = SinonStub()
stub.withArgs(42).returns(1)
self.assertEqual(stub(42), 1)
def test382_chained_function_returns(self):
stub = SinonStub(C_func)
stub.withArgs(42).onFirstCall().returns(1).onSecondCall().returns(2)
self.assertEqual(stub.g.C_func(42), 1)
self.assertEqual(stub.g.C_func(42), 2)
def test301_withArgs_os(self):
stub = SinonStub(os, "system")
stub.withArgs("pwd").returns("customized result")
self.assertEqual(os.system("pwd"), "customized result")
self.assertFalse(os.system())
