Mock and MagicMock in PyTest - Build an Automation Script

Automate testing a function using MagicMock to mock a dependency

Preconditions (2)

Step 1: Create a function 'get_data' that calls an external API function 'fetch_api_data'.

Step 2: Mock 'fetch_api_data' using MagicMock to return a fixed dictionary {'key': 'value'}.

Step 3: Call 'get_data' and verify it returns the mocked dictionary.

Step 4: Verify that 'fetch_api_data' was called exactly once during the test.

✅ Expected Result: The test passes confirming 'get_data' returns the mocked data and 'fetch_api_data' was called once.

Automation Requirements - pytest with unittest.mock

Assertions Needed:

Assert that the return value of 'get_data' matches the mocked dictionary

Assert that the mocked 'fetch_api_data' was called exactly once

Best Practices:

Use MagicMock to mock dependencies

Use patch decorator or context manager to replace the real function

Use assert_called_once() to verify call count

Keep test isolated from real external calls

Automated Solution

PyTest

from unittest.mock import MagicMock, patch
import pytest

# Module code to test
def fetch_api_data():
    # Imagine this calls an external API
    return {'real': 'data'}

def get_data():
    return fetch_api_data()

# Test code
def test_get_data_with_mock():
    with patch('__main__.fetch_api_data', new=MagicMock(return_value={'key': 'value'})) as mock_fetch:
        result = get_data()
        assert result == {'key': 'value'}, f"Expected mocked data, got {result}"
        mock_fetch.assert_called_once()

The test uses patch as a context manager to replace fetch_api_data with a MagicMock that returns a fixed dictionary.

Calling get_data() then returns the mocked data instead of calling the real API.

Assertions check that the returned data matches the mock and that the mocked function was called exactly once.

This keeps the test isolated and fast without real API calls.

Common Mistakes - 3 Pitfalls

Not using patch to replace the real function, so the real API call runs during the test

Using MagicMock without setting return_value, so the mock returns another MagicMock instead of expected data

Not verifying that the mocked function was called, missing important behavior checks

Bonus Challenge

Now add data-driven testing with 3 different mocked return values for 'fetch_api_data' and verify 'get_data' returns each correctly.

Show Hint

Practice

1. What is the main difference between Mock and MagicMock in pytest?

easy

A. MagicMock supports special methods like __len__ and __getitem__, Mock does not.

B. Mock can only mock functions, while MagicMock can mock classes.

C. Mock is used for asynchronous code, MagicMock is not.

D. MagicMock is slower and less flexible than Mock.

Solution

1. Step 1: Understand the purpose of Mock and MagicMock

   Mock creates fake objects for testing, but does not automatically support Python's special methods like __len__ or __getitem__.

2. Step 2: Recognize MagicMock's special feature

   MagicMock extends Mock by including default implementations for many special methods, making it useful when those methods are called during tests.

3. Final Answer:

   MagicMock supports special methods like __len__ and __getitem__, Mock does not. -> Option A

4. Quick Check:

   Special methods = MagicMock [OK]

Hint: MagicMock handles special methods automatically [OK]

Common Mistakes:

• Thinking Mock supports special methods by default
• Confusing MagicMock as only for classes
• Believing Mock is for async only

2. Which of the following is the correct way to create a MagicMock object in pytest?

easy

A. mock = MagicMock[]

B. mock = MagicMock()

C. mock = MagicMock

D. mock = MagicMock{}

Solution

1. Step 1: Recall the syntax for creating instances in Python

   To create an instance of a class, you call it with parentheses, e.g., MagicMock().

2. Step 2: Identify the correct syntax for MagicMock

   MagicMock is a class, so MagicMock() creates a new mock object. The other options use invalid syntax for instantiation.

3. Final Answer:

   mock = MagicMock() -> Option B

4. Quick Check:

   Instance creation uses parentheses () [OK]

Hint: Use parentheses () to create MagicMock instances [OK]

Common Mistakes:

• Forgetting parentheses when creating mock objects
• Using square or curly brackets instead of parentheses
• Assigning class without instantiation

3. What will be the output of this pytest code snippet?

from unittest.mock import MagicMock
mock = MagicMock()
mock.__len__.return_value = 5
print(len(mock))

medium

A. 0

B. TypeError

C. 5

D. AttributeError

Solution

1. Step 1: Understand MagicMock and special methods

   Setting mock.__len__.return_value = 5 tells the mock to return 5 when len() is called on it, because len() calls the __len__ method.

2. Step 2: Predict the output of print(len(mock))

   Since __len__ is mocked to return 5, len(mock) will print 5.

3. Final Answer:

   5 -> Option C

4. Quick Check:

   len(mock) returns mocked __len__ value [OK]

Hint: Set __len__.return_value to mock len() calls [OK]

Common Mistakes:

• Expecting default length 0 instead of mocked value
• Confusing len(mock) with attribute access
• Thinking TypeError occurs without __len__

4. Identify the error in this pytest mocking code:

from unittest.mock import Mock
mock = Mock()
mock.__getitem__.return_value = 'value'
print(mock[0])

medium

A. Mock does not support special methods like __getitem__ by default.

B. The syntax for setting return_value is incorrect.

C. You cannot use indexing on mock objects.

D. Missing import for MagicMock.

Solution

1. Step 1: Check support for special methods in Mock

   Mock does not implement special methods like __getitem__ automatically, so setting mock.__getitem__.return_value has no effect on indexing.

2. Step 2: Understand why the code fails

   Using mock[0] calls __getitem__, but since Mock does not handle it, this raises a TypeError. To mock indexing, use MagicMock.

3. Final Answer:

   Mock does not support special methods like __getitem__ by default. -> Option A

4. Quick Check:

   Special methods need MagicMock [OK]

Hint: Use MagicMock for special methods like indexing [OK]

Common Mistakes:

• Assuming Mock supports all special methods
• Confusing syntax errors with capability errors
• Not importing MagicMock when needed

5. You want to test a function that calls obj.__str__() internally. Which mock object and setup will correctly simulate str(obj) returning 'mocked string'?

hard

A. Use mock = Mock() and set mock.str = 'mocked string'

B. Use mock = Mock() and set mock.__str__ = lambda: 'mocked string'

C. Use mock = MagicMock() and set mock.str.return_value = 'mocked string'

D. Use mock = MagicMock() and set mock.__str__.return_value = 'mocked string'

Solution

1. Step 1: Identify the need to mock __str__ special method

   The str(obj) call internally uses the __str__ method, which is a special method.

2. Step 2: Choose the correct mock and setup

   MagicMock supports special methods like __str__. Setting mock.__str__.return_value = 'mocked string' makes str(mock) return 'mocked string'.

3. Step 3: Eliminate incorrect options

   Use mock = Mock() and set mock.__str__ = lambda: 'mocked string' uses Mock which does not support special methods by default. Options C and D use incorrect attribute names or types.

4. Final Answer:

   Use mock = MagicMock() and set mock.__str__.return_value = 'mocked string' -> Option D

5. Quick Check:

   Mock special methods with MagicMock and return_value [OK]

Hint: Set __str__.return_value on MagicMock for str() calls [OK]

Common Mistakes:

• Using Mock instead of MagicMock for special methods
• Setting wrong attribute names like str instead of __str__
• Assigning string directly instead of return_value