2 files changed, 96 insertions, 56 deletions

diff --git a/crates/ide_assists/src/handlers/fill_match_arms.rs b/crates/ide_assists/src/handlers/fill_match_arms.rs
index 97435f021..3d2cd739a 100644
--- a/crates/ide_assists/src/handlers/fill_match_arms.rs
+++ b/crates/ide_assists/src/handlers/fill_match_arms.rs
@@ -31,8 +31,8 @@ use crate::{
 //
 // fn handle(action: Action) {
 //     match action {
-//         $0Action::Move { distance } => {}
-//         Action::Stop => {}
+//         $0Action::Move { distance } => todo!(),
+//         Action::Stop => todo!(),
 //     }
 // }
 // ```
@@ -129,7 +129,7 @@ pub(crate) fn fill_match_arms(acc: &mut Assists, ctx: &AssistContext) -> Option<
         |builder| {
             let new_match_arm_list = match_arm_list.clone_for_update();
             let missing_arms = missing_pats
-                .map(|pat| make::match_arm(iter::once(pat), make::expr_empty_block()))
+                .map(|pat| make::match_arm(iter::once(pat), make::ext::expr_todo()))
                 .map(|it| it.clone_for_update());
 
             let catch_all_arm = new_match_arm_list
@@ -350,8 +350,8 @@ fn foo(a: bool) {
             r#"
 fn foo(a: bool) {
     match a {
-        $0true => {}
-        false => {}
+        $0true => todo!(),
+        false => todo!(),
     }
 }
 "#,
@@ -373,7 +373,7 @@ fn foo(a: bool) {
 fn foo(a: bool) {
     match a {
         true => {}
-        $0false => {}
+        $0false => todo!(),
     }
 }
 "#,
@@ -410,10 +410,10 @@ fn foo(a: bool) {
             r#"
 fn foo(a: bool) {
     match (a, a) {
-        $0(true, true) => {}
-        (true, false) => {}
-        (false, true) => {}
-        (false, false) => {}
+        $0(true, true) => todo!(),
+        (true, false) => todo!(),
+        (false, true) => todo!(),
+        (false, false) => todo!(),
     }
 }
 "#,
@@ -435,9 +435,9 @@ fn foo(a: bool) {
 fn foo(a: bool) {
     match (a, a) {
         (false, true) => {}
-        $0(true, true) => {}
-        (true, false) => {}
-        (false, false) => {}
+        $0(true, true) => todo!(),
+        (true, false) => todo!(),
+        (false, false) => todo!(),
     }
 }
 "#,
@@ -471,7 +471,7 @@ fn main() {
     match A::As {
         A::Bs { x, y: Some(_) } => {}
         A::Cs(_, Some(_)) => {}
-        $0A::As => {}
+        $0A::As => todo!(),
     }
 }
 "#,
@@ -499,7 +499,7 @@ use Option::*;
 fn main() {
     match None {
         None => {}
-        Some(${0:_}) => {}
+        Some(${0:_}) => todo!(),
     }
 }
             "#,
@@ -523,7 +523,7 @@ enum A { As, Bs, Cs(Option<i32>) }
 fn main() {
     match A::As {
         A::Cs(_) | A::Bs => {}
-        $0A::As => {}
+        $0A::As => todo!(),
     }
 }
 "#,
@@ -553,8 +553,8 @@ fn main() {
         A::Bs if 0 < 1 => {}
         A::Ds(_value) => { let x = 1; }
         A::Es(B::Xs) => (),
-        $0A::As => {}
-        A::Cs => {}
+        $0A::As => todo!(),
+        A::Cs => todo!(),
     }
 }
 "#,
@@ -580,7 +580,7 @@ fn main() {
     match A::As {
         A::As(_) => {}
         a @ A::Bs(_) => {}
-        A::Cs(${0:_}) => {}
+        A::Cs(${0:_}) => todo!(),
     }
 }
 "#,
@@ -605,11 +605,11 @@ enum A { As, Bs, Cs(String), Ds(String, String), Es { x: usize, y: usize } }
 fn main() {
     let a = A::As;
     match a {
-        $0A::As => {}
-        A::Bs => {}
-        A::Cs(_) => {}
-        A::Ds(_, _) => {}
-        A::Es { x, y } => {}
+        $0A::As => todo!(),
+        A::Bs => todo!(),
+        A::Cs(_) => todo!(),
+        A::Ds(_, _) => todo!(),
+        A::Es { x, y } => todo!(),
     }
 }
 "#,
@@ -638,10 +638,10 @@ fn main() {
     let a = A::One;
     let b = B::One;
     match (a, b) {
-        $0(A::One, B::One) => {}
-        (A::One, B::Two) => {}
-        (A::Two, B::One) => {}
-        (A::Two, B::Two) => {}
+        $0(A::One, B::One) => todo!(),
+        (A::One, B::Two) => todo!(),
+        (A::Two, B::One) => todo!(),
+        (A::Two, B::Two) => todo!(),
     }
 }
 "#,
@@ -670,10 +670,10 @@ fn main() {
     let a = A::One;
     let b = B::One;
     match (&a, &b) {
-        $0(A::One, B::One) => {}
-        (A::One, B::Two) => {}
-        (A::Two, B::One) => {}
-        (A::Two, B::Two) => {}
+        $0(A::One, B::One) => todo!(),
+        (A::One, B::Two) => todo!(),
+        (A::Two, B::One) => todo!(),
+        (A::Two, B::Two) => todo!(),
     }
 }
 "#,
@@ -705,9 +705,9 @@ fn main() {
     let b = B::One;
     match (a, b) {
         (A::Two, B::One) => {}
-        $0(A::One, B::One) => {}
-        (A::One, B::Two) => {}
-        (A::Two, B::Two) => {}
+        $0(A::One, B::One) => todo!(),
+        (A::One, B::Two) => todo!(),
+        (A::Two, B::Two) => todo!(),
     }
 }
 "#,
@@ -736,7 +736,7 @@ fn main() {
     match (a, b) {
         (Some(_), _) => {}
         (None, Some(_)) => {}
-        $0(None, None) => {}
+        $0(None, None) => todo!(),
     }
 }
 "#,
@@ -801,8 +801,8 @@ enum A { One, Two }
 fn main() {
     let a = A::One;
     match (a, ) {
-        $0(A::One,) => {}
-        (A::Two,) => {}
+        $0(A::One,) => todo!(),
+        (A::Two,) => todo!(),
     }
 }
 "#,
@@ -826,7 +826,7 @@ enum A { As }
 
 fn foo(a: &A) {
     match a {
-        $0A::As => {}
+        $0A::As => todo!(),
     }
 }
 "#,
@@ -851,7 +851,7 @@ enum A {
 
 fn foo(a: &mut A) {
     match a {
-        $0A::Es { x, y } => {}
+        $0A::Es { x, y } => todo!(),
     }
 }
 "#,
@@ -891,8 +891,8 @@ enum E { X, Y }
 
 fn main() {
     match E::X {
-        $0E::X => {}
-        E::Y => {}
+        $0E::X => todo!(),
+        E::Y => todo!(),
     }
 }
 "#,
@@ -919,8 +919,8 @@ use foo::E::X;
 
 fn main() {
     match X {
-        $0X => {}
-        foo::E::Y => {}
+        $0X => todo!(),
+        foo::E::Y => todo!(),
     }
 }
 "#,
@@ -947,7 +947,7 @@ fn foo(a: A) {
     match a {
         // foo bar baz
         A::One => {}
-        $0A::Two => {}
+        $0A::Two => todo!(),
         // This is where the rest should be
     }
 }
@@ -971,8 +971,8 @@ fn foo(a: A) {
 enum A { One, Two }
 fn foo(a: A) {
     match a {
-        $0A::One => {}
-        A::Two => {}
+        $0A::One => todo!(),
+        A::Two => todo!(),
         // foo bar baz
     }
 }
@@ -996,8 +996,8 @@ fn foo(a: A) {
 enum A { One, Two, }
 fn foo(a: A) {
     match a {
-        $0A::One => {}
-        A::Two => {}
+        $0A::One => todo!(),
+        A::Two => todo!(),
     }
 }
 "#,
@@ -1021,8 +1021,8 @@ fn foo(opt: Option<i32>) {
             r#"
 fn foo(opt: Option<i32>) {
     match opt {
-        Some(${0:_}) => {}
-        None => {}
+        Some(${0:_}) => todo!(),
+        None => todo!(),
     }
 }
 "#,
@@ -1054,9 +1054,9 @@ enum Test {
 
 fn foo(t: Test) {
     m!(match t {
-    $0Test::A => {}
-    Test::B => {}
-    Test::C => {}
+    $0Test::A => todo!(),
+    Test::B => todo!(),
+    Test::C => todo!(),
 });
 }"#,
         );
@@ -1076,4 +1076,44 @@ fn foo(tuple: (A, A)) {
 "#,
         );
     }
+
+    #[test]
+    fn adds_comma_before_new_arms() {
+        check_assist(
+            fill_match_arms,
+            r#"
+fn foo(t: bool) {
+    match $0t {
+        true => 1 + 2
+    }
+}"#,
+            r#"
+fn foo(t: bool) {
+    match t {
+        true => 1 + 2,
+        $0false => todo!(),
+    }
+}"#,
+        );
+    }
+
+    #[test]
+    fn does_not_add_extra_comma() {
+        check_assist(
+            fill_match_arms,
+            r#"
+fn foo(t: bool) {
+    match $0t {
+        true => 1 + 2,
+    }
+}"#,
+            r#"
+fn foo(t: bool) {
+    match t {
+        true => 1 + 2,
+        $0false => todo!(),
+    }
+}"#,
+        );
+    }
 }
diff --git a/crates/ide_assists/src/tests/generated.rs b/crates/ide_assists/src/tests/generated.rs
index 8a9b0777c..de5d9e55a 100644
--- a/crates/ide_assists/src/tests/generated.rs
+++ b/crates/ide_assists/src/tests/generated.rs
@@ -455,8 +455,8 @@ enum Action { Move { distance: u32 }, Stop }
 
 fn handle(action: Action) {
     match action {
-        $0Action::Move { distance } => {}
-        Action::Stop => {}
+        $0Action::Move { distance } => todo!(),
+        Action::Stop => todo!(),
     }
 }
 "#####,