Assign Readstrategy

Signed-off-by: Muhammad Awawdi <[email protected]>
valkey-io · Feb 4, 2025 · 4a0e350 · 4a0e350
1 parent c10caf0
commit 4a0e350
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/glide-core/redis-rs/redis/src/cluster_async/connections_container.rs b/glide-core/redis-rs/redis/src/cluster_async/connections_container.rs
@@ -842,7 +842,10 @@ mod tests {
 
     #[test]
     fn get_connection_for_az_affinity_route() {
-        let container = create_container_with_az_strategy(false, None);
+        let container = create_container_with_az_strategy(
+            false,
+            Some(ReadFromReplicaStrategy::AZAffinity("use-1a".to_string())),
+        );
 
         // slot number is not exits
         assert!(container
@@ -905,7 +908,10 @@ mod tests {
 
     #[test]
     fn get_connection_for_az_affinity_route_round_robin() {
-        let container = create_container_with_az_strategy(false, None);
+        let container = create_container_with_az_strategy(
+            false,
+            Some(ReadFromReplicaStrategy::AZAffinity("use-1a".to_string())),
+        );
 
         let mut addresses = vec![
             container