| #!/usr/bin/env python |
| # Copyright 2013 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import unittest |
| |
| from mock_function import MockFunction |
| |
| |
| class MockFunctionUnittest(unittest.TestCase): |
| def testMockFunction(self): |
| @MockFunction |
| def calc(a, b, mult=1): |
| return (a + b) * mult |
| |
| self.assertTrue(*calc.CheckAndReset(0)) |
| self.assertEqual( |
| (False, 'calc: expected 1 call(s), got 0'), calc.CheckAndReset(1)) |
| |
| self.assertEqual(20, calc(2, 3, mult=4)) |
| self.assertTrue(*calc.CheckAndReset(1)) |
| self.assertTrue(*calc.CheckAndReset(0)) |
| |
| self.assertEqual(20, calc(2, 3, mult=4)) |
| self.assertEqual( |
| (False, 'calc: expected 0 call(s), got 1'), calc.CheckAndReset(0)) |
| |
| self.assertEqual(3, calc(1, 2)) |
| self.assertEqual(0, calc(3, 4, mult=0)) |
| self.assertTrue(*calc.CheckAndReset(2)) |
| self.assertTrue(*calc.CheckAndReset(0)) |
| |
| self.assertEqual(3, calc(1, 2)) |
| self.assertEqual(0, calc(3, 4, mult=0)) |
| self.assertEqual( |
| (False, 'calc: expected 3 call(s), got 2'), calc.CheckAndReset(3)) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
