0
0
mirror of https://github.com/go-gitea/gitea.git synced 2024-12-13 05:40:12 +01:00
gitea/tests/integration/linguist_test.go

260 lines
7.0 KiB
Go

// Copyright 2024 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT
package integration
import (
"context"
"net/url"
"strings"
"testing"
"time"
"code.gitea.io/gitea/models/db"
repo_model "code.gitea.io/gitea/models/repo"
"code.gitea.io/gitea/models/unittest"
user_model "code.gitea.io/gitea/models/user"
"code.gitea.io/gitea/modules/git"
"code.gitea.io/gitea/modules/indexer/stats"
"code.gitea.io/gitea/modules/queue"
repo_service "code.gitea.io/gitea/services/repository"
files_service "code.gitea.io/gitea/services/repository/files"
"github.com/stretchr/testify/assert"
)
func TestLinguist(t *testing.T) {
onGiteaRun(t, func(t *testing.T, _ *url.URL) {
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
cppContent := "#include <iostream>\nint main() {\nstd::cout << \"Hello Gitea!\";\nreturn 0;\n}"
pyContent := "print(\"Hello Gitea!\")"
phpContent := "<?php\necho 'Hallo Welt';\n?>"
lockContent := "# This file is automatically @generated by Poetry 1.7.1 and should not be changed by hand."
mdContent := "markdown"
cases := []struct {
GitAttributesContent string
FilesToAdd []*files_service.ChangeRepoFile
ExpectedLanguageOrder []string
}{
// case 0
{
ExpectedLanguageOrder: []string{},
},
// case 1
{
FilesToAdd: []*files_service.ChangeRepoFile{
{
TreePath: "cplusplus.cpp",
ContentReader: strings.NewReader(cppContent),
},
{
TreePath: "python.py",
ContentReader: strings.NewReader(pyContent),
},
{
TreePath: "php.php",
ContentReader: strings.NewReader(phpContent),
},
},
ExpectedLanguageOrder: []string{"C++", "PHP", "Python"},
},
// case 2
{
FilesToAdd: []*files_service.ChangeRepoFile{
{
TreePath: ".cplusplus.cpp",
ContentReader: strings.NewReader(cppContent),
},
{
TreePath: "python.py",
ContentReader: strings.NewReader(pyContent),
},
{
TreePath: "vendor/php.php",
ContentReader: strings.NewReader(phpContent),
},
},
ExpectedLanguageOrder: []string{"Python"},
},
// case 3
{
GitAttributesContent: "*.cpp linguist-language=Go",
FilesToAdd: []*files_service.ChangeRepoFile{
{
TreePath: "cplusplus.cpp",
ContentReader: strings.NewReader(cppContent),
},
},
ExpectedLanguageOrder: []string{"Go"},
},
// case 4
{
GitAttributesContent: "*.cpp gitlab-language=Go?parent=json",
FilesToAdd: []*files_service.ChangeRepoFile{
{
TreePath: "cplusplus.cpp",
ContentReader: strings.NewReader(cppContent),
},
},
ExpectedLanguageOrder: []string{"Go"},
},
// case 5
{
GitAttributesContent: "*.cpp linguist-language=HTML gitlab-language=Go?parent=json",
FilesToAdd: []*files_service.ChangeRepoFile{
{
TreePath: "cplusplus.cpp",
ContentReader: strings.NewReader(cppContent),
},
},
ExpectedLanguageOrder: []string{"HTML"},
},
// case 6
{
GitAttributesContent: "vendor/** linguist-vendored=false",
FilesToAdd: []*files_service.ChangeRepoFile{
{
TreePath: "vendor/php.php",
ContentReader: strings.NewReader(phpContent),
},
},
ExpectedLanguageOrder: []string{"PHP"},
},
// case 7
{
GitAttributesContent: "*.cpp linguist-vendored=true\n*.py linguist-vendored\nvendor/** -linguist-vendored",
FilesToAdd: []*files_service.ChangeRepoFile{
{
TreePath: "cplusplus.cpp",
ContentReader: strings.NewReader(cppContent),
},
{
TreePath: "python.py",
ContentReader: strings.NewReader(pyContent),
},
{
TreePath: "vendor/php.php",
ContentReader: strings.NewReader(phpContent),
},
},
ExpectedLanguageOrder: []string{"PHP"},
},
// case 8
{
GitAttributesContent: "poetry.lock linguist-language=Go",
FilesToAdd: []*files_service.ChangeRepoFile{
{
TreePath: "poetry.lock",
ContentReader: strings.NewReader(lockContent),
},
},
ExpectedLanguageOrder: []string{"Go"},
},
// case 9
{
GitAttributesContent: "poetry.lock linguist-generated=false",
FilesToAdd: []*files_service.ChangeRepoFile{
{
TreePath: "poetry.lock",
ContentReader: strings.NewReader(lockContent),
},
},
ExpectedLanguageOrder: []string{"TOML"},
},
// case 10
{
GitAttributesContent: "*.cpp -linguist-detectable",
FilesToAdd: []*files_service.ChangeRepoFile{
{
TreePath: "cplusplus.cpp",
ContentReader: strings.NewReader(cppContent),
},
},
ExpectedLanguageOrder: []string{},
},
// case 11
{
GitAttributesContent: "*.md linguist-detectable",
FilesToAdd: []*files_service.ChangeRepoFile{
{
TreePath: "test.md",
ContentReader: strings.NewReader(mdContent),
},
},
ExpectedLanguageOrder: []string{"Markdown"},
},
// case 12
{
GitAttributesContent: "test2.md linguist-detectable",
FilesToAdd: []*files_service.ChangeRepoFile{
{
TreePath: "cplusplus.cpp",
ContentReader: strings.NewReader(cppContent),
},
{
TreePath: "test.md",
ContentReader: strings.NewReader(mdContent),
},
{
TreePath: "test2.md",
ContentReader: strings.NewReader(mdContent),
},
},
ExpectedLanguageOrder: []string{"C++", "Markdown"},
},
// case 13
{
GitAttributesContent: "README.md linguist-documentation=false",
FilesToAdd: []*files_service.ChangeRepoFile{
{
TreePath: "README.md",
ContentReader: strings.NewReader(mdContent),
},
},
ExpectedLanguageOrder: []string{"Markdown"},
},
}
for i, c := range cases {
repo, err := repo_service.CreateRepository(db.DefaultContext, user, user, repo_service.CreateRepoOptions{
Name: "linguist-test",
})
assert.NoError(t, err)
files := []*files_service.ChangeRepoFile{
{
TreePath: ".gitattributes",
ContentReader: strings.NewReader(c.GitAttributesContent),
},
}
files = append(files, c.FilesToAdd...)
for _, f := range files {
f.Operation = "create"
}
_, err = files_service.ChangeRepoFiles(git.DefaultContext, repo, user, &files_service.ChangeRepoFilesOptions{
Files: files,
OldBranch: repo.DefaultBranch,
NewBranch: repo.DefaultBranch,
})
assert.NoError(t, err)
assert.NoError(t, stats.UpdateRepoIndexer(repo))
assert.NoError(t, queue.GetManager().FlushAll(context.Background(), 10*time.Second))
stats, err := repo_model.GetTopLanguageStats(db.DefaultContext, repo, len(c.FilesToAdd))
assert.NoError(t, err)
languages := make([]string, 0, len(stats))
for _, s := range stats {
languages = append(languages, s.Language)
}
assert.Equal(t, c.ExpectedLanguageOrder, languages, "case %d: unexpected language stats", i)
assert.NoError(t, repo_service.DeleteRepository(db.DefaultContext, user, repo, false))
}
})
}