test_util.py 1023 B

123456789101112131415
  1. def assert_calls(method, expected_calls):
  2. method.assert_has_calls(expected_calls, any_order=True)
  3. assert method.call_count == len(expected_calls)
  4. def assert_db_calls(db, expected_calls):
  5. assert_calls(db.add_symbol_legacy_definition, expected_calls.get('add_symbol_legacy_definition', []))
  6. assert_calls(db.add_symbol_definition, expected_calls.get('add_symbol_definition', []))
  7. assert_calls(db.add_symbol_usage_in_legacy, expected_calls.get('add_symbol_usage_in_legacy', []))
  8. assert_calls(db.add_symbol_usage, expected_calls.get('add_symbol_usage', []))
  9. assert_calls(db.add_symbol_legacy_usage, expected_calls.get('add_symbol_legacy_usage', []))
  10. assert_calls(db.add_symbol_select, expected_calls.get('add_symbol_select', []))
  11. assert_calls(db.add_symbol_helper, expected_calls.get('add_symbol_helper', []))
  12. assert_calls(db.add_symbol_legacy_note, expected_calls.get('add_symbol_legacy_note', []))
  13. assert_calls(db.add_symbol_virtual, expected_calls.get('add_symbol_virtual', []))