Fill missing fields of enum variants

1 files changed, 29 insertions, 0 deletions

diff --git a/crates/ra_ide/src/diagnostics.rs b/crates/ra_ide/src/diagnostics.rs
index 82596c665..9cf86b26d 100644
--- a/crates/ra_ide/src/diagnostics.rs
+++ b/crates/ra_ide/src/diagnostics.rs
@@ -470,6 +470,35 @@ mod tests {
     }
 
     #[test]
+    fn test_fill_struct_fields_enum() {
+        let before = r"
+            enum Expr {
+                Bin { lhs: Box<Expr>, rhs: Box<Expr> }
+            }
+
+            impl Expr {
+                fn new_bin(lhs: Box<Expr>, rhs: Box<Expr>) -> Expr {
+                    Expr::Bin { <|> }
+                }
+            }
+
+        ";
+        let after = r"
+            enum Expr {
+                Bin { lhs: Box<Expr>, rhs: Box<Expr> }
+            }
+
+            impl Expr {
+                fn new_bin(lhs: Box<Expr>, rhs: Box<Expr>) -> Expr {
+                    Expr::Bin { lhs: (), rhs: () <|> }
+                }
+            }
+
+        ";
+        check_apply_diagnostic_fix(before, after);
+    }
+
+    #[test]
     fn test_fill_struct_fields_partial() {
         let before = r"
             struct TestStruct {