1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
package shared
import (
"fmt"
"net/http"
"net/url"
"testing"
"github.com/cli/cli/v2/internal/ghrepo"
"github.com/cli/cli/v2/pkg/httpmock"
"github.com/stretchr/testify/assert"
)
func TestDownloadWorkflowArtifactsPageinates(t *testing.T) {
testRepoOwner := "OWNER"
testRepoName := "REPO"
testRunId := "1234567890"
reg := &httpmock.Registry{}
defer reg.Verify(t)
firstReq := httpmock.QueryMatcher(
"GET",
fmt.Sprintf("repos/%s/%s/actions/runs/%s/artifacts", testRepoOwner, testRepoName, testRunId),
url.Values{"per_page": []string{"100"}},
)
firstArtifact := Artifact{
Name: "artifact-0",
Size: 2,
DownloadURL: fmt.Sprintf("https://api.github.com/repos/%s/%s/actions/artifacts/987654320/zip", testRepoOwner, testRepoName),
Expired: true,
}
firstRes := httpmock.JSONResponse(artifactsPayload{Artifacts: []Artifact{firstArtifact}})
testLinkUri := fmt.Sprintf("repositories/123456789/actions/runs/%s/artifacts", testRunId)
testLinkUrl := fmt.Sprintf("https://api.github.com/%s", testLinkUri)
firstRes = httpmock.WithHeader(
firstRes,
"Link",
fmt.Sprintf(`<%s?per_page=100&page=2>; rel="next", <%s?per_page=100&page=2>; rel="last"`, testLinkUrl, testLinkUrl),
)
secondReq := httpmock.QueryMatcher(
"GET",
testLinkUri,
url.Values{"per_page": []string{"100"}, "page": []string{"2"}},
)
secondArtifact := Artifact{
Name: "artifact-1",
Size: 2,
DownloadURL: fmt.Sprintf("https://api.github.com/repos/%s/%s/actions/artifacts/987654321/zip", testRepoOwner, testRepoName),
Expired: false,
}
secondRes := httpmock.JSONResponse(artifactsPayload{Artifacts: []Artifact{secondArtifact}})
reg.Register(firstReq, firstRes)
reg.Register(secondReq, secondRes)
httpClient := &http.Client{Transport: reg}
repo := ghrepo.New(testRepoOwner, testRepoName)
result, err := ListArtifacts(httpClient, repo, testRunId)
assert.NoError(t, err)
assert.Equal(t, []Artifact{firstArtifact, secondArtifact}, result)
}
|
