fix typing

getsentry · Nov 8, 2024 · 9a8f16e · 9a8f16e
1 parent 958926d
commit 9a8f16e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/snuba/subscriptions/codecs.py b/snuba/subscriptions/codecs.py
@@ -54,8 +54,8 @@ def encode(self, value: SubscriptionTaskResult) -> KafkaPayload:
                 "request": base64.b64encode(request.SerializeToString()).decode(
                     "utf-8"
                 ),
-                "request_name": request.__name__,
-                "request_version": request.__module__.split(".", 3)[2],
+                "request_name": request.__class__.__name__,
+                "request_version": request.__class__.__module__.split(".", 3)[2],
             }
         else:
             original_body = {**request.original_body}

diff --git a/tests/subscriptions/test_codecs.py b/tests/subscriptions/test_codecs.py
@@ -274,7 +274,7 @@ def test_decode_rpc(
         build_rpc_subscription_data(entity_key=EntityKey.EAP_SPANS, metadata={}),
         EntityKey.EAP_SPANS,
         {
-            "request": "Ci0IARIJc29tZXRoaW5nGglzb21ldGhpbmciAwECAyoGCPCDuLkGMgYInIa4uQYSFCISCgcIARIDZm9vEAYaBRIDYmFyGhoIARIPCAMSC3Rlc3RfbWV0cmljGgNzdW0gASCsAg==",
+            "request": "Ci0IARIJc29tZXRoaW5nGglzb21ldGhpbmciAwECAyoGCMiIuLkGMgYI9Iq4uQYSFCISCgcIARIDZm9vEAYaBRIDYmFyGhoIARIPCAMSC3Rlc3RfbWV0cmljGgNzdW0gASCsAg==",
             "request_name": "TimeSeriesRequest",
             "request_version": "v1",
         },