package git_commands import ( "testing" "github.com/jesseduffield/lazygit/pkg/commands/models" "github.com/jesseduffield/lazygit/pkg/commands/oscommands" "github.com/jesseduffield/lazygit/pkg/utils" "github.com/stretchr/testify/assert" ) func TestFileGetStatusFiles(t *testing.T) { type scenario struct { testName string runner oscommands.ICmdObjRunner expectedFiles []*models.File } scenarios := []scenario{ { "No files found", oscommands.NewFakeRunner(t). Expect(`git status --untracked-files=yes --porcelain -z`, "", nil), []*models.File{}, }, { "Several files found", oscommands.NewFakeRunner(t). Expect( `git status --untracked-files=yes --porcelain -z`, "MM file1.txt\x00A file3.txt\x00AM file2.txt\x00?? file4.txt\x00UU file5.txt", nil, ), []*models.File{ { Name: "file1.txt", HasStagedChanges: true, HasUnstagedChanges: true, Tracked: true, Added: false, Deleted: false, HasMergeConflicts: false, HasInlineMergeConflicts: false, DisplayString: "MM file1.txt", Type: "file", ShortStatus: "MM", }, { Name: "file3.txt", HasStagedChanges: true, HasUnstagedChanges: false, Tracked: false, Added: true, Deleted: false, HasMergeConflicts: false, HasInlineMergeConflicts: false, DisplayString: "A file3.txt", Type: "file", ShortStatus: "A ", }, { Name: "file2.txt", HasStagedChanges: true, HasUnstagedChanges: true, Tracked: false, Added: true, Deleted: false, HasMergeConflicts: false, HasInlineMergeConflicts: false, DisplayString: "AM file2.txt", Type: "file", ShortStatus: "AM", }, { Name: "file4.txt", HasStagedChanges: false, HasUnstagedChanges: true, Tracked: false, Added: true, Deleted: false, HasMergeConflicts: false, HasInlineMergeConflicts: false, DisplayString: "?? file4.txt", Type: "file", ShortStatus: "??", }, { Name: "file5.txt", HasStagedChanges: false, HasUnstagedChanges: true, Tracked: true, Added: false, Deleted: false, HasMergeConflicts: true, HasInlineMergeConflicts: true, DisplayString: "UU file5.txt", Type: "file", ShortStatus: "UU", }, }, }, { "File with new line char", oscommands.NewFakeRunner(t). Expect(`git status --untracked-files=yes --porcelain -z`, "MM a\nb.txt", nil), []*models.File{ { Name: "a\nb.txt", HasStagedChanges: true, HasUnstagedChanges: true, Tracked: true, Added: false, Deleted: false, HasMergeConflicts: false, HasInlineMergeConflicts: false, DisplayString: "MM a\nb.txt", Type: "file", ShortStatus: "MM", }, }, }, { "Renamed files", oscommands.NewFakeRunner(t). Expect( `git status --untracked-files=yes --porcelain -z`, "R after1.txt\x00before1.txt\x00RM after2.txt\x00before2.txt", nil, ), []*models.File{ { Name: "after1.txt", PreviousName: "before1.txt", HasStagedChanges: true, HasUnstagedChanges: false, Tracked: true, Added: false, Deleted: false, HasMergeConflicts: false, HasInlineMergeConflicts: false, DisplayString: "R before1.txt -> after1.txt", Type: "file", ShortStatus: "R ", }, { Name: "after2.txt", PreviousName: "before2.txt", HasStagedChanges: true, HasUnstagedChanges: true, Tracked: true, Added: false, Deleted: false, HasMergeConflicts: false, HasInlineMergeConflicts: false, DisplayString: "RM before2.txt -> after2.txt", Type: "file", ShortStatus: "RM", }, }, }, { "File with arrow in name", oscommands.NewFakeRunner(t). Expect( `git status --untracked-files=yes --porcelain -z`, `?? a -> b.txt`, nil, ), []*models.File{ { Name: "a -> b.txt", HasStagedChanges: false, HasUnstagedChanges: true, Tracked: false, Added: true, Deleted: false, HasMergeConflicts: false, HasInlineMergeConflicts: false, DisplayString: "?? a -> b.txt", Type: "file", ShortStatus: "??", }, }, }, } for _, s := range scenarios { s := s t.Run(s.testName, func(t *testing.T) { cmd := oscommands.NewDummyCmdObjBuilder(s.runner) loader := &FileLoader{ Common: utils.NewDummyCommon(), cmd: cmd, config: &FakeFileLoaderConfig{showUntrackedFiles: "yes"}, getFileType: func(string) string { return "file" }, } assert.EqualValues(t, s.expectedFiles, loader.GetStatusFiles(GetStatusFileOptions{})) }) } } type FakeFileLoaderConfig struct { showUntrackedFiles string } func (self *FakeFileLoaderConfig) GetShowUntrackedFiles() string { return self.showUntrackedFiles }