test_constants.py 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. from unittest.mock import patch
  2. from sentry.constants import (
  3. INTEGRATION_ID_TO_PLATFORM_DATA,
  4. get_integration_id_for_event,
  5. get_integration_id_for_marketing_slug,
  6. )
  7. mock_integration_ids = {
  8. "java": {},
  9. "java-log4j": {},
  10. "java-log4j2": {},
  11. "java-android": {},
  12. "javascript": {},
  13. }
  14. def test_marketing_slug_to_integration_id():
  15. with patch.dict(INTEGRATION_ID_TO_PLATFORM_DATA, mock_integration_ids):
  16. assert get_integration_id_for_marketing_slug("java") == "java"
  17. # kotlin uses the java library, too
  18. assert get_integration_id_for_marketing_slug("kotlin") == "java"
  19. assert get_integration_id_for_marketing_slug("android") == "java-android"
  20. assert get_integration_id_for_marketing_slug("foobar") is None
  21. def test_integration_id_for_event():
  22. with patch.dict(INTEGRATION_ID_TO_PLATFORM_DATA, mock_integration_ids):
  23. assert get_integration_id_for_event("java", "sentry-java", None) == "java"
  24. assert get_integration_id_for_event("java", "raven-java", None) == "java"
  25. assert get_integration_id_for_event("java", "raven-java:log4j", None) == "java-log4j"
  26. assert get_integration_id_for_event("java", "sentry-java", ["android"]) == "java-android"
  27. assert (
  28. get_integration_id_for_event("java", "sentry-java", ["foobar", "log4j2"])
  29. == "java-log4j2"
  30. )
  31. assert get_integration_id_for_event("foobar", "sentry-java", None) == "java"
  32. assert get_integration_id_for_event("java", "foobar", None) == "java"
  33. assert get_integration_id_for_event("foobar", "foobar", None) is None