fix: add test case for BugFixSuggestion

nedpals · Feb 3, 2024 · c7570f5 · c7570f5
1 parent 240d01c
commit c7570f5
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/translation_test.go b/translation_test.go
@@ -262,6 +262,20 @@ func TestBugFixGenerator(t *testing.T) {
 			})
 		})
 
+		t.Run("With punctuation", func(t *testing.T) {
+			gen := &lib.BugFixGenerator{
+				Document: doc,
+			}
+
+			gen.Add("This is a simple error message.", func(s *lib.BugFixSuggestion) {
+				s.AddStep("Oh wow!", func(step *lib.BugFixStep) {
+					if step.Content != "Oh wow!" {
+						t.Errorf("Expected 'Oh wow!', got %s", step.Content)
+					}
+				})
+			})
+		})
+
 		t.Run("With string data", func(t *testing.T) {
 			gen := &lib.BugFixGenerator{
 				Document: doc,