diff --git a/pkg/integration/tests/conflicts/resolve_without_trailing_lf.go b/pkg/integration/tests/conflicts/resolve_without_trailing_lf.go new file mode 100644 index 00000000000..2af1eac5445 --- /dev/null +++ b/pkg/integration/tests/conflicts/resolve_without_trailing_lf.go @@ -0,0 +1,57 @@ +package conflicts + +import ( + "github.com/jesseduffield/lazygit/pkg/config" + . "github.com/jesseduffield/lazygit/pkg/integration/components" +) + +var ResolveWithoutTrailingLf = NewIntegrationTest(NewIntegrationTestArgs{ + Description: "Regression test for resolving a merge conflict when the file doesn't have a trailing newline", + ExtraCmdArgs: []string{}, + Skip: false, + SetupConfig: func(config *config.AppConfig) {}, + SetupRepo: func(shell *Shell) { + shell. + NewBranch("branch1"). + CreateFileAndAdd("file", "a\n\nno eol"). + Commit("initial commit"). + UpdateFileAndAdd("file", "a1\n\nno eol"). + Commit("commit on branch1"). + NewBranchFrom("branch2", "HEAD^"). + UpdateFileAndAdd("file", "a2\n\nno eol"). + Commit("commit on branch2"). + Checkout("branch1"). + RunCommandExpectError([]string{"git", "merge", "--no-edit", "branch2"}) + }, + Run: func(t *TestDriver, keys config.KeybindingConfig) { + t.Views().Files(). + IsFocused(). + Lines( + Contains("UU file").IsSelected(), + ). + PressEnter() + + t.Views().MergeConflicts(). + IsFocused(). + SelectedLines( + Contains("<<<<<<< HEAD"), + Contains("a1"), + Contains("======="), + ). + SelectNextItem(). + PressPrimaryAction() + + t.ExpectPopup().Alert(). + Title(Equals("Continue")). + Content(Contains("All merge conflicts resolved. Continue?")). + Cancel() + + t.Views().Files(). + Focus(). + Lines( + Contains("M file").IsSelected(), + ) + + t.Views().Main().Content(Contains("-a1\n+a2\n").DoesNotContain("-no eol")) + }, +}) diff --git a/pkg/integration/tests/test_list.go b/pkg/integration/tests/test_list.go index 6c6d484cf75..0c0142e40a9 100644 --- a/pkg/integration/tests/test_list.go +++ b/pkg/integration/tests/test_list.go @@ -123,6 +123,7 @@ var tests = []*components.IntegrationTest{ conflicts.ResolveExternally, conflicts.ResolveMultipleFiles, conflicts.ResolveNoAutoStage, + conflicts.ResolveWithoutTrailingLf, conflicts.UndoChooseHunk, custom_commands.AccessCommitProperties, custom_commands.BasicCommand, diff --git a/pkg/utils/io.go b/pkg/utils/io.go index d31c5fc073e..1d222746b95 100644 --- a/pkg/utils/io.go +++ b/pkg/utils/io.go @@ -2,6 +2,7 @@ package utils import ( "bufio" + "io" "os" ) @@ -12,14 +13,18 @@ func ForEachLineInFile(path string, f func(string, int)) error { } defer file.Close() - reader := bufio.NewReader(file) + forEachLineInStream(file, f) + + return nil +} + +func forEachLineInStream(reader io.Reader, f func(string, int)) { + bufferedReader := bufio.NewReader(reader) for i := 0; true; i++ { - line, err := reader.ReadString('\n') - if err != nil { + line, _ := bufferedReader.ReadString('\n') + if len(line) == 0 { break } f(line, i) } - - return nil } diff --git a/pkg/utils/io_test.go b/pkg/utils/io_test.go new file mode 100644 index 00000000000..b8dfa957c43 --- /dev/null +++ b/pkg/utils/io_test.go @@ -0,0 +1,57 @@ +package utils + +import ( + "strings" + "testing" + + "github.com/stretchr/testify/assert" +) + +func Test_forEachLineInStream(t *testing.T) { + scenarios := []struct { + name string + input string + expectedLines []string + }{ + { + name: "empty input", + input: "", + expectedLines: []string{}, + }, + { + name: "single line", + input: "abc\n", + expectedLines: []string{"abc\n"}, + }, + { + name: "single line without line feed", + input: "abc", + expectedLines: []string{"abc"}, + }, + { + name: "multiple lines", + input: "abc\ndef\n", + expectedLines: []string{"abc\n", "def\n"}, + }, + { + name: "multiple lines including empty lines", + input: "abc\n\ndef\n", + expectedLines: []string{"abc\n", "\n", "def\n"}, + }, + { + name: "multiple lines without linefeed at end of file", + input: "abc\ndef\nghi", + expectedLines: []string{"abc\n", "def\n", "ghi"}, + }, + } + + for _, s := range scenarios { + t.Run(s.name, func(t *testing.T) { + lines := []string{} + forEachLineInStream(strings.NewReader(s.input), func(line string, i int) { + lines = append(lines, line) + }) + assert.EqualValues(t, s.expectedLines, lines) + }) + } +}