Rust: Add async block lambda tests

Author: paldepind

rust/ql/test/library-tests/dataflow/lambdas/Cargo.lock

@@ -113,6 +113,61 @@ fn test_external_call() {
     may_invoke_callback4(|x| sink(a)); // $ hasValueFlow=81
 }
 
+mod async_blocks {
+    use super::{sink, source};
+
+    async fn async_block_flow_out() {
+        let future = async { source(206) };
+        sink(future.await); // $ MISSING: hasValueFlow=206
+    }
+
+    async fn async_block_flow_in() {
+        let data = source(207);
+        async {
+            sink(data); // $ MISSING: hasValueFlow=207
+        }
+        .await;
+    }
+
+    async fn async_block_flow_through() {
+        let data = source(208);
+        let future = async { data };
+        sink(future.await); // $ MISSING: hasValueFlow=208
+    }
+
+    async fn async_block_captured_variable() {
+        let mut captured = 1;
+        sink(captured);
+        async {
+            captured = source(209);
+        }
+        .await;
+        sink(captured); // $ MISSING: hasValueFlow=209
+    }
+
+    async fn boxed_async_block_captured_variable() {
+        let mut captured = 1;
+        sink(captured);
+        Box::pin(async {
+            captured = source(210);
+        })
+        .await;
+        sink(captured); // $ MISSING: hasValueFlow=210
+    }
+
+    async fn test_async_blocks() {
+        async_block_flow_out().await;
+        async_block_flow_in().await;
+        async_block_flow_through().await;
+        async_block_captured_variable().await;
+        boxed_async_block_captured_variable().await;
+    }
+
+    pub fn main() {
+        futures::executor::block_on(test_async_blocks());
+    }
+}
+
 fn main() {
     closure_flow_out();
     closure_flow_in();
@@ -123,4 +178,5 @@ fn main() {
     function_flows_through();
     test_apply();
     test_apply_wrap();
+    async_blocks::main();
 }

rust/ql/test/library-tests/dataflow/lambdas/main.rs

@@ -0,0 +1,3 @@
+qltest_cargo_check: true
+qltest_dependencies:
+    - futures = { version = "0.3" }