Additional tests

replicatedhq · Sep 15, 2023 · 3587249 · 3587249
1 parent 61efb30
commit 3587249
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/pkg/redact/multi_line.go b/pkg/redact/multi_line.go
@@ -53,7 +53,7 @@ func (r *MultiLineRedactor) Redact(input io.Reader, path string) io.Reader {
 		line1, line2, err := getNextTwoLines(reader, nil)
 		if err != nil {
 			// this will print 2 blank lines for empty input...
-			// Append newlines since scanner strip them
+			// Append newlines since scanner strips them
 			err = writeBytes(writer, line1, NEW_LINE, line2, NEW_LINE)
 			if err != nil {
 				return
@@ -96,7 +96,7 @@ func (r *MultiLineRedactor) Redact(input io.Reader, path string) io.Reader {
 			flushLastLine = false
 			clean := r.re2.ReplaceAll(line2, substStr)
 
-			// Append newlines since scanner strip them
+			// Append newlines since scanner strips them
 			err = writeBytes(writer, line1, NEW_LINE, clean, NEW_LINE)
 			if err != nil {
 				return

diff --git a/pkg/redact/multi_line_test.go b/pkg/redact/multi_line_test.go
@@ -128,6 +128,14 @@ func Test_writeBytes(t *testing.T) {
 			inputBytes: [][]byte{[]byte("hello"), NEW_LINE, []byte("world"), NEW_LINE},
 			want:       "hello\nworld\n",
 		},
+		{
+			name:       "Empty line",
+			inputBytes: [][]byte{NEW_LINE},
+			want:       "\n",
+		},
+		{
+			name: "Nothing",
+		},
 	}
 
 	for _, tt := range tests {