16 lines
1023 B
Python
16 lines
1023 B
Python
|
def assert_calls(method, expected_calls):
|
||
|
|
method.assert_has_calls(expected_calls, any_order=True)
|
||
|
|
assert method.call_count == len(expected_calls)
|
||
|
|
|
||
|
|
|
||
|
|
def assert_db_calls(db, expected_calls):
|
||
|
|
assert_calls(db.add_symbol_legacy_definition, expected_calls.get('add_symbol_legacy_definition', []))
|
||
|
|
assert_calls(db.add_symbol_definition, expected_calls.get('add_symbol_definition', []))
|
||
|
|
assert_calls(db.add_symbol_usage_in_legacy, expected_calls.get('add_symbol_usage_in_legacy', []))
|
||
|
|
assert_calls(db.add_symbol_usage, expected_calls.get('add_symbol_usage', []))
|
||
|
|
assert_calls(db.add_symbol_legacy_usage, expected_calls.get('add_symbol_legacy_usage', []))
|
||
|
|
assert_calls(db.add_symbol_select, expected_calls.get('add_symbol_select', []))
|
||
|
|
assert_calls(db.add_symbol_helper, expected_calls.get('add_symbol_helper', []))
|
||
|
|
assert_calls(db.add_symbol_legacy_note, expected_calls.get('add_symbol_legacy_note', []))
|
||
|
|
assert_calls(db.add_symbol_virtual, expected_calls.get('add_symbol_virtual', []))
|