Add DemandReplica two replicas test

StackExchange · Sep 15, 2023 · f3e325a · f3e325a
1 parent 106cd2d
commit f3e325a
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/StackExchange.Redis.Tests/SentinelTwoReplicasTest.cs b/tests/StackExchange.Redis.Tests/SentinelTwoReplicasTest.cs
@@ -58,8 +58,10 @@ public async Task PrimaryConnectTest()
         Assert.Equal(expected, value);
     }
 
-    [Fact]
-    public async Task ReplicasRoundRobinServerSelectionTest()
+    [Theory]
+    [InlineData(CommandFlags.DemandReplica)]
+    [InlineData(CommandFlags.PreferReplica)]
+    public async Task ReplicasRoundRobinServerSelectionTest(CommandFlags replicaFlags)
     {
         var connectionString = $"{TestConfig.Current.SentinelServer}:{TestConfig.Current.SentinelTwoReplicasPortA},serviceName={ServiceOptions.ServiceName},allowAdmin=true";
         using var conn = await ConnectionMultiplexer.ConnectAsync(connectionString);
@@ -98,7 +100,7 @@ public async Task ReplicasRoundRobinServerSelectionTest()
         conn.RegisterProfiler(() => profilingSession);
         for (var i = 0; i < 100; i++)
         {
-            await db.StringGetAsync(key, CommandFlags.DemandReplica);
+            await db.StringGetAsync(key, replicaFlags);
         }
 
         var commands = profilingSession.FinishProfiling();