mirror of
https://github.com/go-gitea/gitea.git
synced 2026-05-28 02:38:44 +09:00
wxiaoguang
4a6db5a7c2
Fix #37849 --------- Signed-off-by: wxiaoguang <wxiaoguang@gmail.com> Co-authored-by: TheFox0x7 <thefox0x7@gmail.com>
219 lines
4.7 KiB
Go
219 lines
4.7 KiB
Go
// Copyright 2021 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package highlight
|
|
|
|
import (
|
|
"html/template"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func lines(s string) (out []template.HTML) {
|
|
// "" => [], "a" => ["a"], "a\n" => ["a\n"], "a\nb" => ["a\n", "b"] (each line always includes EOL "\n" if it exists)
|
|
out = make([]template.HTML, 0)
|
|
s = strings.ReplaceAll(strings.ReplaceAll(strings.TrimSpace(s), "\n", ""), `\n`, "\n")
|
|
for {
|
|
if p := strings.IndexByte(s, '\n'); p != -1 {
|
|
out = append(out, template.HTML(s[:p+1]))
|
|
s = s[p+1:]
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
if s != "" {
|
|
out = append(out, template.HTML(s))
|
|
}
|
|
return out
|
|
}
|
|
|
|
func TestFile(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
code string
|
|
want []template.HTML
|
|
lexerName string
|
|
}{
|
|
{
|
|
name: "empty.py",
|
|
code: "",
|
|
want: lines(""),
|
|
lexerName: "Python",
|
|
},
|
|
{
|
|
name: "empty.js",
|
|
code: "",
|
|
want: lines(""),
|
|
lexerName: "JavaScript",
|
|
},
|
|
{
|
|
name: "empty.yaml",
|
|
code: "",
|
|
want: lines(""),
|
|
lexerName: "YAML",
|
|
},
|
|
{
|
|
name: "tags.txt",
|
|
code: "<>",
|
|
want: lines("<>"),
|
|
lexerName: "Plaintext",
|
|
},
|
|
{
|
|
name: "tags.py",
|
|
code: "<>",
|
|
want: lines(`<span class="o"><></span>`),
|
|
lexerName: "Python",
|
|
},
|
|
{
|
|
name: "eol-no.py",
|
|
code: "a=1",
|
|
want: lines(`<span class="n">a</span><span class="o">=</span><span class="mi">1</span>`),
|
|
lexerName: "Python",
|
|
},
|
|
{
|
|
name: "eol-newline1.py",
|
|
code: "a=1\n",
|
|
want: lines(`<span class="n">a</span><span class="o">=</span><span class="mi">1</span>\n`),
|
|
lexerName: "Python",
|
|
},
|
|
{
|
|
name: "eol-newline2.py",
|
|
code: "a=1\n\n",
|
|
want: lines(`
|
|
<span class="n">a</span><span class="o">=</span><span class="mi">1</span>\n
|
|
\n
|
|
`,
|
|
),
|
|
lexerName: "Python",
|
|
},
|
|
{
|
|
name: "empty-line-with-space.py",
|
|
code: strings.ReplaceAll(strings.TrimSpace(`
|
|
def:
|
|
a=1
|
|
|
|
b=''
|
|
{space}
|
|
c=2
|
|
`), "{space}", " "),
|
|
want: lines(`
|
|
<span class="n">def</span><span class="p">:</span>\n
|
|
<span class="n">a</span><span class="o">=</span><span class="mi">1</span>\n
|
|
\n
|
|
<span class="n">b</span><span class="o">=</span><span class="s1">''</span>\n
|
|
\n
|
|
<span class="n">c</span><span class="o">=</span><span class="mi">2</span>`,
|
|
),
|
|
lexerName: "Python",
|
|
},
|
|
{
|
|
name: "test.sql",
|
|
code: "--\nSELECT",
|
|
want: []template.HTML{"<span class=\"c1\">--\n</span>", `<span class="k">SELECT</span>`},
|
|
lexerName: "SQL",
|
|
},
|
|
{
|
|
name: "test.http",
|
|
code: `HTTP/1.0 400 Bad request
|
|
Content-Type: text/html
|
|
|
|
<html></html>`,
|
|
want: lines(`<span class="kr">HTTP</span><span class="o">/</span><span class="m">1.0</span> <span class="m">400</span> <span class="ne">Bad request</span>\n
|
|
<span class="n">Content-Type</span><span class="o">:</span> <span class="l">text/html</span>\n
|
|
\n
|
|
<span class="p"><</span><span class="nt">html</span><span class="p">></</span><span class="nt">html</span><span class="p">></span>`),
|
|
lexerName: "HTTP",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
out, lexerName := RenderFullFile(tt.name, "", []byte(tt.code))
|
|
assert.Equal(t, tt.want, out)
|
|
assert.Equal(t, tt.lexerName, lexerName)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPlainText(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
code string
|
|
want []template.HTML
|
|
}{
|
|
{
|
|
name: "empty.py",
|
|
code: "",
|
|
want: lines(""),
|
|
},
|
|
{
|
|
name: "tags.py",
|
|
code: "<>",
|
|
want: lines("<>"),
|
|
},
|
|
{
|
|
name: "eol-no.py",
|
|
code: "a=1",
|
|
want: lines(`a=1`),
|
|
},
|
|
{
|
|
name: "eol-newline1.py",
|
|
code: "a=1\n",
|
|
want: lines(`a=1\n`),
|
|
},
|
|
{
|
|
name: "eol-newline2.py",
|
|
code: "a=1\n\n",
|
|
want: lines(`
|
|
a=1\n
|
|
\n
|
|
`),
|
|
},
|
|
{
|
|
name: "empty-line-with-space.py",
|
|
code: strings.ReplaceAll(strings.TrimSpace(`
|
|
def:
|
|
a=1
|
|
|
|
b=''
|
|
{space}
|
|
c=2
|
|
`), "{space}", " "),
|
|
want: lines(`
|
|
def:\n
|
|
a=1\n
|
|
\n
|
|
b=''\n
|
|
\n
|
|
c=2`),
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
out := renderPlainText([]byte(tt.code))
|
|
assert.Equal(t, tt.want, out)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestUnsafeSplitHighlightedLines(t *testing.T) {
|
|
ret := UnsafeSplitHighlightedLines("")
|
|
assert.Empty(t, ret)
|
|
|
|
ret = UnsafeSplitHighlightedLines("a")
|
|
assert.Len(t, ret, 1)
|
|
assert.Equal(t, "a", string(ret[0]))
|
|
|
|
ret = UnsafeSplitHighlightedLines("\n")
|
|
assert.Len(t, ret, 1)
|
|
assert.Equal(t, "\n", string(ret[0]))
|
|
|
|
ret = UnsafeSplitHighlightedLines("<span>a</span>\n<span>b\n</span>")
|
|
assert.Len(t, ret, 2)
|
|
assert.Equal(t, "<span>a</span>\n", string(ret[0]))
|
|
assert.Equal(t, "<span>b\n</span>", string(ret[1]))
|
|
}
|