refactor: decouple context from migration structs (#33399)
Use context as much as possible. --------- Co-authored-by: wxiaoguang <wxiaoguang@gmail.com>
This commit is contained in:
parent
466cc725bc
commit
1ec8d80fa3
23 changed files with 455 additions and 509 deletions
|
@ -21,11 +21,12 @@ func TestGitHubDownloadRepo(t *testing.T) {
|
|||
if token == "" {
|
||||
t.Skip("Skipping GitHub migration test because GITHUB_READ_TOKEN is empty")
|
||||
}
|
||||
downloader := NewGithubDownloaderV3(context.Background(), "https://github.com", "", "", token, "go-gitea", "test_repo")
|
||||
err := downloader.RefreshRate()
|
||||
ctx := context.Background()
|
||||
downloader := NewGithubDownloaderV3(ctx, "https://github.com", "", "", token, "go-gitea", "test_repo")
|
||||
err := downloader.RefreshRate(ctx)
|
||||
assert.NoError(t, err)
|
||||
|
||||
repo, err := downloader.GetRepoInfo()
|
||||
repo, err := downloader.GetRepoInfo(ctx)
|
||||
assert.NoError(t, err)
|
||||
assertRepositoryEqual(t, &base.Repository{
|
||||
Name: "test_repo",
|
||||
|
@ -36,11 +37,11 @@ func TestGitHubDownloadRepo(t *testing.T) {
|
|||
DefaultBranch: "master",
|
||||
}, repo)
|
||||
|
||||
topics, err := downloader.GetTopics()
|
||||
topics, err := downloader.GetTopics(ctx)
|
||||
assert.NoError(t, err)
|
||||
assert.Contains(t, topics, "gitea")
|
||||
|
||||
milestones, err := downloader.GetMilestones()
|
||||
milestones, err := downloader.GetMilestones(ctx)
|
||||
assert.NoError(t, err)
|
||||
assertMilestonesEqual(t, []*base.Milestone{
|
||||
{
|
||||
|
@ -63,7 +64,7 @@ func TestGitHubDownloadRepo(t *testing.T) {
|
|||
},
|
||||
}, milestones)
|
||||
|
||||
labels, err := downloader.GetLabels()
|
||||
labels, err := downloader.GetLabels(ctx)
|
||||
assert.NoError(t, err)
|
||||
assertLabelsEqual(t, []*base.Label{
|
||||
{
|
||||
|
@ -113,7 +114,7 @@ func TestGitHubDownloadRepo(t *testing.T) {
|
|||
},
|
||||
}, labels)
|
||||
|
||||
releases, err := downloader.GetReleases()
|
||||
releases, err := downloader.GetReleases(ctx)
|
||||
assert.NoError(t, err)
|
||||
assertReleasesEqual(t, []*base.Release{
|
||||
{
|
||||
|
@ -129,7 +130,7 @@ func TestGitHubDownloadRepo(t *testing.T) {
|
|||
}, releases)
|
||||
|
||||
// downloader.GetIssues()
|
||||
issues, isEnd, err := downloader.GetIssues(1, 2)
|
||||
issues, isEnd, err := downloader.GetIssues(ctx, 1, 2)
|
||||
assert.NoError(t, err)
|
||||
assert.False(t, isEnd)
|
||||
assertIssuesEqual(t, []*base.Issue{
|
||||
|
@ -218,7 +219,7 @@ func TestGitHubDownloadRepo(t *testing.T) {
|
|||
}, issues)
|
||||
|
||||
// downloader.GetComments()
|
||||
comments, _, err := downloader.GetComments(&base.Issue{Number: 2, ForeignIndex: 2})
|
||||
comments, _, err := downloader.GetComments(ctx, &base.Issue{Number: 2, ForeignIndex: 2})
|
||||
assert.NoError(t, err)
|
||||
assertCommentsEqual(t, []*base.Comment{
|
||||
{
|
||||
|
@ -248,7 +249,7 @@ func TestGitHubDownloadRepo(t *testing.T) {
|
|||
}, comments)
|
||||
|
||||
// downloader.GetPullRequests()
|
||||
prs, _, err := downloader.GetPullRequests(1, 2)
|
||||
prs, _, err := downloader.GetPullRequests(ctx, 1, 2)
|
||||
assert.NoError(t, err)
|
||||
assertPullRequestsEqual(t, []*base.PullRequest{
|
||||
{
|
||||
|
@ -338,7 +339,7 @@ func TestGitHubDownloadRepo(t *testing.T) {
|
|||
},
|
||||
}, prs)
|
||||
|
||||
reviews, err := downloader.GetReviews(&base.PullRequest{Number: 3, ForeignIndex: 3})
|
||||
reviews, err := downloader.GetReviews(ctx, &base.PullRequest{Number: 3, ForeignIndex: 3})
|
||||
assert.NoError(t, err)
|
||||
assertReviewsEqual(t, []*base.Review{
|
||||
{
|
||||
|
@ -370,7 +371,7 @@ func TestGitHubDownloadRepo(t *testing.T) {
|
|||
},
|
||||
}, reviews)
|
||||
|
||||
reviews, err = downloader.GetReviews(&base.PullRequest{Number: 4, ForeignIndex: 4})
|
||||
reviews, err = downloader.GetReviews(ctx, &base.PullRequest{Number: 4, ForeignIndex: 4})
|
||||
assert.NoError(t, err)
|
||||
assertReviewsEqual(t, []*base.Review{
|
||||
{
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue