[#66] fix ReplacePathFileProc.Process mapping entity.DataFile.Data

kozmod · Feb 18, 2023 · ad4ab02 · ad4ab02
1 parent cc08091
commit ad4ab02
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/internal/exec/file.go b/internal/exec/file.go
@@ -110,7 +110,10 @@ func NewReplacePathFileProc(paths map[string]string) *ReplacePathFileProc {
 func (p *ReplacePathFileProc) Process(file entity.DataFile) (entity.DataFile, error) {
 	newPath, ok := p.paths[file.Path()]
 	if ok {
-		file = entity.DataFile{FileInfo: entity.NewFileInfo(newPath)}
+		file = entity.DataFile{
+			FileInfo: entity.NewFileInfo(newPath),
+			Data:     file.Data,
+		}
 	}
 	return file, nil
 }

diff --git a/internal/exec/file_test.go b/internal/exec/file_test.go
@@ -228,6 +228,7 @@ func Test_ReplacePathFileProc(t *testing.T) {
 		oldPathA = "old_1/old_A.go"
 		oldPathB = "old_1/old_B.go"
 		newPathA = "old_1/new_A.go"
+		someData = "DATA_!!!"
 	)
 
 	var (
@@ -238,15 +239,17 @@ func Test_ReplacePathFileProc(t *testing.T) {
 
 	t.Run("success_replace_old_path_to_new_one", func(t *testing.T) {
 		proc := ReplacePathFileProc{paths: paths}
-		res, err := proc.Process(entity.DataFile{Data: nil, FileInfo: entity.NewFileInfo(oldPathA)})
+		res, err := proc.Process(entity.DataFile{Data: []byte(someData), FileInfo: entity.NewFileInfo(oldPathA)})
 		assert.NoError(t, err)
 		assert.Equal(t, newPathA, res.Path())
+		assert.Equal(t, []byte(someData), res.Data)
 	})
 	t.Run("success_nit_replace_old_path_when_new_one_not_present", func(t *testing.T) {
 		proc := ReplacePathFileProc{paths: paths}
-		res, err := proc.Process(entity.DataFile{Data: nil, FileInfo: entity.NewFileInfo(oldPathB)})
+		res, err := proc.Process(entity.DataFile{Data: []byte(someData), FileInfo: entity.NewFileInfo(oldPathB)})
 		assert.NoError(t, err)
 		assert.Equal(t, oldPathB, res.Path())
+		assert.Equal(t, []byte(someData), res.Data)
 	})
 }