diff options
author | Kushal Das <kushaldas@gmail.com> | 2014-04-16 18:02:21 (GMT) |
---|---|---|
committer | Kushal Das <kushaldas@gmail.com> | 2014-04-16 18:02:21 (GMT) |
commit | 8c14534df6c7bd561fac31985fba60306e181265 (patch) | |
tree | 6a5b70b694cef522be062194b0d6953fb412ab3a /Lib/unittest/test/testmock | |
parent | c3ac9af6d00ec9b5939a0ce31984113ddb9b0a9c (diff) | |
download | cpython-8c14534df6c7bd561fac31985fba60306e181265.zip cpython-8c14534df6c7bd561fac31985fba60306e181265.tar.gz cpython-8c14534df6c7bd561fac31985fba60306e181265.tar.bz2 |
Closes Issue 21238: New keyword argument `unsafe` to Mock.
It raises `AttributeError` incase of an attribute startswith assert
or assret.
Diffstat (limited to 'Lib/unittest/test/testmock')
-rw-r--r-- | Lib/unittest/test/testmock/testmock.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/Lib/unittest/test/testmock/testmock.py b/Lib/unittest/test/testmock/testmock.py index 23675b9..59353a0 100644 --- a/Lib/unittest/test/testmock/testmock.py +++ b/Lib/unittest/test/testmock/testmock.py @@ -1187,6 +1187,17 @@ class MockTest(unittest.TestCase): m = mock.create_autospec(object(), name='sweet_func') self.assertIn('sweet_func', repr(m)) + #Issue21238 + def test_mock_unsafe(self): + m = Mock() + with self.assertRaises(AttributeError): + m.assert_foo_call() + with self.assertRaises(AttributeError): + m.assret_foo_call() + m = Mock(unsafe=True) + m.assert_foo_call() + m.assret_foo_call() + def test_mock_add_spec(self): class _One(object): one = 1 |