Rust: Add async type inference tests

hvitved · hvitved · commit 24a4208bf2b0 · 2025-05-26T14:45:31.000+02:00
diff --git a/rust/ql/test/library-tests/type-inference/main.rs b/rust/ql/test/library-tests/type-inference/main.rs
@@ -1239,6 +1239,46 @@ mod operators {
     }
 }
 
+mod async_ {
+    use std::future::Future;
+
+    struct S1;
+
+    impl S1 {
+        pub fn f(self) {} // S1f
+    }
+
+    async fn f1() -> S1 {
+        S1
+    }
+
+    fn f2() -> impl Future<Output = S1> {
+        async {
+            S1
+        }
+    }
+
+    struct S2;
+
+    impl Future for S2 {
+        type Output = S1;
+
+        fn poll(self: std::pin::Pin<&mut Self>, _cx: &mut std::task::Context<'_>) -> std::task::Poll<Self::Output> {
+            std::task::Poll::Ready(S1)
+        }
+    }
+
+    fn f3() -> impl Future<Output = S1> {
+        S2
+    }
+
+    pub async fn f() {
+        f1().await.f(); // $ MISSING: method=S1f
+        f2().await.f(); // $ MISSING: method=S1f
+        f3().await.f(); // $ MISSING: method=S1f
+    }
+}
+
 fn main() {
     field_access::f();
     method_impl::f();
@@ -1258,4 +1298,5 @@ fn main() {
     try_expressions::f();
     builtins::f();
     operators::f();
+    async_::f();
 }
diff --git a/rust/ql/test/library-tests/type-inference/type-inference.expected b/rust/ql/test/library-tests/type-inference/type-inference.expected
@@ -1601,7 +1601,26 @@ inferType
 | main.rs:1236:13:1236:17 | ... = ... |  | file://:0:0:0:0 | () |
 | main.rs:1236:17:1236:17 | 2 |  | file:///BUILTINS/types.rs:12:1:12:15 | i32 |
 | main.rs:1238:9:1238:9 | a |  | file:///BUILTINS/types.rs:12:1:12:15 | i32 |
-| main.rs:1244:5:1244:20 | ...::f(...) |  | main.rs:67:5:67:21 | Foo |
-| main.rs:1245:5:1245:60 | ...::g(...) |  | main.rs:67:5:67:21 | Foo |
-| main.rs:1245:20:1245:38 | ...::Foo {...} |  | main.rs:67:5:67:21 | Foo |
-| main.rs:1245:41:1245:59 | ...::Foo {...} |  | main.rs:67:5:67:21 | Foo |
+| main.rs:1248:18:1248:21 | SelfParam |  | main.rs:1245:5:1245:14 | S1 |
+| main.rs:1251:25:1253:5 | { ... } |  | main.rs:1245:5:1245:14 | S1 |
+| main.rs:1252:9:1252:10 | S1 |  | main.rs:1245:5:1245:14 | S1 |
+| main.rs:1255:41:1259:5 | { ... } |  | main.rs:1245:5:1245:14 | S1 |
+| main.rs:1256:9:1258:9 | { ... } |  | main.rs:1245:5:1245:14 | S1 |
+| main.rs:1257:13:1257:14 | S1 |  | main.rs:1245:5:1245:14 | S1 |
+| main.rs:1266:17:1266:46 | SelfParam |  | file:///RUSTUP_HOME/toolchain/lib/rustlib/src/rust/library/core/src/pin.rs:934:1:1104:1 | Pin |
+| main.rs:1266:17:1266:46 | SelfParam | Ptr | file://:0:0:0:0 | & |
+| main.rs:1266:17:1266:46 | SelfParam | Ptr.&T | main.rs:1261:5:1261:14 | S2 |
+| main.rs:1266:49:1266:51 | _cx |  | file://:0:0:0:0 | & |
+| main.rs:1266:49:1266:51 | _cx | &T | file:///RUSTUP_HOME/toolchain/lib/rustlib/src/rust/library/core/src/task/wake.rs:214:1:232:1 | Context |
+| main.rs:1266:116:1268:9 | { ... } |  | file:///RUSTUP_HOME/toolchain/lib/rustlib/src/rust/library/core/src/task/poll.rs:6:1:28:1 | Poll |
+| main.rs:1266:116:1268:9 | { ... } | T | main.rs:1245:5:1245:14 | S1 |
+| main.rs:1267:13:1267:38 | ...::Ready(...) |  | file:///RUSTUP_HOME/toolchain/lib/rustlib/src/rust/library/core/src/task/poll.rs:6:1:28:1 | Poll |
+| main.rs:1267:13:1267:38 | ...::Ready(...) | T | main.rs:1245:5:1245:14 | S1 |
+| main.rs:1267:36:1267:37 | S1 |  | main.rs:1245:5:1245:14 | S1 |
+| main.rs:1271:41:1273:5 | { ... } |  | main.rs:1261:5:1261:14 | S2 |
+| main.rs:1272:9:1272:10 | S2 |  | main.rs:1261:5:1261:14 | S2 |
+| main.rs:1276:9:1276:12 | f1(...) |  | main.rs:1245:5:1245:14 | S1 |
+| main.rs:1284:5:1284:20 | ...::f(...) |  | main.rs:67:5:67:21 | Foo |
+| main.rs:1285:5:1285:60 | ...::g(...) |  | main.rs:67:5:67:21 | Foo |
+| main.rs:1285:20:1285:38 | ...::Foo {...} |  | main.rs:67:5:67:21 | Foo |
+| main.rs:1285:41:1285:59 | ...::Foo {...} |  | main.rs:67:5:67:21 | Foo |
