fix tests

getsentry · Jul 28, 2023 · 0be3b1b · 0be3b1b
1 parent dfa12a4
commit 0be3b1b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/sentry_sdk/integrations/pymongo.py b/sentry_sdk/integrations/pymongo.py
@@ -86,7 +86,7 @@ def _strip_pii(command):
 
 
 def _get_db_data(event):
-    # type: (Union[CommandFailedEvent, CommandStartedEvent, CommandSucceededEvent]) -> Dict[str, Any]
+    # type: (Any) -> Dict[str, Any]
     data = {}
 
     data[SPANDATA.DB_SYSTEM] = "mongodb"

diff --git a/tests/integrations/sqlalchemy/test_sqlalchemy.py b/tests/integrations/sqlalchemy/test_sqlalchemy.py
@@ -123,12 +123,8 @@ class Address(Base):
     for span in event["spans"]:
         assert span["data"][SPANDATA.DB_SYSTEM] == "sqlite"
         assert span["data"][SPANDATA.DB_NAME] == ":memory:"
-        assert (
-            span["data"][SPANDATA.SERVER_ADDRESS] is None
-        )  # sqlite does not have a server
-        assert (
-            span["data"][SPANDATA.SERVER_PORT] is None
-        )  # sqlite does not have a server
+        assert SPANDATA.SERVER_ADDRESS not in span["data"]
+        assert SPANDATA.SERVER_PORT not in span["data"]
 
     assert (
         render_span_tree(event)