blob: 2db9f78f0243a8754a537117f4d82f7409375da2 (
plain)
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
|
package vault
import (
"fmt"
"net/http"
"net/http/httptest"
"net/url"
"github.com/hashicorp/vault/api"
)
// MockServer -
func MockServer(code int, body string) (*httptest.Server, *Vault) {
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(code)
// nolint: errcheck
fmt.Fprintln(w, body)
}))
tr := &http.Transport{
Proxy: func(req *http.Request) (*url.URL, error) {
return url.Parse(server.URL)
},
}
httpClient := &http.Client{Transport: tr}
config := &api.Config{
Address: server.URL,
HttpClient: httpClient,
}
// nolint: gosec
c, _ := api.NewClient(config)
return server, &Vault{c}
}
|
