diff --git a/sentry_sdk/integrations/pymongo.py b/sentry_sdk/integrations/pymongo.py index 0a94d46813..0b057fe548 100644 --- a/sentry_sdk/integrations/pymongo.py +++ b/sentry_sdk/integrations/pymongo.py @@ -2,6 +2,7 @@ import copy from sentry_sdk import Hub +from sentry_sdk.consts import SPANDATA from sentry_sdk.hub import _should_send_default_pii from sentry_sdk.integrations import DidNotEnable, Integration from sentry_sdk.tracing import Span @@ -119,10 +120,11 @@ def started(self, event): except TypeError: pass - data = {"operation_ids": {}} # type: Dict[str, Dict[str, Any]] + data = {"operation_ids": {}} # type: Dict[str, Any] data["operation_ids"]["operation"] = event.operation_id data["operation_ids"]["request"] = event.request_id + data[SPANDATA.DB_SYSTEM] = "mongodb" try: lsid = command.pop("lsid")["id"] diff --git a/tests/integrations/pymongo/test_pymongo.py b/tests/integrations/pymongo/test_pymongo.py index 16438ac971..786c775e41 100644 --- a/tests/integrations/pymongo/test_pymongo.py +++ b/tests/integrations/pymongo/test_pymongo.py @@ -1,4 +1,5 @@ from sentry_sdk import capture_message, start_transaction +from sentry_sdk.consts import SPANDATA from sentry_sdk.integrations.pymongo import PyMongoIntegration, _strip_pii from mockupdb import MockupDB, OpQuery @@ -55,6 +56,7 @@ def test_transactions(sentry_init, capture_events, mongo_server, with_pii): "net.peer.port": str(mongo_server.port), } for span in find, insert_success, insert_fail: + assert span["data"][SPANDATA.DB_SYSTEM] == "mongodb" for field, value in common_tags.items(): assert span["tags"][field] == value