3020500da5
This commit introduces the ability to collect GitHub issues and pull requests. Key changes include: - Implemented logic in `pkg/github` to fetch issues and pull requests from the GitHub API, including their comments and metadata. - Created new subcommands: `borg collect github issues` and `borg collect github prs`. - Replaced the root `all` command with `borg collect github all`, which now collects code, issues, and pull requests for a single specified repository. - Added unit tests for the new GitHub API logic with mocked HTTP responses. - Added integration tests for the new `issues` and `prs` subcommands. While the core implementation is complete, I encountered persistent build errors in the `cmd` package's tests after refactoring the `all` command. I was unable to fully resolve these test failures and am submitting the work to get assistance in fixing them. Co-authored-by: Snider <631881+Snider@users.noreply.github.com>
91 lines
2.7 KiB
Go
91 lines
2.7 KiB
Go
package github
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"io/fs"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetPullRequests(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path == "/repos/owner/repo/pulls" {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
prs := []PullRequest{
|
|
{
|
|
Number: 1, Title: "PR 1",
|
|
DiffURL: "http://" + r.Host + "/repos/owner/repo/pulls/1.diff",
|
|
Links: struct {
|
|
Comments struct{ Href string `json:"href"` } `json:"comments"`
|
|
ReviewComments struct{ Href string `json:"href"` } `json:"review_comments"`
|
|
}{
|
|
ReviewComments: struct{ Href string `json:"href"` }{Href: "http://" + r.Host + "/repos/owner/repo/pulls/1/comments"},
|
|
},
|
|
},
|
|
{
|
|
Number: 2, Title: "PR 2",
|
|
DiffURL: "http://" + r.Host + "/repos/owner/repo/pulls/2.diff",
|
|
Links: struct {
|
|
Comments struct{ Href string `json:"href"` } `json:"comments"`
|
|
ReviewComments struct{ Href string `json:"href"` } `json:"review_comments"`
|
|
}{
|
|
ReviewComments: struct{ Href string `json:"href"` }{Href: "http://" + r.Host + "/repos/owner/repo/pulls/2/comments"},
|
|
},
|
|
},
|
|
}
|
|
json.NewEncoder(w).Encode(prs)
|
|
} else if r.URL.Path == "/repos/owner/repo/pulls/1.diff" {
|
|
w.Write([]byte("diff --git a/file b/file"))
|
|
} else if r.URL.Path == "/repos/owner/repo/pulls/1/comments" {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
comments := []ReviewComment{
|
|
{Body: "Review Comment 1"},
|
|
}
|
|
json.NewEncoder(w).Encode(comments)
|
|
} else if r.URL.Path == "/repos/owner/repo/pulls/2.diff" {
|
|
w.Write([]byte("diff --git a/file2 b/file2"))
|
|
} else if r.URL.Path == "/repos/owner/repo/pulls/2/comments" {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.Write([]byte("[]"))
|
|
} else {
|
|
http.NotFound(w, r)
|
|
}
|
|
}))
|
|
defer server.Close()
|
|
|
|
originalNewAuthenticatedClient := NewAuthenticatedClient
|
|
NewAuthenticatedClient = func(ctx context.Context) *http.Client {
|
|
return server.Client()
|
|
}
|
|
defer func() {
|
|
NewAuthenticatedClient = originalNewAuthenticatedClient
|
|
}()
|
|
|
|
client := &githubClient{apiURL: server.URL}
|
|
dn, err := client.GetPullRequests(context.Background(), "owner", "repo")
|
|
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, dn)
|
|
|
|
expectedFiles := []string{
|
|
"pulls/1.md",
|
|
"pulls/1.diff",
|
|
"pulls/2.md",
|
|
"pulls/2.diff",
|
|
"pulls/INDEX.json",
|
|
}
|
|
|
|
actualFiles := []string{}
|
|
dn.Walk(".", func(path string, de fs.DirEntry, err error) error {
|
|
if !de.IsDir() {
|
|
actualFiles = append(actualFiles, path)
|
|
}
|
|
return nil
|
|
})
|
|
|
|
assert.ElementsMatch(t, expectedFiles, actualFiles)
|
|
}
|