summaryrefslogtreecommitdiffstats
path: root/models/git_diff_test.go
blob: f88e6fb6104bb1f5921f599226cc9b53273c198b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
package models

import (
	dmp "github.com/sergi/go-diff/diffmatchpatch"
	"html/template"
	"testing"
)

func assertEqual(t *testing.T, s1 string, s2 template.HTML) {
	if s1 != string(s2) {
		t.Errorf("%s should be equal %s", s2, s1)
	}
}

func assertLineEqual(t *testing.T, d1 *DiffLine, d2 *DiffLine) {
	if d1 != d2 {
		t.Errorf("%v should be equal %v", d1, d2)
	}
}

func TestDiffToHTML(t *testing.T) {
	assertEqual(t, "+foo <span class=\"added-code\">bar</span> biz", diffToHTML([]dmp.Diff{
		{dmp.DiffEqual, "foo "},
		{dmp.DiffInsert, "bar"},
		{dmp.DiffDelete, " baz"},
		{dmp.DiffEqual, " biz"},
	}, DiffLineAdd))

	assertEqual(t, "-foo <span class=\"removed-code\">bar</span> biz", diffToHTML([]dmp.Diff{
		{dmp.DiffEqual, "foo "},
		{dmp.DiffDelete, "bar"},
		{dmp.DiffInsert, " baz"},
		{dmp.DiffEqual, " biz"},
	}, DiffLineDel))
}