7a8db80420
We no longer guess the head repository using heuristics; instead, we present the user with the choice of pushable repositories and an additional option to create a new fork. The new `pr create --head` flag is available for the user to specify the head branch in `branch` or `owner:branch` format and completely skip any forking or auto-pushing checks.
120 lines
3.1 KiB
Go
120 lines
3.1 KiB
Go
package httpmock
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
)
|
|
|
|
// TODO: clean up methods in this file when there are no more callers
|
|
|
|
func (r *Registry) StubResponse(status int, body io.Reader) {
|
|
r.Register(MatchAny, func(req *http.Request) (*http.Response, error) {
|
|
return httpResponse(status, req, body), nil
|
|
})
|
|
}
|
|
|
|
func (r *Registry) StubWithFixture(status int, fixtureFileName string) func() {
|
|
fixturePath := path.Join("../test/fixtures/", fixtureFileName)
|
|
fixtureFile, err := os.Open(fixturePath)
|
|
r.Register(MatchAny, func(req *http.Request) (*http.Response, error) {
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return httpResponse(200, req, fixtureFile), nil
|
|
})
|
|
return func() {
|
|
if err == nil {
|
|
fixtureFile.Close()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (r *Registry) StubWithFixturePath(status int, fixturePath string) func() {
|
|
fixtureFile, err := os.Open(fixturePath)
|
|
r.Register(MatchAny, func(req *http.Request) (*http.Response, error) {
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return httpResponse(200, req, fixtureFile), nil
|
|
})
|
|
return func() {
|
|
if err == nil {
|
|
fixtureFile.Close()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (r *Registry) StubRepoInfoResponse(owner, repo, branch string) {
|
|
r.Register(
|
|
GraphQL(`query RepositoryInfo\b`),
|
|
StringResponse(fmt.Sprintf(`
|
|
{ "data": { "repository": {
|
|
"id": "REPOID",
|
|
"name": "%s",
|
|
"owner": {"login": "%s"},
|
|
"description": "",
|
|
"defaultBranchRef": {"name": "%s"},
|
|
"hasIssuesEnabled": true,
|
|
"viewerPermission": "WRITE"
|
|
} } }
|
|
`, repo, owner, branch)))
|
|
}
|
|
|
|
func (r *Registry) StubRepoResponse(owner, repo string) {
|
|
r.StubRepoResponseWithPermission(owner, repo, "WRITE")
|
|
}
|
|
|
|
func (r *Registry) StubRepoResponseWithPermission(owner, repo, permission string) {
|
|
r.Register(GraphQL(`query RepositoryNetwork\b`), StringResponse(RepoNetworkStubResponse(owner, repo, "master", permission)))
|
|
}
|
|
|
|
func (r *Registry) StubRepoResponseWithDefaultBranch(owner, repo, defaultBranch string) {
|
|
r.Register(GraphQL(`query RepositoryNetwork\b`), StringResponse(RepoNetworkStubResponse(owner, repo, defaultBranch, "WRITE")))
|
|
}
|
|
|
|
func (r *Registry) StubForkedRepoResponse(ownRepo, parentRepo string) {
|
|
r.Register(GraphQL(`query RepositoryNetwork\b`), StringResponse(RepoNetworkStubForkResponse(ownRepo, parentRepo)))
|
|
}
|
|
|
|
func RepoNetworkStubResponse(owner, repo, defaultBranch, permission string) string {
|
|
return fmt.Sprintf(`
|
|
{ "data": { "repo_000": {
|
|
"id": "REPOID",
|
|
"name": "%s",
|
|
"owner": {"login": "%s"},
|
|
"defaultBranchRef": {
|
|
"name": "%s"
|
|
},
|
|
"viewerPermission": "%s"
|
|
} } }
|
|
`, repo, owner, defaultBranch, permission)
|
|
}
|
|
|
|
func RepoNetworkStubForkResponse(forkFullName, parentFullName string) string {
|
|
forkRepo := strings.SplitN(forkFullName, "/", 2)
|
|
parentRepo := strings.SplitN(parentFullName, "/", 2)
|
|
return fmt.Sprintf(`
|
|
{ "data": { "repo_000": {
|
|
"id": "REPOID2",
|
|
"name": "%s",
|
|
"owner": {"login": "%s"},
|
|
"defaultBranchRef": {
|
|
"name": "master"
|
|
},
|
|
"viewerPermission": "ADMIN",
|
|
"parent": {
|
|
"id": "REPOID1",
|
|
"name": "%s",
|
|
"owner": {"login": "%s"},
|
|
"defaultBranchRef": {
|
|
"name": "master"
|
|
},
|
|
"viewerPermission": "READ"
|
|
}
|
|
} } }
|
|
`, forkRepo[1], forkRepo[0], parentRepo[1], parentRepo[0])
|
|
}
|