714ab71ddc
* Ensure that all migration requests are cancellable Signed-off-by: Andrew Thornton <art27@cantab.net> * Use WithContext as RequestWithContext is go 1.14 Signed-off-by: Andrew Thornton <art27@cantab.net> Co-authored-by: techknowlogick <techknowlogick@gitea.io>
112 lines
3.1 KiB
Go
112 lines
3.1 KiB
Go
// Copyright 2019 The Gitea Authors. All rights reserved.
|
|
// Copyright 2018 Jonas Franz. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package migrations
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"code.gitea.io/gitea/models"
|
|
"code.gitea.io/gitea/modules/graceful"
|
|
"code.gitea.io/gitea/modules/structs"
|
|
"code.gitea.io/gitea/modules/util"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGiteaUploadRepo(t *testing.T) {
|
|
// FIXME: Since no accesskey or user/password will trigger rate limit of github, just skip
|
|
t.Skip()
|
|
|
|
models.PrepareTestEnv(t)
|
|
|
|
user := models.AssertExistsAndLoadBean(t, &models.User{ID: 1}).(*models.User)
|
|
|
|
var (
|
|
downloader = NewGithubDownloaderV3(context.Background(), "", "", "", "go-xorm", "builder")
|
|
repoName = "builder-" + time.Now().Format("2006-01-02-15-04-05")
|
|
uploader = NewGiteaLocalUploader(graceful.GetManager().HammerContext(), user, user.Name, repoName)
|
|
)
|
|
|
|
err := migrateRepository(downloader, uploader, structs.MigrateRepoOption{
|
|
CloneAddr: "https://github.com/go-xorm/builder",
|
|
RepoName: repoName,
|
|
AuthUsername: "",
|
|
|
|
Wiki: true,
|
|
Issues: true,
|
|
Milestones: true,
|
|
Labels: true,
|
|
Releases: true,
|
|
Comments: true,
|
|
PullRequests: true,
|
|
Private: true,
|
|
Mirror: false,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
repo := models.AssertExistsAndLoadBean(t, &models.Repository{OwnerID: user.ID, Name: repoName}).(*models.Repository)
|
|
assert.True(t, repo.HasWiki())
|
|
|
|
milestones, err := models.GetMilestones(models.GetMilestonesOption{
|
|
RepoID: repo.ID,
|
|
State: structs.StateOpen,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, len(milestones))
|
|
|
|
milestones, err = models.GetMilestones(models.GetMilestonesOption{
|
|
RepoID: repo.ID,
|
|
State: structs.StateClosed,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 0, len(milestones))
|
|
|
|
labels, err := models.GetLabelsByRepoID(repo.ID, "", models.ListOptions{})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 11, len(labels))
|
|
|
|
releases, err := models.GetReleasesByRepoID(repo.ID, models.FindReleasesOptions{
|
|
ListOptions: models.ListOptions{
|
|
PageSize: 10,
|
|
Page: 0,
|
|
},
|
|
IncludeTags: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 8, len(releases))
|
|
|
|
releases, err = models.GetReleasesByRepoID(repo.ID, models.FindReleasesOptions{
|
|
ListOptions: models.ListOptions{
|
|
PageSize: 10,
|
|
Page: 0,
|
|
},
|
|
IncludeTags: false,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, len(releases))
|
|
|
|
issues, err := models.Issues(&models.IssuesOptions{
|
|
RepoIDs: []int64{repo.ID},
|
|
IsPull: util.OptionalBoolFalse,
|
|
SortType: "oldest",
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 14, len(issues))
|
|
assert.NoError(t, issues[0].LoadDiscussComments())
|
|
assert.EqualValues(t, 0, len(issues[0].Comments))
|
|
|
|
pulls, _, err := models.PullRequests(repo.ID, &models.PullRequestsOptions{
|
|
SortType: "oldest",
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 34, len(pulls))
|
|
assert.NoError(t, pulls[0].LoadIssue())
|
|
assert.NoError(t, pulls[0].Issue.LoadDiscussComments())
|
|
assert.EqualValues(t, 2, len(pulls[0].Issue.Comments))
|
|
}
|