unittest-mocker
Inspired by the pytest-mock, but written from scratch for using with unittest and convenient tool - patch_class
Installation
pip install unittest-mocker
Usage
General mocker
@activate_mocker
def test_create_client(self, mocker: Mocker):
post_mock = mocker.patch('requests.post')
post_mock.return_value.status_code = 200
post_mock.return_value.json.return_value = {'msg': done}
send_clients_to_crm()
post_mock.assert_called_once_with(...)
or you can put the mocking logic in a separate method:
def mock_post(mocker, status):
response = requests.Response()
response.status_code = status
response.headers = {'Location': '...'}
response._content = 'some content'
return mocker.patch('requests.post', return_value=response)
@activate_mocker
def test_upload_post_success(mocker):
mocked_post = mock_post(mocker, 201)
...
@activate_mocker
def test_upload_post_error(mocker):
mocked_post = mock_post(mocker, 500)
...
Details
mocker.patch under the hood calls unittest.mock.patch So you can apply any arguments of the standard mock library
The same with mocker.patch.object. It has the same parameters asunittest.mock.patch.object
Method patch_class
Basic usage:
from unittest_mocker import activate_mocker, Mocker
from api import ApiClient
@activate_mocker
def test_api_client(self, mocker: Mocker):
client_mock = mocker.patch_class('api.ApiClient')
ApiClient(url='example.com')(command='start')
client_mock.assert_initialized_with(url='example.com')
client_mock.mocks['__call__'].assert_called_once_with(command='start')
Syntax
# Variant 1
mocker.patch_class(target: str, methods=Optional[list[str]])
# Example:
mock = mocker.patch_class('some_module.SomeClass', methods=['method_1', 'method_2'])
# Variant 2
mocker.patch_class.object(target: Union[module, class], attribute: str, methods=Optional[list[str]])
# Example:
import some_module
mock = mocker.patch_class(some_module, 'SomeClass', methods=['method_1', 'method_2'])
All mocked methods are stored in mock.mocks dictionary and can be accessed:
mock.mocks['__init__'].assert_called_once_with(...)
patch_class always patches class's __init__ method.
Also, methods parameter also defaults to [__call__]. So that's correct:
mock = mocker.patch_class('some_module.SomeClass')
...
mock.mocks['__call__'].assert_called_once_with(...)
Helper assert_initialized_with
mock.assert_initialized_with() is just shortcut for mock.mocks['__init__'].assert_called_once_with()
14 Dec 6, 2022
1 Nov 17, 2022
12 Dec 26, 2022
3.2k Dec 31, 2022
3.8k Jan 3, 2023
909 Dec 15, 2022
7.8k Jan 2, 2023
19 Sep 10, 2022
3 Jan 23, 2022
2 Jan 31, 2022
56 Oct 29, 2022
23 Dec 05, 2022
6 Dec 14, 2022
1.6k Dec 09, 2022
30 Dec 04, 2022
103 Dec 21, 2022
4 Nov 27, 2022
1.1k Dec 28, 2022
2 Jan 16, 2022
66 Dec 21, 2022
24 Oct 25, 2022
3.8k Jan 06, 2023
93 Nov 16, 2022
13 Apr 15, 2022
13 Jul 07, 2022
2.3k Jan 01, 2023
111 Dec 27, 2022
30 Oct 26, 2022