Платформа ЦРНП "Мирокод" для разработки проектов
https://git.mirocod.ru
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.3 KiB
43 lines
1.3 KiB
// Copyright 2022 The Gitea Authors. All rights reserved. |
|
// Use of this source code is governed by a MIT-style |
|
// license that can be found in the LICENSE file. |
|
|
|
package repository |
|
|
|
import ( |
|
"testing" |
|
|
|
"code.gitea.io/gitea/models/db" |
|
repo_model "code.gitea.io/gitea/models/repo" |
|
"code.gitea.io/gitea/models/unit" |
|
"code.gitea.io/gitea/models/unittest" |
|
|
|
"github.com/stretchr/testify/assert" |
|
) |
|
|
|
func TestLinkedRepository(t *testing.T) { |
|
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
testCases := []struct { |
|
name string |
|
attachID int64 |
|
expectedRepo *repo_model.Repository |
|
expectedUnitType unit.Type |
|
}{ |
|
{"LinkedIssue", 1, &repo_model.Repository{ID: 1}, unit.TypeIssues}, |
|
{"LinkedComment", 3, &repo_model.Repository{ID: 1}, unit.TypePullRequests}, |
|
{"LinkedRelease", 9, &repo_model.Repository{ID: 1}, unit.TypeReleases}, |
|
{"Notlinked", 10, nil, -1}, |
|
} |
|
for _, tc := range testCases { |
|
t.Run(tc.name, func(t *testing.T) { |
|
attach, err := repo_model.GetAttachmentByID(db.DefaultContext, tc.attachID) |
|
assert.NoError(t, err) |
|
repo, unitType, err := LinkedRepository(attach) |
|
assert.NoError(t, err) |
|
if tc.expectedRepo != nil { |
|
assert.Equal(t, tc.expectedRepo.ID, repo.ID) |
|
} |
|
assert.Equal(t, tc.expectedUnitType, unitType) |
|
}) |
|
} |
|
}
|
|
|