fix: missing newline

katanomi · Jan 18, 2023 · d7b352c · d7b352c
1 parent 2c25279
commit d7b352c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/command/options/exit_code_option_test.go b/command/options/exit_code_option_test.go
@@ -17,6 +17,7 @@ limitations under the License.
 package options
 
 import (
+	"strings"
 	"testing"
 
 	. "github.com/onsi/gomega"
@@ -40,7 +41,7 @@ func TestExitCodeOption(t *testing.T) {
 		g.Expect(err).Should(Succeed(), "parse flag succeed.")
 		success, code, err := obj.Succeed()
 		g.Expect(err).To(Succeed())
-		g.Expect(code).To(Equal("0"))
+		g.Expect(strings.TrimRight(code, "\n")).To(Equal("0"))
 		g.Expect(success).To(Equal(true))
 	})
 
@@ -98,7 +99,7 @@ func TestExitCodeOption(t *testing.T) {
 		g.Expect(err).Should(Succeed(), "parse flag succeed.")
 		success, code, err := obj.Succeed()
 		g.Expect(err).To(Succeed())
-		g.Expect(code).To(Equal("-1"))
+		g.Expect(strings.TrimRight(code, "\n")).To(Equal("-1"))
 		g.Expect(success).To(Equal(false))
 	})
 
@@ -118,7 +119,7 @@ func TestExitCodeOption(t *testing.T) {
 		g.Expect(err).Should(Succeed(), "parse flag succeed.")
 		success, code, err := obj.Succeed()
 		g.Expect(err).To(Succeed())
-		g.Expect(code).To(Equal(""))
+		g.Expect(strings.TrimRight(code, "\n")).To(Equal(""))
 		g.Expect(success).To(Equal(true))
 	})
 }
diff --git a/command/options/testdata/exit-code-option-failed.txt b/command/options/testdata/exit-code-option-failed.txt
@@ -1 +1 @@
--1
+-1
diff --git a/command/options/testdata/exit-code-option-succeed.txt b/command/options/testdata/exit-code-option-succeed.txt
@@ -1 +1 @@
-0
+0