Fix perf test

Author: coolreader18

crates/testing/tests/standalone_integration_test.rs

@@ -235,9 +235,10 @@ fn test_index_scans() {
                 .await
                 .unwrap();
 
-            // TODO(1011): Uncomment once multi-column prefix scans are supported
-            // let json = r#"{"call": {"fn": "test_index_scan_on_x_z", "args": []}}"#;
-            // module.send(json.to_string()).await.unwrap();
+            module
+                .call_reducer_json("test_index_scan_on_x_z", product![])
+                .await
+                .unwrap();
 
             let logs = read_logs(&module).await;
 
@@ -249,7 +250,7 @@ fn test_index_scans() {
             assert!(timing(&logs[0]));
             assert!(timing(&logs[1]));
             assert!(timing(&logs[2]));
-            // assert!(timing(&logs[3]));
+            assert!(timing(&logs[3]));
         },
     );
 }

modules/perf-test/src/lib.rs

@@ -63,12 +63,7 @@ pub fn test_index_scan_on_x_z_dimension(ctx: &ReducerContext) {
     let z = CHUNK as i32;
     let dimension = ID as u32;
     let span = LogStopwatch::new("Index scan on {x, z, dimension}");
-    let n = ctx
-        .db
-        .location()
-        .iter()
-        .filter(|r| r.x == 0 && r.z == z && r.dimension == dimension)
-        .count();
+    let n = ctx.db.location().coordinates().filter((0, z, dimension)).count();
     span.end();
     assert_eq!(n, 1);
 }
@@ -78,7 +73,7 @@ pub fn test_index_scan_on_x_z_dimension(ctx: &ReducerContext) {
 pub fn test_index_scan_on_x_z(ctx: &ReducerContext) {
     let z = CHUNK as i32;
     let span = LogStopwatch::new("Index scan on {x, z}");
-    let n = ctx.db.location().iter().filter(|r| r.x == 0 && r.z == z).count();
+    let n = ctx.db.location().coordinates().filter((0, z)).count();
     span.end();
     assert_eq!(n as u64, ROWS_PER_CHUNK);
 }