|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081 |
- // Copyright 2018 Google LLC
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
-
- package storage
-
- import (
- "context"
- "encoding/json"
- "fmt"
- "io"
- "io/ioutil"
- "net/http"
- "strings"
- "testing"
-
- "google.golang.org/api/option"
- )
-
- type mockTransport struct {
- gotReq *http.Request
- gotBody []byte
- results []transportResult
- }
-
- type transportResult struct {
- res *http.Response
- err error
- }
-
- func (t *mockTransport) addResult(res *http.Response, err error) {
- t.results = append(t.results, transportResult{res, err})
- }
-
- func (t *mockTransport) RoundTrip(req *http.Request) (*http.Response, error) {
- t.gotReq = req
- t.gotBody = nil
- if req.Body != nil {
- bytes, err := ioutil.ReadAll(req.Body)
- if err != nil {
- return nil, err
- }
- t.gotBody = bytes
- }
- if len(t.results) == 0 {
- return nil, fmt.Errorf("error handling request")
- }
- result := t.results[0]
- t.results = t.results[1:]
- return result.res, result.err
- }
-
- func (t *mockTransport) gotJSONBody() map[string]interface{} {
- m := map[string]interface{}{}
- if err := json.Unmarshal(t.gotBody, &m); err != nil {
- panic(err)
- }
- return m
- }
-
- func mockClient(t *testing.T, m *mockTransport) *Client {
- client, err := NewClient(context.Background(), option.WithHTTPClient(&http.Client{Transport: m}))
- if err != nil {
- t.Fatal(err)
- }
- return client
- }
-
- func bodyReader(s string) io.ReadCloser {
- return ioutil.NopCloser(strings.NewReader(s))
- }
|