f3eb092d7e
This commit is another isolation refactor, this time for repo fork. However, I got fed up with the --remote="true|false|prompt" style of flags and took this opportunity to switch to a set of bool flags: --remote and --clone --no-remote and --no-clone the string args were really non standard and confusing; with only two bools it was impossible to tell when to prompt.
104 lines
2.8 KiB
Go
104 lines
2.8 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) 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])
|
|
}
|