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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
// Author: mpolden
package cmd
import (
"fmt"
"os"
"path/filepath"
"testing"
"github.com/stretchr/testify/require"
"github.com/stretchr/testify/assert"
"github.com/vespa-engine/vespa/client/go/internal/mock"
"github.com/vespa-engine/vespa/client/go/internal/vespa"
)
func TestCert(t *testing.T) {
t.Run("auth cert", func(t *testing.T) {
testCert(t, []string{"auth", "cert"})
})
}
func configureCloud(t *testing.T, cli *CLI) {
require.Nil(t, cli.Run("config", "set", "application", "t1.a1.i1"))
require.Nil(t, cli.Run("config", "set", "target", "cloud"))
require.Nil(t, cli.Run("auth", "api-key"))
}
func testCert(t *testing.T, subcommand []string) {
appDir, pkgDir := mock.ApplicationPackageDir(t, false, false)
cli, stdout, _ := newTestCLI(t)
configureCloud(t, cli)
stdout.Reset()
args := append(subcommand, pkgDir)
require.Nil(t, cli.Run(args...))
app, err := vespa.ApplicationFromString("t1.a1.i1")
assert.Nil(t, err)
pkgCertificate := filepath.Join(appDir, "security", "clients.pem")
homeDir := cli.config.homeDir
certificate := filepath.Join(homeDir, app.String(), "data-plane-public-cert.pem")
privateKey := filepath.Join(homeDir, app.String(), "data-plane-private-key.pem")
assert.Equal(t, fmt.Sprintf("Success: Certificate written to %s\nSuccess: Private key written to %s\nSuccess: Copied certificate from %s to %s\n", certificate, privateKey, certificate, pkgCertificate), stdout.String())
}
func TestCertCompressedPackage(t *testing.T) {
t.Run("auth cert", func(t *testing.T) {
testCertCompressedPackage(t, []string{"auth", "cert"})
})
}
func testCertCompressedPackage(t *testing.T, subcommand []string) {
_, pkgDir := mock.ApplicationPackageDir(t, true, false)
zipFile := filepath.Join(pkgDir, "target", "application.zip")
err := os.MkdirAll(filepath.Dir(zipFile), 0755)
assert.Nil(t, err)
_, err = os.Create(zipFile)
assert.Nil(t, err)
cli, stdout, stderr := newTestCLI(t)
configureCloud(t, cli)
stdout.Reset()
stderr.Reset()
args := append(subcommand, pkgDir)
err = cli.Run(args...)
assert.NotNil(t, err)
assert.Contains(t, stderr.String(), "Error: cannot add certificate to compressed application package")
err = os.Remove(zipFile)
assert.Nil(t, err)
args = append(subcommand, "-f", pkgDir)
err = cli.Run(args...)
assert.Nil(t, err)
assert.Contains(t, stdout.String(), "Success: Certificate written to")
assert.Contains(t, stdout.String(), "Success: Private key written to")
}
func TestCertAdd(t *testing.T) {
cli, stdout, stderr := newTestCLI(t)
configureCloud(t, cli)
stdout.Reset()
require.Nil(t, cli.Run("auth", "cert", "-N"))
appDir, pkgDir := mock.ApplicationPackageDir(t, false, false)
stdout.Reset()
require.Nil(t, cli.Run("auth", "cert", "add", pkgDir))
pkgCertificate := filepath.Join(appDir, "security", "clients.pem")
homeDir := cli.config.homeDir
certificate := filepath.Join(homeDir, "t1.a1.i1", "data-plane-public-cert.pem")
assert.Equal(t, fmt.Sprintf("Success: Copied certificate from %s to %s\n", certificate, pkgCertificate), stdout.String())
require.NotNil(t, cli.Run("auth", "cert", "add", pkgDir))
assert.Contains(t, stderr.String(), fmt.Sprintf("Error: application package %s already contains a certificate", appDir))
stdout.Reset()
require.Nil(t, cli.Run("auth", "cert", "add", "-f", pkgDir))
assert.Equal(t, fmt.Sprintf("Success: Copied certificate from %s to %s\n", certificate, pkgCertificate), stdout.String())
}
func TestCertNoAdd(t *testing.T) {
cli, stdout, stderr := newTestCLI(t)
configureCloud(t, cli)
stdout.Reset()
require.Nil(t, cli.Run("auth", "cert", "-N"))
homeDir := cli.config.homeDir
app, err := vespa.ApplicationFromString("t1.a1.i1")
assert.Nil(t, err)
certificate := filepath.Join(homeDir, app.String(), "data-plane-public-cert.pem")
privateKey := filepath.Join(homeDir, app.String(), "data-plane-private-key.pem")
assert.Equal(t, fmt.Sprintf("Success: Certificate written to %s\nSuccess: Private key written to %s\n", certificate, privateKey), stdout.String())
require.NotNil(t, cli.Run("auth", "cert", "-N"))
assert.Contains(t, stderr.String(), fmt.Sprintf("Error: private key %s already exists", privateKey))
require.Nil(t, os.Remove(privateKey))
stderr.Reset()
require.NotNil(t, cli.Run("auth", "cert", "-N"))
assert.Contains(t, stderr.String(), fmt.Sprintf("Error: certificate %s already exists", certificate))
stdout.Reset()
require.Nil(t, cli.Run("auth", "cert", "-N", "-f"))
assert.Equal(t, fmt.Sprintf("Success: Certificate written to %s\nSuccess: Private key written to %s\n", certificate, privateKey), stdout.String())
}
|
