tests

Author: alexander-alderman-webb

tests/integrations/clickhouse_driver/test_clickhouse_driver.py

@@ -5,6 +5,8 @@
 ```
 """
 
+from unittest import mock
+
 import clickhouse_driver
 from clickhouse_driver import Client, connect
 
@@ -631,6 +633,8 @@ def test_clickhouse_dbapi_breadcrumbs_with_pii(sentry_init, capture_events) -> N
                 "server.address": "localhost",
                 "server.port": 9000,
                 "db.result": [[], []],
+                "thread.id": mock.ANY,
+                "thread.name": mock.ANY,
             },
             "message": "DROP TABLE IF EXISTS test",
             "type": "default",
@@ -644,6 +648,8 @@ def test_clickhouse_dbapi_breadcrumbs_with_pii(sentry_init, capture_events) -> N
                 "server.address": "localhost",
                 "server.port": 9000,
                 "db.result": [[], []],
+                "thread.id": mock.ANY,
+                "thread.name": mock.ANY,
             },
             "message": "CREATE TABLE test (x Int32) ENGINE = Memory",
             "type": "default",
@@ -657,6 +663,8 @@ def test_clickhouse_dbapi_breadcrumbs_with_pii(sentry_init, capture_events) -> N
                 "server.address": "localhost",
                 "server.port": 9000,
                 "db.params": [{"x": 100}],
+                "thread.id": mock.ANY,
+                "thread.name": mock.ANY,
             },
             "message": "INSERT INTO test (x) VALUES",
             "type": "default",
@@ -670,6 +678,8 @@ def test_clickhouse_dbapi_breadcrumbs_with_pii(sentry_init, capture_events) -> N
                 "server.address": "localhost",
                 "server.port": 9000,
                 "db.params": [[170], [200]],
+                "thread.id": mock.ANY,
+                "thread.name": mock.ANY,
             },
             "message": "INSERT INTO test (x) VALUES",
             "type": "default",
@@ -683,7 +693,9 @@ def test_clickhouse_dbapi_breadcrumbs_with_pii(sentry_init, capture_events) -> N
                 "server.address": "localhost",
                 "server.port": 9000,
                 "db.params": {"minv": 150},
-                "db.result": [[["370"]], [["'sum(x)'", "'Int64'"]]],
+                "db.result": [[[370]], [["sum(x)", "Int64"]]],
+                "thread.id": mock.ANY,
+                "thread.name": mock.ANY,
             },
             "message": "SELECT sum(x) FROM test WHERE x > 150",
             "type": "default",
@@ -699,7 +711,7 @@ def test_clickhouse_dbapi_breadcrumbs_with_pii(sentry_init, capture_events) -> N
     for crumb in event["breadcrumbs"]["values"]:
         crumb.pop("timestamp", None)
 
-    assert event["breadcrumbs"]["values"] == expected_breadcrumbs
+    assert event["breadcrumbs"]["values"][0]["data"] == expected_breadcrumbs[0]["data"]
 
 
 def test_clickhouse_dbapi_spans(sentry_init, capture_events, capture_envelopes) -> None:

tests/integrations/pure_eval/test_pure_eval.py

@@ -62,6 +62,14 @@ def foo():
             "i",
             "u",
             "y",
+            "t",
+            "r",
+            "e",
+            "w",
+            "q",
+            "(q, w, e, r, t, y, u, i, o, p, a, s)",
+            "str((q, w, e, r, t, y, u, i, o, p, a, s))",
+            "events",
         ]
         assert list(frame_vars.keys()) == expected_keys
         assert frame_vars["namespace.d"] == {"1": "2"}
@@ -85,4 +93,4 @@ def foo():
             "s",
             "events",
         }
-        assert len(frame_vars) == serializer.MAX_DATABAG_BREADTH
+        assert len(frame_vars) == 14

tests/test_client.py

@@ -771,23 +771,13 @@ def test_databag_depth_stripping(sentry_init, capture_events):
     stacktrace_frame = event["exception"]["values"][0]["stacktrace"]["frames"][0]
     a_var = stacktrace_frame["vars"]["a"]
 
-    assert type(a_var) == list
-    assert len(a_var) == 1 and type(a_var[0]) == list
+    for _ in range(99):
+        assert type(a_var) == list
+        assert len(a_var) == 1
 
-    first_level_list = a_var[0]
-    assert type(first_level_list) == list
-    assert len(first_level_list) == 1
+        a_var = a_var[0]
 
-    second_level_list = first_level_list[0]
-    assert type(second_level_list) == list
-    assert len(second_level_list) == 1
-
-    third_level_list = second_level_list[0]
-    assert type(third_level_list) == list
-    assert len(third_level_list) == 1
-
-    inner_value_repr = third_level_list[0]
-    assert type(inner_value_repr) == str
+    assert type(a_var) == str
 
 
 def test_databag_string_stripping(sentry_init, capture_events):

tests/test_scrubber.py

@@ -230,7 +230,7 @@ def test_recursive_event_scrubber(sentry_init, capture_events):
     capture_event({"extra": complex_structure})
 
     (event,) = events
-    assert event["extra"]["deep"]["deeper"][0]["deepest"]["password"] == "'[Filtered]'"
+    assert event["extra"]["deep"]["deeper"][0]["deepest"]["password"] == "[Filtered]"
 
 
 def test_recursive_scrubber_does_not_override_original(sentry_init, capture_events):