1// Copyright 2020 The Gitea Authors. All rights reserved.
2// Use of this source code is governed by a MIT-style
3// license that can be found in the LICENSE file.
4
5package git
6
7import (
8	"strings"
9	"testing"
10
11	"github.com/stretchr/testify/assert"
12)
13
14const exampleDiff = `diff --git a/README.md b/README.md
15--- a/README.md
16+++ b/README.md
17@@ -1,3 +1,6 @@
18 # gitea-github-migrator
19+
20+ Build Status
21- Latest Release
22 Docker Pulls
23+ cut off
24+ cut off`
25
26const breakingDiff = `diff --git a/aaa.sql b/aaa.sql
27index d8e4c92..19dc8ad 100644
28--- a/aaa.sql
29+++ b/aaa.sql
30@@ -1,9 +1,10 @@
31 --some comment
32--- some comment 5
33+--some coment 2
34+-- some comment 3
35 create or replace procedure test(p1 varchar2)
36 is
37 begin
38---new comment
39 dbms_output.put_line(p1);
40+--some other comment
41 end;
42 /
43`
44
45var issue17875Diff = `diff --git a/Geschäftsordnung.md b/Geschäftsordnung.md
46index d46c152..a7d2d55 100644
47--- a/Geschäftsordnung.md
48+++ b/Geschäftsordnung.md
49@@ -1,5 +1,5 @@
50 ---
51-date: "23.01.2021"
52+date: "30.11.2021"
53 ...
54 ` + `
55 # Geschäftsordnung
56@@ -16,4 +16,22 @@ Diese Geschäftsordnung regelt alle Prozesse des Vereins, solange diese nicht du
57 ` + `
58 ## § 3 Datenschutzverantwortlichkeit
59 ` + `
60-1. Der Verein bestellt eine datenschutzverantwortliche Person mit den Aufgaben nach Artikel 39 DSGVO.
61\ No newline at end of file
62+1. Der Verein bestellt eine datenschutzverantwortliche Person mit den Aufgaben nach Artikel 39 DSGVO.
63+
64+## §4 Umgang mit der SARS-Cov-2-Pandemie
65+
66+1. Der Vorstand hat die Befugnis, in Rücksprache mit den Vereinsmitgliedern, verschiedene Hygienemaßnahmen für Präsenzveranstaltungen zu beschließen.
67+
68+2. Die Einführung, Änderung und Abschaffung dieser Maßnahmen sind nur zum Zweck der Eindämmung der SARS-Cov-2-Pandemie zulässig.
69+
70+3. Die Einführung, Änderung und Abschaffung von Maßnahmen nach Abs. 2 bedarf einer wissenschaftlichen Grundlage.
71+
72+4. Die Maßnahmen nach Abs. 2 setzen sich aus den folgenden Bausteinen inklusive einer ihrer Ausprägungen zusammen.
73+
74+	1. Maskenpflicht: Keine; Maskenpflicht, außer am Platz, oder wo Abstände nicht eingehalten werden können; Maskenpflicht, wenn Abstände nicht eingehalten werden können;  Maskenpflicht
75+
76+	2. Geimpft-, Genesen- oder Testnachweis: Kein Nachweis notwendig; Nachweis, dass Person geimpft, genesen oder tagesaktuell getestet ist (3G); Nachweis, dass Person geimpft oder genesen ist (2G); Nachweis, dass Person geimpft bzw. genesen und tagesaktuell getestet ist (2G+)
77+
78+	3. Online-Veranstaltung: Keine, parallele Online-Veranstaltung, ausschließlich Online-Veranstaltung
79+
80+5. Bei Präsenzveranstungen gelten außerdem die Hygienevorschriften des Veranstaltungsorts. Bei Regelkollision greift die restriktivere Regel.
81\ No newline at end of file`
82
83func TestCutDiffAroundLineIssue17875(t *testing.T) {
84	result, err := CutDiffAroundLine(strings.NewReader(issue17875Diff), 23, false, 3)
85	assert.NoError(t, err)
86	expected := `diff --git a/Geschäftsordnung.md b/Geschäftsordnung.md
87--- a/Geschäftsordnung.md
88+++ b/Geschäftsordnung.md
89@@ -20,0 +21,3 @@
90+## §4 Umgang mit der SARS-Cov-2-Pandemie
91+
92+1. Der Vorstand hat die Befugnis, in Rücksprache mit den Vereinsmitgliedern, verschiedene Hygienemaßnahmen für Präsenzveranstaltungen zu beschließen.`
93	assert.Equal(t, expected, result)
94}
95
96func TestCutDiffAroundLine(t *testing.T) {
97	result, err := CutDiffAroundLine(strings.NewReader(exampleDiff), 4, false, 3)
98	assert.NoError(t, err)
99	resultByLine := strings.Split(result, "\n")
100	assert.Len(t, resultByLine, 7)
101	// Check if headers got transferred
102	assert.Equal(t, "diff --git a/README.md b/README.md", resultByLine[0])
103	assert.Equal(t, "--- a/README.md", resultByLine[1])
104	assert.Equal(t, "+++ b/README.md", resultByLine[2])
105	// Check if hunk header is calculated correctly
106	assert.Equal(t, "@@ -2,2 +3,2 @@", resultByLine[3])
107	// Check if line got transferred
108	assert.Equal(t, "+ Build Status", resultByLine[4])
109
110	// Must be same result as before since old line 3 == new line 5
111	newResult, err := CutDiffAroundLine(strings.NewReader(exampleDiff), 3, true, 3)
112	assert.NoError(t, err)
113	assert.Equal(t, result, newResult, "Must be same result as before since old line 3 == new line 5")
114
115	newResult, err = CutDiffAroundLine(strings.NewReader(exampleDiff), 6, false, 300)
116	assert.NoError(t, err)
117	assert.Equal(t, exampleDiff, newResult)
118
119	emptyResult, err := CutDiffAroundLine(strings.NewReader(exampleDiff), 6, false, 0)
120	assert.NoError(t, err)
121	assert.Empty(t, emptyResult)
122
123	// Line is out of scope
124	emptyResult, err = CutDiffAroundLine(strings.NewReader(exampleDiff), 434, false, 0)
125	assert.NoError(t, err)
126	assert.Empty(t, emptyResult)
127
128	// Handle minus diffs properly
129	minusDiff, err := CutDiffAroundLine(strings.NewReader(breakingDiff), 2, false, 4)
130	assert.NoError(t, err)
131
132	expected := `diff --git a/aaa.sql b/aaa.sql
133--- a/aaa.sql
134+++ b/aaa.sql
135@@ -1,9 +1,10 @@
136 --some comment
137--- some comment 5
138+--some coment 2`
139	assert.Equal(t, expected, minusDiff)
140
141	// Handle minus diffs properly
142	minusDiff, err = CutDiffAroundLine(strings.NewReader(breakingDiff), 3, false, 4)
143	assert.NoError(t, err)
144
145	expected = `diff --git a/aaa.sql b/aaa.sql
146--- a/aaa.sql
147+++ b/aaa.sql
148@@ -1,9 +1,10 @@
149 --some comment
150--- some comment 5
151+--some coment 2
152+-- some comment 3`
153
154	assert.Equal(t, expected, minusDiff)
155}
156
157func BenchmarkCutDiffAroundLine(b *testing.B) {
158	for n := 0; n < b.N; n++ {
159		CutDiffAroundLine(strings.NewReader(exampleDiff), 3, true, 3)
160	}
161}
162
163func ExampleCutDiffAroundLine() {
164	const diff = `diff --git a/README.md b/README.md
165--- a/README.md
166+++ b/README.md
167@@ -1,3 +1,6 @@
168 # gitea-github-migrator
169+
170+ Build Status
171- Latest Release
172 Docker Pulls
173+ cut off
174+ cut off`
175	result, _ := CutDiffAroundLine(strings.NewReader(diff), 4, false, 3)
176	println(result)
177}
178
179func TestParseDiffHunkString(t *testing.T) {
180	leftLine, leftHunk, rightLine, rightHunk := ParseDiffHunkString("@@ -19,3 +19,5 @@ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER")
181	assert.EqualValues(t, 19, leftLine)
182	assert.EqualValues(t, 3, leftHunk)
183	assert.EqualValues(t, 19, rightLine)
184	assert.EqualValues(t, 5, rightHunk)
185}
186