Mega mocking capabilities - stop using dot-notated paths!
Project description
MegaMock - The Developer Experience Upgrade for Python Mocking
Pew pew! Sane defaults for mocking behavior! Patch objects, variables, attributes, etc by passing in the thing in question, rather than passing in dot-delimited path strings! Create tests faster than ever!
Supported Python Versions: 3.10+
New! Use the official GPT!
An OpenAI ChatGPT+ GPT has been created for MegaMock! Ask questions about library usage or have it generate tests! The GPT has been coached on test generation and should outperform vanilla GPT-4. The GPT is available at: https://chat.openai.com/g/g-DtZiDVQsz-python-megamock-test-generation-assistant
The GPT is experimental. It has a tendency to regress during conversations. If it starts mixing MegaMock and mock, remind it to double check that its following its instructions. Its been coached on a specific pytest style but you can tell it do something else. It tends to be aggressive with the mocking, so you'll need to reel it in and use your judgement on what ultimately makes sense to mock. Don't mock something you don't need to. Don't create unit tests using mocks that will just shadow required integration tests.
Installation
Pip installation:
pip install megamock
poetry (as a development dependency):
poetry add megamock --group=dev
Why Use MegaMock? (short version)
MegaMock is a library that provides a better interface for mocking and patching in Python. Its version
of patch doesn't have any gotchas based on how you import something, and it also automatically
creates mocks using best practices. Additionally, the generated mock types are unions of the mocked object
and MegaMock
, allowing you to better leverage your IDE's autocomplete.
Mock:
class_mock = mock.create_autospec(ClassICareAbout, instance=True)
# cmd / alt clicking on "method_call" doesn't direct you to the definition
class_mock.method_call.return_value = "some value"
# can't simply cmd / alt click and go to ClassICareAbout
with mock.patch("some.hard.to.remember.and.long.dot.path.ClassICareAbout", class_mock) as mock_instance:
do_something()
MegaMock:
# cmd / alt clicking on ClassICareAbout takes you to the definition
patch = MegaPatch.it(ClassICareAbout)
mock_instance = patch.megainstance
# cmd / alt clicking on "method_call" directs you to the definition
mock_instance.method_call.return_value = "some value"
do_something()
Why Use MegaMock? (long version)
MegaMock was created to address some shortcomings in the built-in Python library:
- Legacy code holds back "best practice" defaults, so many developers write sub-optimal mocks that allow things that should not be allowed. Likewise, writing better mocks are more work, so there's a tendency to write simpler code because, at that point in time, the developer felt that is all that was needed. MegaMock's simple interface provides sane defaults.
mock.patch
is very commonly used, and can work well whenautospec=True
, but has the drawback that you need to pass in a string to the thing that is being patched. Most (all?) IDEs do not properly recognize these strings as references to the objects that are being patched, and thus automated refactoring and reference finding skips them. Likewise, automatically getting a dot referenced path to an object is also commonly missing functionality. This all adds additional burden to the developer. WithMegaPatch
, you can import an object as you normally would into the test, then pass in thing itself you want to patch. This even works for methods, attributes, and nested classes! Additionally, your IDE's autocomplete for attributes will work in many situations as well!mock.patch
has a gotcha where the string you provide must match where the reference lives. So, for example, if you have inmy_module.py
:from other_module import Thing
, then doingmock.patch("other_module.Thing")
won't actually work, because the reference inmy_module
still points to the original. You can work around this by doingimport other_module
and referencingThing
byother_module.Thing
. MegaMock does not have this problem, and it doesn't matter how you import.
Features
See the full features list.
Example Usage
Production Code
from module.submodule import MyClassToMock
def my_method(...):
...
a_thing = MyClassToMock(...)
do_something_with_a_thing(a_thing)
...
def do_something_with_a_thing(a_thing: MyClassToMock) -> None:
result = a_thing.some_method(...)
if result == "a value":
...
Test Code
from megamock import MegaPatch
from module.submodule import MyClassToMock
def test_something(...):
patch = MegaPatch.it(MyClassToMock.some_method)
patch.return_value = "a value"
my_method(...)
Documentation
Usage (pytest)
With pytest, MegaMock is easily leveraged by using the included pytest plugin. You can use it by adding -p megamock.plugins.pytest
to the command line.
Command line example:
pytest -p megamock.plugins.pytest
pyproject.toml
example:
[tool.pytest.ini_options]
addopts = "-p megamock.plugins.pytest"
The pytest plugin also automatically stops MegaPatch
es after each test. If pytest-mock
is installed, the default mocker will be switched to the pytest-mock
mocker
.
Usage (other test frameworks)
If you're not using the pytest plugin, import and execution order is important for MegaMock. When running tests, you will need to execute the start_import_mod
function prior to importing any production or test code. You will also want it so the loader is not used in production.
Core Classes
MegaMock
- the primary class for a mocked object. This is similar to MagicMock
. To create a mock instance of a class, use MegaMock.it(MyClass)
to make MyClass
the spec. To create a mock class (the type) use MegaMock.the_class(MyClass)
. To create mock instances of instantiated objects, functions, etc, use MegaMock.this(some_object)
.
MegaPatch
- the class for patching. This is similar to patch
. Use MegaPath.it(MyObject)
to replace new instances of the MyObject
class.
Mega
- helper class for accessing mock attributes without having to memorize them due to lost type inference. Use Mega(some_megamock)
.
Note that the assert_
methods, such as assert_called_once
, is now called_once
and returns a boolean. The assertion error
is stored in Mega.last_assertion_error
. This is typically for doing asserts against mocked functions and methods.
Dependency injection example:
from megamock import MegaMock
def test_something(...):
manager = MegaMock.it(MyManagerClass)
service = SomeService(manager)
...
MegaPatch example:
from elsewhere import Service
from megamock import MegaPatch
def test_something(...):
patched = MegaPatch.it(Service.make_external_call)
patched.return_value = SomeValue(...)
service = SomeService(...)
code_under_test(service)
...
MegaMock
objects have the same attributes as regular MagicMock
s plus megamock
and megainstance
.
For example, my_mega_mock.megamock.spy
is the object being spied, if set. my_class_mock.megainstance
is the instance returned when the class is instantiated. Note that you typically access the megainstance with MegaMock(my_class_mock).megainstance
due to limitations in the type system.
The guidance document is available to provide in-depth information on using mocking and MegaMock. Continuing reading to quickly jump in to examples.
Learning By Example
All examples below have the following imports:
from my_module import MyClass
from megamock import Mega, MegaMock, MegaPatch
Creating a mock instance of a class:
mock_instance = MegaMock.it(MyClass)
Creating a mock class itself:
mock_class = MegaMock.the_class(MyClass)
func_that_wants_a_type(mock_class)
Spying an object:
my_thing = MyClass()
spied_class = MegaMock.this(spy=my_thing)
# ... do stuff with spied_class...
Mega(spied_class.some_method).call_args_list # same as wraps
# check whether a value was accessed
# if things aren't as expected, you can pull up the debugger and see the stack traces
assert len(spied_class.megamock.spied_access["some_attribute"]) == 1
spy_access_list = spied_class.megamock.spied_access["some_attribute"]
spy_access: SpyAccess = spy_access_list[0]
spy_access.attr_value # shallow copy of what was returned
spy_access.stacktrace # where the access happened
spy_access.time # when it happened (from time.time())
spy_access.top_of_stacktrace # a shorthand property intended to be used when debugging in the IDE
spy_access.format_stacktrace() # return a list of strings for the stacktrace
spy_access.print_stacktrace() # display the stacktrace to the console
Patching a class:
mock_patch = MegaPatch.it(MyClass)
# the class itself
mock_patch.new_value
# the class instance
mock_patch.megainstance
# the return value of the __call__ method on the class
mock_patch.megainstance.return_value
Patching a class attribute:
# temporarily update the max retries to 0
mega_patch = MegaPatch.it(MyClass.max_retries, new=0)
Patching a class method:
mega_patch = MegaPatch.it(MyClass.my_method, return_value=...)
Alternatively:
mega_patch = MegaPatch.it(MyClass.my_method)
mega_patch.mock.return_value = ...
mega_patch = MegaPatch.it(MyClass.my_method)
mega_patch.new_value.return_value = ...
You can also alter the return value of your mock without creating a separate mock object first.
mega_patch.return_value.user = SomeUser()
Working with MegaPatch
and classes:
mega_patch.new_value
is the class type itself
mega_patch = MegaPatch.it(MyClass)
mega_patch.new_value.x is MyClass.x
mega_patch.return_value
is the class instance returned. However, there is the property
megainstance
which is preferred because it has better type hinting.
mega_patch = MegaPatch.it(MyClass)
# instead of this, for which the static type is Any:
mega_patch.return_value is MyClass()
# use this, which has a static type of MegaMock | MyClass:
mega_patch.megainstance is MyClass()
Patching a module attribute:
import my_module
MegaPatch.it(my_module.some_attribute, new=...)
Patching a method of a nested class:
import my_module
MegaPatch.it(
my_module.MyClass.MyNestedClass.some_method,
return_value=...
)
Setting the return value:
my_mock.my_method.return_value = "foo"
Turning on real logic:
import my_module
mega_patch = MegaPatch.it(my_module.SomeClass)
Mega(mega_patch.megainstance.some_pure_logic_method).use_real_logic()
do_something_that_invokes_that_function(...)
Congrats on Reading This Far! Here's an Art Gallery!
Nobody said it was a big art gallery. Feel free to submit a PR that helps fix that. No artistic skill required.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file megamock-0.1.0b9.tar.gz
.
File metadata
- Download URL: megamock-0.1.0b9.tar.gz
- Upload date:
- Size: 26.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/4.0.2 CPython/3.11.6
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 91cbc062cf21dea96f1707159a3fa88c60b78b854f4ccf066cce7a472f3fe513 |
|
MD5 | ac448a0f555e16a987f7b4cec29ccafd |
|
BLAKE2b-256 | 2fc6a0ae91c96a21688dca4553d1ede37f56768fd2bc466ae77c3c777ea32ee2 |
File details
Details for the file megamock-0.1.0b9-py3-none-any.whl
.
File metadata
- Download URL: megamock-0.1.0b9-py3-none-any.whl
- Upload date:
- Size: 26.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/4.0.2 CPython/3.11.6
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | bc5fc960e0c4635dc0964061f2de26e536333c27b4abf83b2df503ec365346e0 |
|
MD5 | 162ca9bf12bae22c52bb6c031ea3df6b |
|
BLAKE2b-256 | 95ba0df2df612c95bc907706394e3e73f23a928fdcdb22efb00b0592d79839e7 |