add test case for atomic pipeline (redis-rs#702)

nonirosenfeldredis · Feb 13, 2023 · 280eaed · 280eaed
1 parent f10c016
commit 280eaed
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/redis-test/src/lib.rs b/redis-test/src/lib.rs
@@ -397,4 +397,27 @@ mod tests {
             .expect("success");
         assert_eq!(results, vec!["hello", "world"]);
     }
+
+    #[test]
+    fn pipeline_atomic_test() {
+        let mut conn = MockRedisConnection::new(vec![MockCmd::with_values(
+            pipe().atomic().cmd("GET").arg("foo").cmd("GET").arg("bar"),
+            Ok(vec![Value::Bulk(
+                vec!["hello", "world"]
+                    .into_iter()
+                    .map(|x| Value::Data(x.as_bytes().into()))
+                    .collect(),
+            )]),
+        )]);
+
+        let results: Vec<String> = pipe()
+            .atomic()
+            .cmd("GET")
+            .arg("foo")
+            .cmd("GET")
+            .arg("bar")
+            .query(&mut conn)
+            .expect("success");
+        assert_eq!(results, vec!["hello", "world"]);
+    }
 }