mirror of
https://github.com/go-gitea/gitea.git
synced 2024-08-18 10:33:24 +02:00
Fixes #16837 if a column is deleted. We were clobbering the columns that were added by looping through the aline (base) and then when bline (head) was looped through, it clobbered what was in the "cells" array that is show in the diff, and then left a nil cell because nothing was shifted. This fix properly shifts the cells, and properly puts the b cell either at its location or after, according to what the aline placed in the cells. This includes test, adding a new test function since adding/removing cells works best with three columns, not two, which results in 4 columns of the resulting cells because it has a deleted column and an added column. If you try this locally, you can try those cases and others, such as adding a column. There was no need to do anything special for the rows when `aline == 0 || bline == 0` so that was removed. This allows the same code to be used for removed or added lines, with the bcell text always being the RightCell, acell text being the LeftCell. I still added the patch zeripath gave at https://github.com/go-gitea/gitea/issues/16837#issuecomment-913007382 so that just in case for some reason a cell is nil (which shouldn't happen now) it doesn't throw a 500 error, so the user can at least view the raw diff. Also fixes in the [view.go](https://github.com/go-gitea/gitea/pull/17018/files#diff-43a7f4747c7ba8bff888c9be11affaafd595fd55d27f3333840eb19df9fad393L521) file how if a CSV file is empty (either created empty or if you edit it and remove all contents) it throws a huge 500 error when you then save it (when you view the file). Since we allow creating, saving and pushing empty files, we shouldn't throw an error on an empty CSV file, but just show its empty contents. This doesn't happen if it is a Markdown file or other type of file that is empty. EDIT: Now handled in the markup/csv renderer code
225 lines
5.5 KiB
Go
225 lines
5.5 KiB
Go
// Copyright 2021 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package gitdiff
|
|
|
|
import (
|
|
"encoding/csv"
|
|
"strings"
|
|
"testing"
|
|
|
|
csv_module "code.gitea.io/gitea/modules/csv"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestCSVDiff(t *testing.T) {
|
|
var cases = []struct {
|
|
diff string
|
|
base string
|
|
head string
|
|
cells [][]TableDiffCellType
|
|
}{
|
|
// case 0 - initial commit of a csv
|
|
{
|
|
diff: `diff --git a/unittest.csv b/unittest.csv
|
|
--- a/unittest.csv
|
|
+++ b/unittest.csv
|
|
@@ -0,0 +1,2 @@
|
|
+col1,col2
|
|
+a,a`,
|
|
base: "",
|
|
head: `col1,col2
|
|
a,a`,
|
|
cells: [][]TableDiffCellType{
|
|
{TableDiffCellAdd, TableDiffCellAdd},
|
|
{TableDiffCellAdd, TableDiffCellAdd}},
|
|
},
|
|
// case 1 - adding 1 row at end
|
|
{
|
|
diff: `diff --git a/unittest.csv b/unittest.csv
|
|
--- a/unittest.csv
|
|
+++ b/unittest.csv
|
|
@@ -1,2 +1,3 @@
|
|
col1,col2
|
|
-a,a
|
|
+a,a
|
|
+b,b`,
|
|
base: `col1,col2
|
|
a,a`,
|
|
head: `col1,col2
|
|
a,a
|
|
b,b`,
|
|
cells: [][]TableDiffCellType{
|
|
{TableDiffCellUnchanged, TableDiffCellUnchanged}, {TableDiffCellUnchanged, TableDiffCellUnchanged},
|
|
{TableDiffCellAdd, TableDiffCellAdd},
|
|
},
|
|
},
|
|
// case 2 - row deleted
|
|
{
|
|
diff: `diff --git a/unittest.csv b/unittest.csv
|
|
--- a/unittest.csv
|
|
+++ b/unittest.csv
|
|
@@ -1,3 +1,2 @@
|
|
col1,col2
|
|
-a,a
|
|
b,b`,
|
|
base: `col1,col2
|
|
a,a
|
|
b,b`,
|
|
head: `col1,col2
|
|
b,b`,
|
|
cells: [][]TableDiffCellType{
|
|
{TableDiffCellUnchanged, TableDiffCellUnchanged}, {TableDiffCellDel, TableDiffCellDel},
|
|
{TableDiffCellUnchanged, TableDiffCellUnchanged},
|
|
},
|
|
},
|
|
// case 3 - row changed
|
|
{
|
|
diff: `diff --git a/unittest.csv b/unittest.csv
|
|
--- a/unittest.csv
|
|
+++ b/unittest.csv
|
|
@@ -1,2 +1,2 @@
|
|
col1,col2
|
|
-b,b
|
|
+b,c`,
|
|
base: `col1,col2
|
|
b,b`,
|
|
head: `col1,col2
|
|
b,c`,
|
|
cells: [][]TableDiffCellType{
|
|
{TableDiffCellUnchanged, TableDiffCellUnchanged},
|
|
{TableDiffCellUnchanged, TableDiffCellChanged},
|
|
},
|
|
},
|
|
// case 4 - all deleted
|
|
{
|
|
diff: `diff --git a/unittest.csv b/unittest.csv
|
|
--- a/unittest.csv
|
|
+++ b/unittest.csv
|
|
@@ -1,2 +0,0 @@
|
|
-col1,col2
|
|
-b,c`,
|
|
base: `col1,col2
|
|
b,c`,
|
|
head: "",
|
|
cells: [][]TableDiffCellType{
|
|
{TableDiffCellDel, TableDiffCellDel},
|
|
{TableDiffCellDel, TableDiffCellDel},
|
|
},
|
|
},
|
|
// case 5 - renames first column
|
|
{
|
|
diff: `diff --git a/unittest.csv b/unittest.csv
|
|
--- a/unittest.csv
|
|
+++ b/unittest.csv
|
|
@@ -1,3 +1,3 @@
|
|
-col1,col2,col3
|
|
+cola,col2,col3
|
|
a,b,c`,
|
|
base: `col1,col2,col3
|
|
a,b,c`,
|
|
head: `cola,col2,col3
|
|
a,b,c`,
|
|
cells: [][]TableDiffCellType{
|
|
{TableDiffCellDel, TableDiffCellAdd, TableDiffCellUnchanged, TableDiffCellUnchanged},
|
|
{TableDiffCellDel, TableDiffCellAdd, TableDiffCellUnchanged, TableDiffCellUnchanged},
|
|
},
|
|
},
|
|
// case 6 - inserts a column after first, deletes last column
|
|
{
|
|
diff: `diff --git a/unittest.csv b/unittest.csv
|
|
--- a/unittest.csv
|
|
+++ b/unittest.csv
|
|
@@ -1,2 +1,2 @@
|
|
-col1,col2,col3
|
|
-a,b,c
|
|
+col1,col1a,col2
|
|
+a,d,b`,
|
|
base: `col1,col2,col3
|
|
a,b,c`,
|
|
head: `col1,col1a,col2
|
|
a,d,b`,
|
|
cells: [][]TableDiffCellType{
|
|
{TableDiffCellUnchanged, TableDiffCellAdd, TableDiffCellDel, TableDiffCellMovedUnchanged},
|
|
{TableDiffCellUnchanged, TableDiffCellAdd, TableDiffCellDel, TableDiffCellMovedUnchanged},
|
|
},
|
|
},
|
|
// case 7 - deletes first column, inserts column after last
|
|
{
|
|
diff: `diff --git a/unittest.csv b/unittest.csv
|
|
--- a/unittest.csv
|
|
+++ b/unittest.csv
|
|
@@ -1,2 +1,2 @@
|
|
-col1,col2,col3
|
|
-a,b,c
|
|
+col2,col3,col4
|
|
+b,c,d`,
|
|
base: `col1,col2,col3
|
|
a,b,c`,
|
|
head: `col2,col3,col4
|
|
b,c,d`,
|
|
cells: [][]TableDiffCellType{
|
|
{TableDiffCellDel, TableDiffCellUnchanged, TableDiffCellUnchanged, TableDiffCellAdd},
|
|
{TableDiffCellDel, TableDiffCellUnchanged, TableDiffCellUnchanged, TableDiffCellAdd},
|
|
},
|
|
},
|
|
// case 8 - two columns deleted, 2 added
|
|
{
|
|
diff: `diff --git a/unittest.csv b/unittest.csv
|
|
--- a/unittest.csv
|
|
+++ b/unittest.csv
|
|
@@ -1,2 +1,2 @@
|
|
-col1,col2,col
|
|
-a,b,c
|
|
+col3,col4,col5
|
|
+c,d,e`,
|
|
base: `col1,col2,col3
|
|
a,b,c`,
|
|
head: `col3,col4,col5
|
|
c,d,e`,
|
|
cells: [][]TableDiffCellType{
|
|
{TableDiffCellDel, TableDiffCellMovedUnchanged, TableDiffCellDel, TableDiffCellAdd, TableDiffCellAdd},
|
|
{TableDiffCellDel, TableDiffCellMovedUnchanged, TableDiffCellDel, TableDiffCellAdd, TableDiffCellAdd},
|
|
},
|
|
},
|
|
}
|
|
|
|
for n, c := range cases {
|
|
diff, err := ParsePatch(setting.Git.MaxGitDiffLines, setting.Git.MaxGitDiffLineCharacters, setting.Git.MaxGitDiffFiles, strings.NewReader(c.diff))
|
|
if err != nil {
|
|
t.Errorf("ParsePatch failed: %s", err)
|
|
}
|
|
|
|
var baseReader *csv.Reader
|
|
if len(c.base) > 0 {
|
|
baseReader, err = csv_module.CreateReaderAndGuessDelimiter(strings.NewReader(c.base))
|
|
if err != nil {
|
|
t.Errorf("CreateReaderAndGuessDelimiter failed: %s", err)
|
|
}
|
|
}
|
|
var headReader *csv.Reader
|
|
if len(c.head) > 0 {
|
|
headReader, err = csv_module.CreateReaderAndGuessDelimiter(strings.NewReader(c.head))
|
|
if err != nil {
|
|
t.Errorf("CreateReaderAndGuessDelimiter failed: %s", err)
|
|
}
|
|
}
|
|
|
|
result, err := CreateCsvDiff(diff.Files[0], baseReader, headReader)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, result, 1, "case %d: should be one section", n)
|
|
|
|
section := result[0]
|
|
assert.Len(t, section.Rows, len(c.cells), "case %d: should be %d rows", n, len(c.cells))
|
|
|
|
for i, row := range section.Rows {
|
|
assert.Len(t, row.Cells, len(c.cells[i]), "case %d: row %d should have %d cells", n, i, len(c.cells[i]))
|
|
for j, cell := range row.Cells {
|
|
assert.Equal(t, c.cells[i][j], cell.Type, "case %d: row %d cell %d should be equal", n, i, j)
|
|
}
|
|
}
|
|
}
|
|
}
|