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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package cmd
import (
"bytes"
"fmt"
"net/http"
"path/filepath"
"strings"
"testing"
"time"
"github.com/vespa-engine/vespa/client/go/internal/cli/auth/auth0"
"github.com/vespa-engine/vespa/client/go/internal/mock"
"github.com/vespa-engine/vespa/client/go/internal/util"
"github.com/vespa-engine/vespa/client/go/internal/vespa"
)
func newTestCLI(t *testing.T, envVars ...string) (*CLI, *bytes.Buffer, *bytes.Buffer) {
t.Helper()
homeDir := filepath.Join(t.TempDir(), ".vespa")
cacheDir := filepath.Join(t.TempDir(), ".cache", "vespa")
env := []string{"VESPA_CLI_HOME=" + homeDir, "VESPA_CLI_CACHE_DIR=" + cacheDir}
env = append(env, envVars...)
var (
stdout bytes.Buffer
stderr bytes.Buffer
)
cli, err := New(&stdout, &stderr, env)
if err != nil {
t.Fatal(err)
}
httpClient := &mock.HTTPClient{}
cli.httpClientFactory = func(timeout time.Duration) util.HTTPClient { return httpClient }
cli.httpClient = httpClient
cli.exec = &mock.Exec{}
cli.auth0Factory = func(httpClient util.HTTPClient, options auth0.Options) (vespa.Authenticator, error) {
return &mockAuthenticator{}, nil
}
cli.ztsFactory = func(httpClient util.HTTPClient, domain, url string) (vespa.Authenticator, error) {
return &mockAuthenticator{}, nil
}
return cli, &stdout, &stderr
}
func mockServiceStatus(client *mock.HTTPClient, clusterNames ...string) {
var serviceObjects []string
for _, name := range clusterNames {
service := fmt.Sprintf(`{
"clusterName": "%s",
"host": "localhost",
"port": 8080,
"type": "container",
"url": "http://localhost:19071/application/v2/tenant/default/application/default/environment/prod/region/default/instance/default/serviceconverge/localhost:8080",
"currentGeneration": 1
}
`, name)
serviceObjects = append(serviceObjects, service)
}
services := "[]"
if len(serviceObjects) > 0 {
services = "[" + strings.Join(serviceObjects, ",") + "]"
}
response := fmt.Sprintf(`
{
"services": %s,
"currentGeneration": 1
}`, services)
client.NextResponse(mock.HTTPResponse{
URI: "/application/v2/tenant/default/application/default/environment/prod/region/default/instance/default/serviceconverge",
Status: 200,
Body: []byte(response),
})
}
type mockAuthenticator struct{}
func (a *mockAuthenticator) Authenticate(request *http.Request) error { return nil }
|
