SNOW-897148: fix oob telemetry tags generation (#1708)

Author: sfc-gh-aling

DESCRIPTION.md

@@ -13,6 +13,7 @@ Source code is also available at: https://github.com/snowflakedb/snowflake-conne
   - Fixed a bug in retry logic for okta authentication to refresh token.
   - Support `RSAPublicKey` when constructing `AuthByKeyPair` in addition to raw bytes.
   - Fixed a bug when connecting through SOCKS5 proxy, the attribute `proxy_header` is missing on `SOCKSProxyManager`.
+  - Fixed a bug in tag generation of OOB telemetry event.
 
 - v3.1.0(July 31,2023)
 

src/snowflake/connector/telemetry_oob.py

@@ -124,13 +124,13 @@ def generate_tags(self):
 
         telemetry = TelemetryService.get_instance()
         # Add telemetry service generated tags
-        tags[TelemetryField.KEY_OOB_DRIVER] = CLIENT_NAME
-        tags[TelemetryField.KEY_OOB_VERSION] = str(SNOWFLAKE_CONNECTOR_VERSION)
+        tags[TelemetryField.KEY_OOB_DRIVER.value] = CLIENT_NAME
+        tags[TelemetryField.KEY_OOB_VERSION.value] = str(SNOWFLAKE_CONNECTOR_VERSION)
         tags[
-            TelemetryField.KEY_OOB_TELEMETRY_SERVER_DEPLOYMENT
+            TelemetryField.KEY_OOB_TELEMETRY_SERVER_DEPLOYMENT.value
         ] = telemetry.deployment.name
         tags[
-            TelemetryField.KEY_OOB_CONNECTION_STRING
+            TelemetryField.KEY_OOB_CONNECTION_STRING.value
         ] = telemetry.get_connection_string()
         if telemetry.context and len(telemetry.context) > 0:
             for k, v in telemetry.context.items():

test/unit/test_telemetry_oob.py

@@ -5,6 +5,8 @@
 
 from __future__ import annotations
 
+import logging
+
 import pytest
 
 import snowflake.connector.errorcode
@@ -49,15 +51,20 @@ def telemetry_setup(request):
     telemetry.flush()
 
 
-def test_telemetry_oob_simple_flush(telemetry_setup):
+def test_telemetry_oob_simple_flush(telemetry_setup, caplog):
     """Tests capturing and sending a simple OCSP Exception message."""
     telemetry = TelemetryService.get_instance()
 
     telemetry.log_ocsp_exception(
         event_type, telemetry_data, exception=exception, stack_trace=stack_trace
     )
     assert telemetry.size() == 1
+    caplog.set_level(logging.DEBUG, "snowflake.connector.telemetry_oob")
     telemetry.flush()
+    assert (
+        "Failed to generate a JSON dump from the passed in telemetry OOB events"
+        not in caplog.text
+    )
     assert telemetry.size() == 0