| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | import unittest | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | from unittest.mock import ( | 
					
						
							|  |  |  |     call, _Call, create_autospec, MagicMock, | 
					
						
							|  |  |  |     Mock, ANY, _CallList, patch, PropertyMock | 
					
						
							|  |  |  | ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | from datetime import datetime | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class SomeClass(object): | 
					
						
							|  |  |  |     def one(self, a, b): | 
					
						
							|  |  |  |         pass | 
					
						
							|  |  |  |     def two(self): | 
					
						
							|  |  |  |         pass | 
					
						
							|  |  |  |     def three(self, a=None): | 
					
						
							|  |  |  |         pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class AnyTest(unittest.TestCase): | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_any(self): | 
					
						
							|  |  |  |         self.assertEqual(ANY, object()) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = Mock() | 
					
						
							|  |  |  |         mock(ANY) | 
					
						
							|  |  |  |         mock.assert_called_with(ANY) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = Mock() | 
					
						
							|  |  |  |         mock(foo=ANY) | 
					
						
							|  |  |  |         mock.assert_called_with(foo=ANY) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_repr(self): | 
					
						
							|  |  |  |         self.assertEqual(repr(ANY), '<ANY>') | 
					
						
							|  |  |  |         self.assertEqual(str(ANY), '<ANY>') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_any_and_datetime(self): | 
					
						
							|  |  |  |         mock = Mock() | 
					
						
							|  |  |  |         mock(datetime.now(), foo=datetime.now()) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock.assert_called_with(ANY, foo=ANY) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_any_mock_calls_comparison_order(self): | 
					
						
							|  |  |  |         mock = Mock() | 
					
						
							|  |  |  |         d = datetime.now() | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             def __eq__(self, other): | 
					
						
							|  |  |  |                 return False | 
					
						
							|  |  |  |             def __ne__(self, other): | 
					
						
							|  |  |  |                 return True | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         for d in datetime.now(), Foo(): | 
					
						
							|  |  |  |             mock.reset_mock() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             mock(d, foo=d, bar=d) | 
					
						
							|  |  |  |             mock.method(d, zinga=d, alpha=d) | 
					
						
							|  |  |  |             mock().method(a1=d, z99=d) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             expected = [ | 
					
						
							|  |  |  |                 call(ANY, foo=ANY, bar=ANY), | 
					
						
							|  |  |  |                 call.method(ANY, zinga=ANY, alpha=ANY), | 
					
						
							|  |  |  |                 call(), call().method(a1=ANY, z99=ANY) | 
					
						
							|  |  |  |             ] | 
					
						
							|  |  |  |             self.assertEqual(expected, mock.mock_calls) | 
					
						
							|  |  |  |             self.assertEqual(mock.mock_calls, expected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class CallTest(unittest.TestCase): | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_call_with_call(self): | 
					
						
							|  |  |  |         kall = _Call() | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call()) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(('',))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(((),))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(({},))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(('', ()))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(('', {}))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(('', (), {}))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(('foo',))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(('bar', ()))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(('baz', {}))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(('spam', (), {}))) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         kall = _Call(((1, 2, 3),)) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(((1, 2, 3),))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(('', (1, 2, 3)))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(((1, 2, 3), {}))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(('', (1, 2, 3), {}))) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         kall = _Call(((1, 2, 4),)) | 
					
						
							|  |  |  |         self.assertNotEqual(kall, _Call(('', (1, 2, 3)))) | 
					
						
							|  |  |  |         self.assertNotEqual(kall, _Call(('', (1, 2, 3), {}))) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         kall = _Call(('foo', (1, 2, 4),)) | 
					
						
							|  |  |  |         self.assertNotEqual(kall, _Call(('', (1, 2, 4)))) | 
					
						
							|  |  |  |         self.assertNotEqual(kall, _Call(('', (1, 2, 4), {}))) | 
					
						
							|  |  |  |         self.assertNotEqual(kall, _Call(('bar', (1, 2, 4)))) | 
					
						
							|  |  |  |         self.assertNotEqual(kall, _Call(('bar', (1, 2, 4), {}))) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         kall = _Call(({'a': 3},)) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(('', (), {'a': 3}))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(('', {'a': 3}))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(((), {'a': 3}))) | 
					
						
							|  |  |  |         self.assertEqual(kall, _Call(({'a': 3},))) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_empty__Call(self): | 
					
						
							|  |  |  |         args = _Call() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertEqual(args, ()) | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo',)) | 
					
						
							|  |  |  |         self.assertEqual(args, ((),)) | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo', ())) | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo',(), {})) | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo', {})) | 
					
						
							|  |  |  |         self.assertEqual(args, ({},)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_named_empty_call(self): | 
					
						
							|  |  |  |         args = _Call(('foo', (), {})) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo',)) | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo', ())) | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo',(), {})) | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo', {})) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertNotEqual(args, ((),)) | 
					
						
							|  |  |  |         self.assertNotEqual(args, ()) | 
					
						
							|  |  |  |         self.assertNotEqual(args, ({},)) | 
					
						
							|  |  |  |         self.assertNotEqual(args, ('bar',)) | 
					
						
							|  |  |  |         self.assertNotEqual(args, ('bar', ())) | 
					
						
							|  |  |  |         self.assertNotEqual(args, ('bar', {})) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_call_with_args(self): | 
					
						
							|  |  |  |         args = _Call(((1, 2, 3), {})) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertEqual(args, ((1, 2, 3),)) | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo', (1, 2, 3))) | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo', (1, 2, 3), {})) | 
					
						
							|  |  |  |         self.assertEqual(args, ((1, 2, 3), {})) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_named_call_with_args(self): | 
					
						
							|  |  |  |         args = _Call(('foo', (1, 2, 3), {})) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo', (1, 2, 3))) | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo', (1, 2, 3), {})) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertNotEqual(args, ((1, 2, 3),)) | 
					
						
							|  |  |  |         self.assertNotEqual(args, ((1, 2, 3), {})) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_call_with_kwargs(self): | 
					
						
							|  |  |  |         args = _Call(((), dict(a=3, b=4))) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertEqual(args, (dict(a=3, b=4),)) | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo', dict(a=3, b=4))) | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo', (), dict(a=3, b=4))) | 
					
						
							|  |  |  |         self.assertEqual(args, ((), dict(a=3, b=4))) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_named_call_with_kwargs(self): | 
					
						
							|  |  |  |         args = _Call(('foo', (), dict(a=3, b=4))) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo', dict(a=3, b=4))) | 
					
						
							|  |  |  |         self.assertEqual(args, ('foo', (), dict(a=3, b=4))) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertNotEqual(args, (dict(a=3, b=4),)) | 
					
						
							|  |  |  |         self.assertNotEqual(args, ((), dict(a=3, b=4))) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_call_with_args_call_empty_name(self): | 
					
						
							|  |  |  |         args = _Call(((1, 2, 3), {})) | 
					
						
							|  |  |  |         self.assertEqual(args, call(1, 2, 3)) | 
					
						
							|  |  |  |         self.assertEqual(call(1, 2, 3), args) | 
					
						
							| 
									
										
										
										
											2013-11-17 00:12:21 +02:00
										 |  |  |         self.assertIn(call(1, 2, 3), [args]) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_call_ne(self): | 
					
						
							|  |  |  |         self.assertNotEqual(_Call(((1, 2, 3),)), call(1, 2)) | 
					
						
							|  |  |  |         self.assertFalse(_Call(((1, 2, 3),)) != call(1, 2, 3)) | 
					
						
							|  |  |  |         self.assertTrue(_Call(((1, 2), {})) != call(1, 2, 3)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_call_non_tuples(self): | 
					
						
							|  |  |  |         kall = _Call(((1, 2, 3),)) | 
					
						
							|  |  |  |         for value in 1, None, self, int: | 
					
						
							|  |  |  |             self.assertNotEqual(kall, value) | 
					
						
							|  |  |  |             self.assertFalse(kall == value) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_repr(self): | 
					
						
							|  |  |  |         self.assertEqual(repr(_Call()), 'call()') | 
					
						
							|  |  |  |         self.assertEqual(repr(_Call(('foo',))), 'call.foo()') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertEqual(repr(_Call(((1, 2, 3), {'a': 'b'}))), | 
					
						
							|  |  |  |                          "call(1, 2, 3, a='b')") | 
					
						
							|  |  |  |         self.assertEqual(repr(_Call(('bar', (1, 2, 3), {'a': 'b'}))), | 
					
						
							|  |  |  |                          "call.bar(1, 2, 3, a='b')") | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertEqual(repr(call), 'call') | 
					
						
							|  |  |  |         self.assertEqual(str(call), 'call') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertEqual(repr(call()), 'call()') | 
					
						
							|  |  |  |         self.assertEqual(repr(call(1)), 'call(1)') | 
					
						
							|  |  |  |         self.assertEqual(repr(call(zz='thing')), "call(zz='thing')") | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertEqual(repr(call().foo), 'call().foo') | 
					
						
							|  |  |  |         self.assertEqual(repr(call(1).foo.bar(a=3).bing), | 
					
						
							|  |  |  |                          'call().foo.bar().bing') | 
					
						
							|  |  |  |         self.assertEqual( | 
					
						
							|  |  |  |             repr(call().foo(1, 2, a=3)), | 
					
						
							|  |  |  |             "call().foo(1, 2, a=3)" | 
					
						
							|  |  |  |         ) | 
					
						
							|  |  |  |         self.assertEqual(repr(call()()), "call()()") | 
					
						
							|  |  |  |         self.assertEqual(repr(call(1)(2)), "call()(2)") | 
					
						
							|  |  |  |         self.assertEqual( | 
					
						
							|  |  |  |             repr(call()().bar().baz.beep(1)), | 
					
						
							|  |  |  |             "call()().bar().baz.beep(1)" | 
					
						
							|  |  |  |         ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_call(self): | 
					
						
							|  |  |  |         self.assertEqual(call(), ('', (), {})) | 
					
						
							|  |  |  |         self.assertEqual(call('foo', 'bar', one=3, two=4), | 
					
						
							|  |  |  |                          ('', ('foo', 'bar'), {'one': 3, 'two': 4})) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = Mock() | 
					
						
							|  |  |  |         mock(1, 2, 3) | 
					
						
							|  |  |  |         mock(a=3, b=6) | 
					
						
							|  |  |  |         self.assertEqual(mock.call_args_list, | 
					
						
							|  |  |  |                          [call(1, 2, 3), call(a=3, b=6)]) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_attribute_call(self): | 
					
						
							|  |  |  |         self.assertEqual(call.foo(1), ('foo', (1,), {})) | 
					
						
							|  |  |  |         self.assertEqual(call.bar.baz(fish='eggs'), | 
					
						
							|  |  |  |                          ('bar.baz', (), {'fish': 'eggs'})) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = Mock() | 
					
						
							|  |  |  |         mock.foo(1, 2 ,3) | 
					
						
							|  |  |  |         mock.bar.baz(a=3, b=6) | 
					
						
							|  |  |  |         self.assertEqual(mock.method_calls, | 
					
						
							|  |  |  |                          [call.foo(1, 2, 3), call.bar.baz(a=3, b=6)]) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_extended_call(self): | 
					
						
							|  |  |  |         result = call(1).foo(2).bar(3, a=4) | 
					
						
							|  |  |  |         self.assertEqual(result, ('().foo().bar', (3,), dict(a=4))) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = MagicMock() | 
					
						
							|  |  |  |         mock(1, 2, a=3, b=4) | 
					
						
							|  |  |  |         self.assertEqual(mock.call_args, call(1, 2, a=3, b=4)) | 
					
						
							|  |  |  |         self.assertNotEqual(mock.call_args, call(1, 2, 3)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertEqual(mock.call_args_list, [call(1, 2, a=3, b=4)]) | 
					
						
							|  |  |  |         self.assertEqual(mock.mock_calls, [call(1, 2, a=3, b=4)]) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = MagicMock() | 
					
						
							|  |  |  |         mock.foo(1).bar()().baz.beep(a=6) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         last_call = call.foo(1).bar()().baz.beep(a=6) | 
					
						
							|  |  |  |         self.assertEqual(mock.mock_calls[-1], last_call) | 
					
						
							|  |  |  |         self.assertEqual(mock.mock_calls, last_call.call_list()) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_call_list(self): | 
					
						
							|  |  |  |         mock = MagicMock() | 
					
						
							|  |  |  |         mock(1) | 
					
						
							|  |  |  |         self.assertEqual(call(1).call_list(), mock.mock_calls) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = MagicMock() | 
					
						
							|  |  |  |         mock(1).method(2) | 
					
						
							|  |  |  |         self.assertEqual(call(1).method(2).call_list(), | 
					
						
							|  |  |  |                          mock.mock_calls) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = MagicMock() | 
					
						
							|  |  |  |         mock(1).method(2)(3) | 
					
						
							|  |  |  |         self.assertEqual(call(1).method(2)(3).call_list(), | 
					
						
							|  |  |  |                          mock.mock_calls) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = MagicMock() | 
					
						
							|  |  |  |         int(mock(1).method(2)(3).foo.bar.baz(4)(5)) | 
					
						
							|  |  |  |         kall = call(1).method(2)(3).foo.bar.baz(4)(5).__int__() | 
					
						
							|  |  |  |         self.assertEqual(kall.call_list(), mock.mock_calls) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_call_any(self): | 
					
						
							|  |  |  |         self.assertEqual(call, ANY) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         m = MagicMock() | 
					
						
							|  |  |  |         int(m) | 
					
						
							|  |  |  |         self.assertEqual(m.mock_calls, [ANY]) | 
					
						
							|  |  |  |         self.assertEqual([ANY], m.mock_calls) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_two_args_call(self): | 
					
						
							|  |  |  |         args = _Call(((1, 2), {'a': 3}), two=True) | 
					
						
							|  |  |  |         self.assertEqual(len(args), 2) | 
					
						
							|  |  |  |         self.assertEqual(args[0], (1, 2)) | 
					
						
							|  |  |  |         self.assertEqual(args[1], {'a': 3}) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         other_args = _Call(((1, 2), {'a': 3})) | 
					
						
							|  |  |  |         self.assertEqual(args, other_args) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class SpecSignatureTest(unittest.TestCase): | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _check_someclass_mock(self, mock): | 
					
						
							|  |  |  |         self.assertRaises(AttributeError, getattr, mock, 'foo') | 
					
						
							|  |  |  |         mock.one(1, 2) | 
					
						
							|  |  |  |         mock.one.assert_called_with(1, 2) | 
					
						
							|  |  |  |         self.assertRaises(AssertionError, | 
					
						
							|  |  |  |                           mock.one.assert_called_with, 3, 4) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, mock.one, 1) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock.two() | 
					
						
							|  |  |  |         mock.two.assert_called_with() | 
					
						
							|  |  |  |         self.assertRaises(AssertionError, | 
					
						
							|  |  |  |                           mock.two.assert_called_with, 3) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, mock.two, 1) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock.three() | 
					
						
							|  |  |  |         mock.three.assert_called_with() | 
					
						
							|  |  |  |         self.assertRaises(AssertionError, | 
					
						
							|  |  |  |                           mock.three.assert_called_with, 3) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, mock.three, 3, 2) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock.three(1) | 
					
						
							|  |  |  |         mock.three.assert_called_with(1) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock.three(a=1) | 
					
						
							|  |  |  |         mock.three.assert_called_with(a=1) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_basic(self): | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         mock = create_autospec(SomeClass) | 
					
						
							|  |  |  |         self._check_someclass_mock(mock) | 
					
						
							|  |  |  |         mock = create_autospec(SomeClass()) | 
					
						
							|  |  |  |         self._check_someclass_mock(mock) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_create_autospec_return_value(self): | 
					
						
							|  |  |  |         def f(): | 
					
						
							|  |  |  |             pass | 
					
						
							|  |  |  |         mock = create_autospec(f, return_value='foo') | 
					
						
							|  |  |  |         self.assertEqual(mock(), 'foo') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = create_autospec(Foo, return_value='foo') | 
					
						
							|  |  |  |         self.assertEqual(mock(), 'foo') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-06-09 17:31:59 +01:00
										 |  |  |     def test_autospec_reset_mock(self): | 
					
						
							|  |  |  |         m = create_autospec(int) | 
					
						
							|  |  |  |         int(m) | 
					
						
							|  |  |  |         m.reset_mock() | 
					
						
							|  |  |  |         self.assertEqual(m.__int__.call_count, 0) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     def test_mocking_unbound_methods(self): | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             def foo(self, foo): | 
					
						
							|  |  |  |                 pass | 
					
						
							|  |  |  |         p = patch.object(Foo, 'foo') | 
					
						
							|  |  |  |         mock_foo = p.start() | 
					
						
							|  |  |  |         Foo().foo(1) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock_foo.assert_called_with(1) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_create_autospec_unbound_methods(self): | 
					
						
							| 
									
										
										
										
											2012-04-21 18:22:28 +01:00
										 |  |  |         # see mock issue 128 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         # this is expected to fail until the issue is fixed | 
					
						
							|  |  |  |         return | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             def foo(self): | 
					
						
							|  |  |  |                 pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         klass = create_autospec(Foo) | 
					
						
							|  |  |  |         instance = klass() | 
					
						
							|  |  |  |         self.assertRaises(TypeError, instance.foo, 1) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # Note: no type checking on the "self" parameter | 
					
						
							|  |  |  |         klass.foo(1) | 
					
						
							|  |  |  |         klass.foo.assert_called_with(1) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, klass.foo) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_create_autospec_keyword_arguments(self): | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             a = 3 | 
					
						
							|  |  |  |         m = create_autospec(Foo, a='3') | 
					
						
							|  |  |  |         self.assertEqual(m.a, '3') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-04-21 18:22:28 +01:00
										 |  |  |     def test_create_autospec_keyword_only_arguments(self): | 
					
						
							|  |  |  |         def foo(a, *, b=None): | 
					
						
							|  |  |  |             pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         m = create_autospec(foo) | 
					
						
							|  |  |  |         m(1) | 
					
						
							|  |  |  |         m.assert_called_with(1) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, m, 1, 2) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         m(2, b=3) | 
					
						
							|  |  |  |         m.assert_called_with(2, b=3) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     def test_function_as_instance_attribute(self): | 
					
						
							|  |  |  |         obj = SomeClass() | 
					
						
							|  |  |  |         def f(a): | 
					
						
							|  |  |  |             pass | 
					
						
							|  |  |  |         obj.f = f | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = create_autospec(obj) | 
					
						
							|  |  |  |         mock.f('bing') | 
					
						
							|  |  |  |         mock.f.assert_called_with('bing') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_spec_as_list(self): | 
					
						
							|  |  |  |         # because spec as a list of strings in the mock constructor means | 
					
						
							|  |  |  |         # something very different we treat a list instance as the type. | 
					
						
							|  |  |  |         mock = create_autospec([]) | 
					
						
							|  |  |  |         mock.append('foo') | 
					
						
							|  |  |  |         mock.append.assert_called_with('foo') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertRaises(AttributeError, getattr, mock, 'foo') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             foo = [] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = create_autospec(Foo) | 
					
						
							|  |  |  |         mock.foo.append(3) | 
					
						
							|  |  |  |         mock.foo.append.assert_called_with(3) | 
					
						
							|  |  |  |         self.assertRaises(AttributeError, getattr, mock.foo, 'foo') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_attributes(self): | 
					
						
							|  |  |  |         class Sub(SomeClass): | 
					
						
							|  |  |  |             attr = SomeClass() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         sub_mock = create_autospec(Sub) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         for mock in (sub_mock, sub_mock.attr): | 
					
						
							|  |  |  |             self._check_someclass_mock(mock) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_builtin_functions_types(self): | 
					
						
							|  |  |  |         # we could replace builtin functions / methods with a function | 
					
						
							|  |  |  |         # with *args / **kwargs signature. Using the builtin method type | 
					
						
							|  |  |  |         # as a spec seems to work fairly well though. | 
					
						
							|  |  |  |         class BuiltinSubclass(list): | 
					
						
							|  |  |  |             def bar(self, arg): | 
					
						
							|  |  |  |                 pass | 
					
						
							|  |  |  |             sorted = sorted | 
					
						
							|  |  |  |             attr = {} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = create_autospec(BuiltinSubclass) | 
					
						
							|  |  |  |         mock.append(3) | 
					
						
							|  |  |  |         mock.append.assert_called_with(3) | 
					
						
							|  |  |  |         self.assertRaises(AttributeError, getattr, mock.append, 'foo') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock.bar('foo') | 
					
						
							|  |  |  |         mock.bar.assert_called_with('foo') | 
					
						
							|  |  |  |         self.assertRaises(TypeError, mock.bar, 'foo', 'bar') | 
					
						
							|  |  |  |         self.assertRaises(AttributeError, getattr, mock.bar, 'foo') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock.sorted([1, 2]) | 
					
						
							|  |  |  |         mock.sorted.assert_called_with([1, 2]) | 
					
						
							|  |  |  |         self.assertRaises(AttributeError, getattr, mock.sorted, 'foo') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock.attr.pop(3) | 
					
						
							|  |  |  |         mock.attr.pop.assert_called_with(3) | 
					
						
							|  |  |  |         self.assertRaises(AttributeError, getattr, mock.attr, 'foo') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_method_calls(self): | 
					
						
							|  |  |  |         class Sub(SomeClass): | 
					
						
							|  |  |  |             attr = SomeClass() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = create_autospec(Sub) | 
					
						
							|  |  |  |         mock.one(1, 2) | 
					
						
							|  |  |  |         mock.two() | 
					
						
							|  |  |  |         mock.three(3) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         expected = [call.one(1, 2), call.two(), call.three(3)] | 
					
						
							|  |  |  |         self.assertEqual(mock.method_calls, expected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock.attr.one(1, 2) | 
					
						
							|  |  |  |         mock.attr.two() | 
					
						
							|  |  |  |         mock.attr.three(3) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         expected.extend( | 
					
						
							|  |  |  |             [call.attr.one(1, 2), call.attr.two(), call.attr.three(3)] | 
					
						
							|  |  |  |         ) | 
					
						
							|  |  |  |         self.assertEqual(mock.method_calls, expected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_magic_methods(self): | 
					
						
							|  |  |  |         class BuiltinSubclass(list): | 
					
						
							|  |  |  |             attr = {} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = create_autospec(BuiltinSubclass) | 
					
						
							|  |  |  |         self.assertEqual(list(mock), []) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, int, mock) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, int, mock.attr) | 
					
						
							|  |  |  |         self.assertEqual(list(mock), []) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertIsInstance(mock['foo'], MagicMock) | 
					
						
							|  |  |  |         self.assertIsInstance(mock.attr['foo'], MagicMock) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_spec_set(self): | 
					
						
							|  |  |  |         class Sub(SomeClass): | 
					
						
							|  |  |  |             attr = SomeClass() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         for spec in (Sub, Sub()): | 
					
						
							|  |  |  |             mock = create_autospec(spec, spec_set=True) | 
					
						
							|  |  |  |             self._check_someclass_mock(mock) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             self.assertRaises(AttributeError, setattr, mock, 'foo', 'bar') | 
					
						
							|  |  |  |             self.assertRaises(AttributeError, setattr, mock.attr, 'foo', 'bar') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_descriptors(self): | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             @classmethod | 
					
						
							|  |  |  |             def f(cls, a, b): | 
					
						
							|  |  |  |                 pass | 
					
						
							|  |  |  |             @staticmethod | 
					
						
							|  |  |  |             def g(a, b): | 
					
						
							|  |  |  |                 pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         class Bar(Foo): | 
					
						
							|  |  |  |             pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         class Baz(SomeClass, Bar): | 
					
						
							|  |  |  |             pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         for spec in (Foo, Foo(), Bar, Bar(), Baz, Baz()): | 
					
						
							|  |  |  |             mock = create_autospec(spec) | 
					
						
							|  |  |  |             mock.f(1, 2) | 
					
						
							|  |  |  |             mock.f.assert_called_once_with(1, 2) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             mock.g(3, 4) | 
					
						
							|  |  |  |             mock.g.assert_called_once_with(3, 4) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_recursive(self): | 
					
						
							|  |  |  |         class A(object): | 
					
						
							|  |  |  |             def a(self): | 
					
						
							|  |  |  |                 pass | 
					
						
							|  |  |  |             foo = 'foo bar baz' | 
					
						
							|  |  |  |             bar = foo | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         A.B = A | 
					
						
							|  |  |  |         mock = create_autospec(A) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock() | 
					
						
							|  |  |  |         self.assertFalse(mock.B.called) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock.a() | 
					
						
							|  |  |  |         mock.B.a() | 
					
						
							|  |  |  |         self.assertEqual(mock.method_calls, [call.a(), call.B.a()]) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertIs(A.foo, A.bar) | 
					
						
							|  |  |  |         self.assertIsNot(mock.foo, mock.bar) | 
					
						
							|  |  |  |         mock.foo.lower() | 
					
						
							|  |  |  |         self.assertRaises(AssertionError, mock.bar.lower.assert_called_with) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_spec_inheritance_for_classes(self): | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |             def a(self, x): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                 pass | 
					
						
							|  |  |  |             class Bar(object): | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |                 def f(self, y): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                     pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         class_mock = create_autospec(Foo) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertIsNot(class_mock, class_mock()) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         for this_mock in class_mock, class_mock(): | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |             this_mock.a(x=5) | 
					
						
							|  |  |  |             this_mock.a.assert_called_with(x=5) | 
					
						
							|  |  |  |             this_mock.a.assert_called_with(5) | 
					
						
							|  |  |  |             self.assertRaises(TypeError, this_mock.a, 'foo', 'bar') | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             self.assertRaises(AttributeError, getattr, this_mock, 'b') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         instance_mock = create_autospec(Foo()) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         instance_mock.a(5) | 
					
						
							|  |  |  |         instance_mock.a.assert_called_with(5) | 
					
						
							|  |  |  |         instance_mock.a.assert_called_with(x=5) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, instance_mock.a, 'foo', 'bar') | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         self.assertRaises(AttributeError, getattr, instance_mock, 'b') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # The return value isn't isn't callable | 
					
						
							|  |  |  |         self.assertRaises(TypeError, instance_mock) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         instance_mock.Bar.f(6) | 
					
						
							|  |  |  |         instance_mock.Bar.f.assert_called_with(6) | 
					
						
							|  |  |  |         instance_mock.Bar.f.assert_called_with(y=6) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         self.assertRaises(AttributeError, getattr, instance_mock.Bar, 'g') | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         instance_mock.Bar().f(6) | 
					
						
							|  |  |  |         instance_mock.Bar().f.assert_called_with(6) | 
					
						
							|  |  |  |         instance_mock.Bar().f.assert_called_with(y=6) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         self.assertRaises(AttributeError, getattr, instance_mock.Bar(), 'g') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_inherit(self): | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             a = 3 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         Foo.Foo = Foo | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # class | 
					
						
							|  |  |  |         mock = create_autospec(Foo) | 
					
						
							|  |  |  |         instance = mock() | 
					
						
							|  |  |  |         self.assertRaises(AttributeError, getattr, instance, 'b') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         attr_instance = mock.Foo() | 
					
						
							|  |  |  |         self.assertRaises(AttributeError, getattr, attr_instance, 'b') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # instance | 
					
						
							|  |  |  |         mock = create_autospec(Foo()) | 
					
						
							|  |  |  |         self.assertRaises(AttributeError, getattr, mock, 'b') | 
					
						
							|  |  |  |         self.assertRaises(TypeError, mock) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # attribute instance | 
					
						
							|  |  |  |         call_result = mock.Foo() | 
					
						
							|  |  |  |         self.assertRaises(AttributeError, getattr, call_result, 'b') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_builtins(self): | 
					
						
							|  |  |  |         # used to fail with infinite recursion | 
					
						
							|  |  |  |         create_autospec(1) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         create_autospec(int) | 
					
						
							|  |  |  |         create_autospec('foo') | 
					
						
							|  |  |  |         create_autospec(str) | 
					
						
							|  |  |  |         create_autospec({}) | 
					
						
							|  |  |  |         create_autospec(dict) | 
					
						
							|  |  |  |         create_autospec([]) | 
					
						
							|  |  |  |         create_autospec(list) | 
					
						
							|  |  |  |         create_autospec(set()) | 
					
						
							|  |  |  |         create_autospec(set) | 
					
						
							|  |  |  |         create_autospec(1.0) | 
					
						
							|  |  |  |         create_autospec(float) | 
					
						
							|  |  |  |         create_autospec(1j) | 
					
						
							|  |  |  |         create_autospec(complex) | 
					
						
							|  |  |  |         create_autospec(False) | 
					
						
							|  |  |  |         create_autospec(True) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_function(self): | 
					
						
							|  |  |  |         def f(a, b): | 
					
						
							|  |  |  |             pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = create_autospec(f) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, mock) | 
					
						
							|  |  |  |         mock(1, 2) | 
					
						
							|  |  |  |         mock.assert_called_with(1, 2) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         mock.assert_called_with(1, b=2) | 
					
						
							|  |  |  |         mock.assert_called_with(a=1, b=2) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         f.f = f | 
					
						
							|  |  |  |         mock = create_autospec(f) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, mock.f) | 
					
						
							|  |  |  |         mock.f(3, 4) | 
					
						
							|  |  |  |         mock.f.assert_called_with(3, 4) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         mock.f.assert_called_with(a=3, b=4) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-04-13 17:39:16 +01:00
										 |  |  |     def test_skip_attributeerrors(self): | 
					
						
							|  |  |  |         class Raiser(object): | 
					
						
							|  |  |  |             def __get__(self, obj, type=None): | 
					
						
							|  |  |  |                 if obj is None: | 
					
						
							|  |  |  |                     raise AttributeError('Can only be accessed via an instance') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         class RaiserClass(object): | 
					
						
							|  |  |  |             raiser = Raiser() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             @staticmethod | 
					
						
							|  |  |  |             def existing(a, b): | 
					
						
							|  |  |  |                 return a + b | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         s = create_autospec(RaiserClass) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, lambda x: s.existing(1, 2, 3)) | 
					
						
							|  |  |  |         s.existing(1, 2) | 
					
						
							|  |  |  |         self.assertRaises(AttributeError, lambda: s.nonexisting) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # check we can fetch the raiser attribute and it has no spec | 
					
						
							|  |  |  |         obj = s.raiser | 
					
						
							|  |  |  |         obj.foo, obj.bar | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     def test_signature_class(self): | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             def __init__(self, a, b=3): | 
					
						
							|  |  |  |                 pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = create_autospec(Foo) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.assertRaises(TypeError, mock) | 
					
						
							|  |  |  |         mock(1) | 
					
						
							|  |  |  |         mock.assert_called_once_with(1) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         mock.assert_called_once_with(a=1) | 
					
						
							|  |  |  |         self.assertRaises(AssertionError, mock.assert_called_once_with, 2) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         mock(4, 5) | 
					
						
							|  |  |  |         mock.assert_called_with(4, 5) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         mock.assert_called_with(a=4, b=5) | 
					
						
							|  |  |  |         self.assertRaises(AssertionError, mock.assert_called_with, a=5, b=4) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_class_with_no_init(self): | 
					
						
							|  |  |  |         # this used to raise an exception | 
					
						
							|  |  |  |         # due to trying to get a signature from object.__init__ | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             pass | 
					
						
							|  |  |  |         create_autospec(Foo) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_signature_callable(self): | 
					
						
							|  |  |  |         class Callable(object): | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |             def __init__(self, x, y): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                 pass | 
					
						
							|  |  |  |             def __call__(self, a): | 
					
						
							|  |  |  |                 pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = create_autospec(Callable) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         mock(1, 2) | 
					
						
							|  |  |  |         mock.assert_called_once_with(1, 2) | 
					
						
							|  |  |  |         mock.assert_called_once_with(x=1, y=2) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         self.assertRaises(TypeError, mock, 'a') | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         instance = mock(1, 2) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         self.assertRaises(TypeError, instance) | 
					
						
							|  |  |  |         instance(a='a') | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         instance.assert_called_once_with('a') | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         instance.assert_called_once_with(a='a') | 
					
						
							|  |  |  |         instance('a') | 
					
						
							|  |  |  |         instance.assert_called_with('a') | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         instance.assert_called_with(a='a') | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         mock = create_autospec(Callable(1, 2)) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         mock(a='a') | 
					
						
							|  |  |  |         mock.assert_called_once_with(a='a') | 
					
						
							|  |  |  |         self.assertRaises(TypeError, mock) | 
					
						
							|  |  |  |         mock('a') | 
					
						
							|  |  |  |         mock.assert_called_with('a') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_signature_noncallable(self): | 
					
						
							|  |  |  |         class NonCallable(object): | 
					
						
							|  |  |  |             def __init__(self): | 
					
						
							|  |  |  |                 pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = create_autospec(NonCallable) | 
					
						
							|  |  |  |         instance = mock() | 
					
						
							|  |  |  |         mock.assert_called_once_with() | 
					
						
							|  |  |  |         self.assertRaises(TypeError, mock, 'a') | 
					
						
							|  |  |  |         self.assertRaises(TypeError, instance) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, instance, 'a') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = create_autospec(NonCallable()) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, mock) | 
					
						
							|  |  |  |         self.assertRaises(TypeError, mock, 'a') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_create_autospec_none(self): | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             bar = None | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock = create_autospec(Foo) | 
					
						
							|  |  |  |         none = mock.bar | 
					
						
							|  |  |  |         self.assertNotIsInstance(none, type(None)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         none.foo() | 
					
						
							|  |  |  |         none.foo.assert_called_once_with() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_autospec_functions_with_self_in_odd_place(self): | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             def f(a, self): | 
					
						
							|  |  |  |                 pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         a = create_autospec(Foo) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         a.f(10) | 
					
						
							|  |  |  |         a.f.assert_called_with(10) | 
					
						
							|  |  |  |         a.f.assert_called_with(self=10) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         a.f(self=10) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         a.f.assert_called_with(10) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         a.f.assert_called_with(self=10) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_autospec_property(self): | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             @property | 
					
						
							|  |  |  |             def foo(self): | 
					
						
							|  |  |  |                 return 3 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         foo = create_autospec(Foo) | 
					
						
							|  |  |  |         mock_property = foo.foo | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # no spec on properties | 
					
						
							| 
									
										
										
										
											2013-11-17 00:12:21 +02:00
										 |  |  |         self.assertIsInstance(mock_property, MagicMock) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         mock_property(1, 2, 3) | 
					
						
							|  |  |  |         mock_property.abc(4, 5, 6) | 
					
						
							|  |  |  |         mock_property.assert_called_once_with(1, 2, 3) | 
					
						
							|  |  |  |         mock_property.abc.assert_called_once_with(4, 5, 6) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_autospec_slots(self): | 
					
						
							|  |  |  |         class Foo(object): | 
					
						
							|  |  |  |             __slots__ = ['a'] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         foo = create_autospec(Foo) | 
					
						
							|  |  |  |         mock_slot = foo.a | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # no spec on slots | 
					
						
							|  |  |  |         mock_slot(1, 2, 3) | 
					
						
							|  |  |  |         mock_slot.abc(4, 5, 6) | 
					
						
							|  |  |  |         mock_slot.assert_called_once_with(1, 2, 3) | 
					
						
							|  |  |  |         mock_slot.abc.assert_called_once_with(4, 5, 6) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class TestCallList(unittest.TestCase): | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_args_list_contains_call_list(self): | 
					
						
							|  |  |  |         mock = Mock() | 
					
						
							|  |  |  |         self.assertIsInstance(mock.call_args_list, _CallList) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         mock(1, 2) | 
					
						
							|  |  |  |         mock(a=3) | 
					
						
							|  |  |  |         mock(3, 4) | 
					
						
							|  |  |  |         mock(b=6) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         for kall in call(1, 2), call(a=3), call(3, 4), call(b=6): | 
					
						
							| 
									
										
										
										
											2013-11-17 00:12:21 +02:00
										 |  |  |             self.assertIn(kall, mock.call_args_list) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         calls = [call(a=3), call(3, 4)] | 
					
						
							| 
									
										
										
										
											2013-11-17 00:12:21 +02:00
										 |  |  |         self.assertIn(calls, mock.call_args_list) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         calls = [call(1, 2), call(a=3)] | 
					
						
							| 
									
										
										
										
											2013-11-17 00:12:21 +02:00
										 |  |  |         self.assertIn(calls, mock.call_args_list) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         calls = [call(3, 4), call(b=6)] | 
					
						
							| 
									
										
										
										
											2013-11-17 00:12:21 +02:00
										 |  |  |         self.assertIn(calls, mock.call_args_list) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         calls = [call(3, 4)] | 
					
						
							| 
									
										
										
										
											2013-11-17 00:12:21 +02:00
										 |  |  |         self.assertIn(calls, mock.call_args_list) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-11-17 00:12:21 +02:00
										 |  |  |         self.assertNotIn(call('fish'), mock.call_args_list) | 
					
						
							|  |  |  |         self.assertNotIn([call('fish')], mock.call_args_list) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_call_list_str(self): | 
					
						
							|  |  |  |         mock = Mock() | 
					
						
							|  |  |  |         mock(1, 2) | 
					
						
							|  |  |  |         mock.foo(a=3) | 
					
						
							|  |  |  |         mock.foo.bar().baz('fish', cat='dog') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         expected = ( | 
					
						
							|  |  |  |             "[call(1, 2),\n" | 
					
						
							|  |  |  |             " call.foo(a=3),\n" | 
					
						
							|  |  |  |             " call.foo.bar(),\n" | 
					
						
							|  |  |  |             " call.foo.bar().baz('fish', cat='dog')]" | 
					
						
							|  |  |  |         ) | 
					
						
							|  |  |  |         self.assertEqual(str(mock.mock_calls), expected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_propertymock(self): | 
					
						
							|  |  |  |         p = patch('%s.SomeClass.one' % __name__, new_callable=PropertyMock) | 
					
						
							|  |  |  |         mock = p.start() | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             SomeClass.one | 
					
						
							|  |  |  |             mock.assert_called_once_with() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             s = SomeClass() | 
					
						
							|  |  |  |             s.one | 
					
						
							|  |  |  |             mock.assert_called_with() | 
					
						
							|  |  |  |             self.assertEqual(mock.mock_calls, [call(), call()]) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             s.one = 3 | 
					
						
							|  |  |  |             self.assertEqual(mock.mock_calls, [call(), call(), call(3)]) | 
					
						
							|  |  |  |         finally: | 
					
						
							|  |  |  |             p.stop() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-04-13 16:57:22 +01:00
										 |  |  |     def test_propertymock_returnvalue(self): | 
					
						
							|  |  |  |         m = MagicMock() | 
					
						
							|  |  |  |         p = PropertyMock() | 
					
						
							|  |  |  |         type(m).foo = p | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         returned = m.foo | 
					
						
							|  |  |  |         p.assert_called_once_with() | 
					
						
							|  |  |  |         self.assertIsInstance(returned, MagicMock) | 
					
						
							|  |  |  |         self.assertNotIsInstance(returned, PropertyMock) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | if __name__ == '__main__': | 
					
						
							|  |  |  |     unittest.main() |