fix wildcard type annotation

verus-lang · Nov 10, 2023 · ad3654f · ad3654f
1 parent 73910ea
commit ad3654f
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/src/lib.rs b/src/lib.rs
@@ -788,7 +788,7 @@ fn to_doc<'a>(
                 .brackets()
         }
         Rule::slice_type => map_to_doc(ctx, arena, pair).brackets(),
-        Rule::infer_type => map_to_doc(ctx, arena, pair),
+        Rule::infer_type => arena.text("_").append(map_to_doc(ctx, arena, pair)),
         Rule::fn_ptr_type => map_to_doc(ctx, arena, pair),
         Rule::fn_trait_type => map_to_doc(ctx, arena, pair),
         Rule::for_type => map_to_doc(ctx, arena, pair),

diff --git a/tests/rustfmt-tests.rs b/tests/rustfmt-tests.rs
@@ -354,3 +354,20 @@ use LongLongLongLongLongLongLongLongLongLongType as LongerLongLongLongLongLongLo
 "#;
     compare(file);
 }
+
+#[test]
+fn rust_wildcard_type_annotation() {
+    let file = r#"
+fn foo() {
+    let x: _ = bar();
+}
+
+fn blah() {
+let (temp_owl__x607, Tracked(itree)): ( _
+, Tracked<ITreeToken<(Seq<u8>, state_Initiator), Endpoint>> ) = { baz(); };
+}
+"#;
+    compare(file);
+
+}
+