1 #!/usr/bin/env python 2 # Copyright 2013 The Chromium Authors. All rights reserved. 3 # Use of this source code is governed by a BSD-style license that can be 4 # found in the LICENSE file. 5 6 import unittest 7 8 from mock_function import MockFunction 9 10 11 class MockFunctionUnittest(unittest.TestCase): 12 def testMockFunction(self): 13 @MockFunction 14 def calc(a, b, mult=1): 15 return (a + b) * mult 16 17 self.assertTrue(*calc.CheckAndReset(0)) 18 self.assertEqual( 19 (False, 'calc: expected 1 call(s), got 0'), calc.CheckAndReset(1)) 20 21 self.assertEqual(20, calc(2, 3, mult=4)) 22 self.assertTrue(*calc.CheckAndReset(1)) 23 self.assertTrue(*calc.CheckAndReset(0)) 24 25 self.assertEqual(20, calc(2, 3, mult=4)) 26 self.assertEqual( 27 (False, 'calc: expected 0 call(s), got 1'), calc.CheckAndReset(0)) 28 29 self.assertEqual(3, calc(1, 2)) 30 self.assertEqual(0, calc(3, 4, mult=0)) 31 self.assertTrue(*calc.CheckAndReset(2)) 32 self.assertTrue(*calc.CheckAndReset(0)) 33 34 self.assertEqual(3, calc(1, 2)) 35 self.assertEqual(0, calc(3, 4, mult=0)) 36 self.assertEqual( 37 (False, 'calc: expected 3 call(s), got 2'), calc.CheckAndReset(3)) 38 39 40 if __name__ == '__main__': 41 unittest.main() 42
