test for returning op

cozo-core/src/runtime/tests.rs

@@ -1189,13 +1189,13 @@ fn insertion() {
     assert!(db
         .run_script(
             r"?[x, y] <- [[1, 2]] :insert a {x => y}",
-            Default::default()
+            Default::default(),
         )
         .is_ok());
     assert!(db
         .run_script(
             r"?[x, y] <- [[1, 3]] :insert a {x => y}",
-            Default::default()
+            Default::default(),
         )
         .is_err());
 }
@@ -1211,7 +1211,7 @@ fn deletion() {
     assert!(db
         .run_script(
             r"?[x, y] <- [[1, 2]] :insert a {x => y}",
-            Default::default()
+            Default::default(),
         )
         .is_ok());
     db
@@ -1229,9 +1229,10 @@ fn returning() {
             Default::default(),
         )
         .unwrap();
-    for row in res.into_json()["rows"].as_array().unwrap() {
-        println!("{}", row);
-    }
+    assert_eq!(res.into_json()["rows"], json!([["OK"]]));
+    // for row in res.into_json()["rows"].as_array().unwrap() {
+    //     println!("{}", row);
+    // }
 
     let res = db
         .run_script(
@@ -1239,10 +1240,11 @@ fn returning() {
             Default::default(),
         )
         .unwrap();
-    println!("{:?}", res.headers);
-    for row in res.into_json()["rows"].as_array().unwrap() {
-        println!("{}", row);
-    }
+    assert_eq!(res.into_json()["rows"], json!([["inserted", 1, 3], ["inserted", 2, 4], ["replaced", 1, 2]]));
+    // println!("{:?}", res.headers);
+    // for row in res.into_json()["rows"].as_array().unwrap() {
+    //     println!("{}", row);
+    // }
 
     let res = db
         .run_script(
@@ -1250,10 +1252,11 @@ fn returning() {
             Default::default(),
         )
         .unwrap();
-    println!("{:?}", res.headers);
-    for row in res.into_json()["rows"].as_array().unwrap() {
-        println!("{}", row);
-    }
+    // println!("{:?}", res.headers);
+    // for row in res.into_json()["rows"].as_array().unwrap() {
+    //     println!("{}", row);
+    // }
+    assert_eq!(res.into_json()["rows"], json!([["requested", 1], ["requested", 4], ["deleted", 1]]));
 }
 
 #[test]