fix float-to-int deserialization in Redis instrumentation for REPLAY mode

Author: sohankshirsagar

drift/instrumentation/redis/e2e-tests/src/app.py

@@ -319,6 +319,25 @@ def test_cluster_pipeline():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
+@app.route("/test/cluster-pipeline-transaction", methods=["GET"])
+def test_cluster_pipeline_transaction():
+    """Test ClusterPipeline with transaction mode.
+
+    Uses TransactionStrategy internally. All keys must be on the same slot.
+    """
+    try:
+        cluster = get_cluster_client()
+        pipe = cluster.pipeline(transaction=True)
+        pipe.set("{tx}:key1", "txval1")
+        pipe.set("{tx}:key2", "txval2")
+        pipe.get("{tx}:key1")
+        pipe.get("{tx}:key2")
+        results = pipe.execute()
+        cluster.delete("{tx}:key1", "{tx}:key2")
+        return jsonify({"success": True, "results": results})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
 
 if __name__ == "__main__":
     sdk.mark_app_as_ready()

drift/instrumentation/redis/e2e-tests/src/test_requests.py

@@ -49,4 +49,7 @@
     make_request("GET", "/test/cluster-incr")
     make_request("GET", "/test/cluster-pipeline")
 
+    # RedisCluster
+    make_request("GET", "/test/cluster-pipeline-transaction")
+
     print_request_summary()

drift/instrumentation/redis/instrumentation.py

@@ -1122,6 +1122,8 @@ def _deserialize_value(self, value: Any) -> Any:
             return {k: self._deserialize_value(v) for k, v in value.items()}
         elif isinstance(value, list):
             return [self._deserialize_value(v) for v in value]
+        elif isinstance(value, float) and value.is_integer():
+            return int(value)
         return value
 
     def _deserialize_response(self, mock_data: dict[str, Any]) -> Any: