mirror of
https://github.com/jesseduffield/lazygit.git
synced 2025-03-19 21:28:28 +02:00
This fixes up some git and oscommand tests, and pulls some tests into commit_list_builder_test.go I've also made the NewDummyBlah functions public so that I didn't need to duplicate them across packages I've also given OSCommand a SetCommand() method for setting the command on the struct I've also created a file utils.go in the test package for creating convient 'CommandSwapper's, which basically enable you to assert a sequence of commands on the command line, and swap each one out for a different one to actually be executed
86 lines
1.8 KiB
Go
86 lines
1.8 KiB
Go
package git
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/jesseduffield/lazygit/pkg/commands"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// NewDummyPatchModifier constructs a new dummy patch modifier for testing
|
|
func NewDummyPatchModifier() *PatchModifier {
|
|
return &PatchModifier{
|
|
Log: commands.NewDummyLog(),
|
|
}
|
|
}
|
|
|
|
func TestModifyPatchForLine(t *testing.T) {
|
|
type scenario struct {
|
|
testName string
|
|
patchFilename string
|
|
lineNumber int
|
|
shouldError bool
|
|
expectedPatchFilename string
|
|
}
|
|
|
|
scenarios := []scenario{
|
|
{
|
|
"Removing one line",
|
|
"testdata/testPatchBefore.diff",
|
|
8,
|
|
false,
|
|
"testdata/testPatchAfter1.diff",
|
|
},
|
|
{
|
|
"Adding one line",
|
|
"testdata/testPatchBefore.diff",
|
|
10,
|
|
false,
|
|
"testdata/testPatchAfter2.diff",
|
|
},
|
|
{
|
|
"Adding one line in top hunk in diff with multiple hunks",
|
|
"testdata/testPatchBefore2.diff",
|
|
20,
|
|
false,
|
|
"testdata/testPatchAfter3.diff",
|
|
},
|
|
{
|
|
"Adding one line in top hunk in diff with multiple hunks",
|
|
"testdata/testPatchBefore2.diff",
|
|
53,
|
|
false,
|
|
"testdata/testPatchAfter4.diff",
|
|
},
|
|
{
|
|
"adding unstaged file with a single line",
|
|
"testdata/addedFile.diff",
|
|
6,
|
|
false,
|
|
"testdata/addedFile.diff",
|
|
},
|
|
}
|
|
|
|
for _, s := range scenarios {
|
|
t.Run(s.testName, func(t *testing.T) {
|
|
p := NewDummyPatchModifier()
|
|
beforePatch, err := ioutil.ReadFile(s.patchFilename)
|
|
if err != nil {
|
|
panic("Cannot open file at " + s.patchFilename)
|
|
}
|
|
afterPatch, err := p.ModifyPatchForLine(string(beforePatch), s.lineNumber)
|
|
if s.shouldError {
|
|
assert.Error(t, err)
|
|
} else {
|
|
assert.NoError(t, err)
|
|
expected, err := ioutil.ReadFile(s.expectedPatchFilename)
|
|
if err != nil {
|
|
panic("Cannot open file at " + s.expectedPatchFilename)
|
|
}
|
|
assert.Equal(t, string(expected), afterPatch)
|
|
}
|
|
})
|
|
}
|
|
}
|