From 8772208cb9c7acd46f8d81702cd83683c7906658 Mon Sep 17 00:00:00 2001
From: Kiran Kumar Kolli <kirankk@microsoft.com>
Date: Fri, 4 Oct 2024 17:26:46 -0700
Subject: [PATCH] [INTERNAL] CI : Fixes Quarantine
 StreamOperationsTestDirectModeAsync (#4750)

[INTERNAL] CI : Fixes Quarantine StreamOperationsTestDirectModeAsync
---
 .../ClientTelemetryTests.cs                               | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.EmulatorTests/ClientTelemetryTests.cs b/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.EmulatorTests/ClientTelemetryTests.cs
index ffa3e6eda1..1375c1ae66 100644
--- a/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.EmulatorTests/ClientTelemetryTests.cs
+++ b/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.EmulatorTests/ClientTelemetryTests.cs
@@ -102,13 +102,19 @@ public override async Task StreamReadFailureOperationsTest(ConnectionMode mode)
         }
 
         [TestMethod]
-        [DataRow(ConnectionMode.Direct)]
         [DataRow(ConnectionMode.Gateway)]
         public override async Task StreamOperationsTest(ConnectionMode mode)
         {
             await base.StreamOperationsTest(mode);
         }
 
+        [TestMethod]
+        [TestCategory("Quarantine")] // Flaky test
+        public async Task StreamOperationsTestDirectModeAsync()
+        {
+            await base.StreamOperationsTest(ConnectionMode.Direct);
+        }
+
         [TestMethod]
         [DataRow(ConnectionMode.Direct)]
         [DataRow(ConnectionMode.Gateway)]