477 Add new NeedReset property to File and update tests

Use a boolean to determin if a file needs to be reset. We want to reset
the file when discrading changes if there is a conflict.

pkg/commands/file.go

@@ -12,6 +12,7 @@ type File struct {
 	Deleted                 bool
 	HasMergeConflicts       bool
 	HasInlineMergeConflicts bool
+	NeedReset               bool
 	DisplayString           string
 	Type                    string // one of 'file', 'directory', and 'other'
 	ShortStatus             string // e.g. 'AD', ' A', 'M ', '??'

pkg/commands/git.go

@@ -176,6 +176,8 @@ func (c *GitCommand) GetStatusFiles() []*File {
 		filename := c.OSCommand.Unquote(statusString[3:])
 		_, untracked := map[string]bool{"??": true, "A ": true, "AM": true}[change]
 		_, hasNoStagedChanges := map[string]bool{" ": true, "U": true, "?": true}[stagedChange]
+		hasMergeConflicts := change == "UU" || change == "AA" || change == "DU"
+		hasInlineMergeConflicts := change == "UU" || change == "AA"
 
 		file := &File{
 			Name:                    filename,
@@ -184,8 +186,9 @@ func (c *GitCommand) GetStatusFiles() []*File {
 			HasUnstagedChanges:      unstagedChange != " ",
 			Tracked:                 !untracked,
 			Deleted:                 unstagedChange == "D" || stagedChange == "D",
-			HasMergeConflicts:       change == "UU" || change == "AA" || change == "DU",
+			HasMergeConflicts:       hasMergeConflicts,
-			HasInlineMergeConflicts: change == "UU" || change == "AA",
+			HasInlineMergeConflicts: hasInlineMergeConflicts,
+			NeedReset:               !hasNoStagedChanges || hasMergeConflicts || hasInlineMergeConflicts,
 			Type:                    c.OSCommand.FileType(filename),
 			ShortStatus:             change,
 		}
@@ -471,7 +474,7 @@ func (c *GitCommand) RebaseMode() (string, error) {
 func (c *GitCommand) DiscardAllFileChanges(file *File) error {
 	// if the file isn't tracked, we assume you want to delete it
 	quotedFileName := c.OSCommand.Quote(file.Name)
-	if file.HasStagedChanges || file.HasMergeConflicts || file.HasInlineMergeConflicts {
+	if file.NeedReset {
 		if err := c.OSCommand.RunCommand(fmt.Sprintf("git reset -- %s", quotedFileName)); err != nil {
 			return err
 		}

pkg/commands/git_test.go

@@ -356,11 +356,11 @@ func TestGitCommandGetStatusFiles(t *testing.T) {
 			func(cmd string, args ...string) *exec.Cmd {
 				return exec.Command(
 					"echo",
-					"MM file1.txt\nA  file3.txt\nAM file2.txt\n?? file4.txt",
+					"MM file1.txt\nA  file3.txt\nAM file2.txt\n?? file4.txt\nUU file5.txt",
 				)
 			},
 			func(files []*File) {
-				assert.Len(t, files, 4)
+				assert.Len(t, files, 5)
 
 				expected := []*File{
 					{
@@ -370,6 +370,8 @@ func TestGitCommandGetStatusFiles(t *testing.T) {
 						Tracked:                 true,
 						Deleted:                 false,
 						HasMergeConflicts:       false,
+						HasInlineMergeConflicts: false,
+						NeedReset:               true,
 						DisplayString:           "MM file1.txt",
 						Type:                    "other",
 						ShortStatus:             "MM",
@@ -381,6 +383,8 @@ func TestGitCommandGetStatusFiles(t *testing.T) {
 						Tracked:                 false,
 						Deleted:                 false,
 						HasMergeConflicts:       false,
+						HasInlineMergeConflicts: false,
+						NeedReset:               true,
 						DisplayString:           "A  file3.txt",
 						Type:                    "other",
 						ShortStatus:             "A ",
@@ -392,6 +396,8 @@ func TestGitCommandGetStatusFiles(t *testing.T) {
 						Tracked:                 false,
 						Deleted:                 false,
 						HasMergeConflicts:       false,
+						HasInlineMergeConflicts: false,
+						NeedReset:               true,
 						DisplayString:           "AM file2.txt",
 						Type:                    "other",
 						ShortStatus:             "AM",
@@ -403,10 +409,25 @@ func TestGitCommandGetStatusFiles(t *testing.T) {
 						Tracked:                 false,
 						Deleted:                 false,
 						HasMergeConflicts:       false,
+						HasInlineMergeConflicts: false,
+						NeedReset:               false,
 						DisplayString:           "?? file4.txt",
 						Type:                    "other",
 						ShortStatus:             "??",
 					},
+					{
+						Name:                    "file5.txt",
+						HasStagedChanges:        false,
+						HasUnstagedChanges:      true,
+						Tracked:                 true,
+						Deleted:                 false,
+						HasMergeConflicts:       true,
+						HasInlineMergeConflicts: true,
+						NeedReset:               true,
+						DisplayString:           "UU file5.txt",
+						Type:                    "other",
+						ShortStatus:             "UU",
+					},
 				}
 
 				assert.EqualValues(t, expected, files)
@@ -1196,7 +1217,7 @@ func TestGitCommandDiscardAllFileChanges(t *testing.T) {
 			},
 			&File{
 				Name:      "test",
-				HasStagedChanges: true,
+				NeedReset: true,
 			},
 			func(string) error {
 				return nil
@@ -1298,7 +1319,7 @@ func TestGitCommandDiscardAllFileChanges(t *testing.T) {
 			&File{
 				Name:      "test",
 				Tracked:   true,
-				HasStagedChanges: true,
+				NeedReset: true,
 			},
 			func(string) error {
 				return nil
@@ -1324,7 +1345,7 @@ func TestGitCommandDiscardAllFileChanges(t *testing.T) {
 			&File{
 				Name:      "test",
 				Tracked:   false,
-				HasStagedChanges: true,
+				NeedReset: true,
 			},
 			func(filename string) error {
 				assert.Equal(t, "test", filename)