def _pytest_mock_dependencies_generate(dependencies, prepare_asserts_calls,
include_mock_autogen_import,
mock_autogen_alias):
generated_code = ""
unique_dependencies = set()
mock_names = []
if dependencies:
generated_code += "# mocked dependencies\n"
for (
obj_path,
obj_name,
) in dependencies:
unique_name = get_unique_item(unique_dependencies, obj_name)
generated_mock_name, generated_mock_code = \
_single_pytest_mock_module_entry_with_alias(unique_name,
obj_name,
obj_path)
mock_names.append(generated_mock_name)
generated_code += generated_mock_code
if prepare_asserts_calls and mock_names:
generated_code += "# calls to generate_asserts, " \
"put this after the 'act'\n"
generated_code = generated_code if not include_mock_autogen_import else \
generated_code + f"import {mock_autogen_alias}\n"
for mock_name in mock_names:
generated_code += _single_call_to_generate_asserts(
mock_name, mock_autogen_alias)
return generated_code
def test_get_unique_item_exist_multiple():
items = set(['a', 'a_2', 'b'])
assert 'a_3' == get_unique_item(items, 'a')
assert 'a' in items
assert 'a_2' in items
assert 'a_3' in items
assert 'b' in items
assert len(items) == 4
def test_get_unique_item_unique():
items = set('b')
assert 'a' == get_unique_item(items, 'a')
assert 'a' in items
assert len(items) == 2
def test_get_unique_item_exist():
items = set('a')
assert 'a_2' == get_unique_item(items, 'a')
assert 'a' in items
assert 'a_2' in items
assert len(items) == 2
def test_get_unique_item_empty_set():
items = set()
assert 'a' == get_unique_item(items, 'a')
assert 'a' in items