mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-12-05 05:27:51 +01:00
e81ccc406b
Change all license headers to comply with REUSE specification. Fix #16132 Co-authored-by: flynnnnnnnnnn <flynnnnnnnnnn@github> Co-authored-by: John Olheiser <john.olheiser@gmail.com>
173 lines
3.4 KiB
Go
173 lines
3.4 KiB
Go
// Copyright 2021 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package highlight
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func lines(s string) []string {
|
|
return strings.Split(strings.ReplaceAll(strings.TrimSpace(s), `\n`, "\n"), "\n")
|
|
}
|
|
|
|
func TestFile(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
code string
|
|
want []string
|
|
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><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="sa"></span><span class="s1">'</span><span class="s1">'</span>\n
|
|
\n
|
|
<span class="n">c</span><span class="o">=</span><span class="mi">2</span>`,
|
|
),
|
|
lexerName: "Python",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
out, lexerName, err := File(tt.name, "", []byte(tt.code))
|
|
assert.NoError(t, err)
|
|
expected := strings.Join(tt.want, "\n")
|
|
actual := strings.Join(out, "\n")
|
|
assert.Equal(t, strings.Count(actual, "<span"), strings.Count(actual, "</span>"))
|
|
assert.EqualValues(t, expected, actual)
|
|
assert.Equal(t, tt.lexerName, lexerName)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPlainText(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
code string
|
|
want []string
|
|
}{
|
|
{
|
|
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 := PlainText([]byte(tt.code))
|
|
expected := strings.Join(tt.want, "\n")
|
|
actual := strings.Join(out, "\n")
|
|
assert.EqualValues(t, expected, actual)
|
|
})
|
|
}
|
|
}
|