55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
// +build integration
|
|
|
|
package vault_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/influxdata/influxdb/v2"
|
|
influxdbtesting "github.com/influxdata/influxdb/v2/testing"
|
|
"github.com/influxdata/influxdb/v2/vault"
|
|
testcontainer "github.com/testcontainers/testcontainers-go"
|
|
)
|
|
|
|
func initSecretService(f influxdbtesting.SecretServiceFields, t *testing.T) (influxdb.SecretService, func()) {
|
|
token := "test"
|
|
ctx := context.Background()
|
|
vaultC, err := testcontainer.RunContainer(ctx, "vault", testcontainer.RequestContainer{
|
|
ExportedPort: []string{
|
|
"8200/tcp",
|
|
},
|
|
Cmd: fmt.Sprintf(`vault server -dev -dev-listen-address 0.0.0.0:8200 -dev-root-token-id=%s`, token),
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("failed to initialize vault testcontiner: %v", err)
|
|
}
|
|
ip, port, err := vaultC.GetHostEndpoint(ctx, "8200/tcp")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
s, err := vault.NewSecretService()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
s.Client.SetToken(token)
|
|
s.Client.SetAddress(fmt.Sprintf("http://%v:%v", ip, port))
|
|
|
|
for _, sec := range f.Secrets {
|
|
for k, v := range sec.Env {
|
|
if err := s.PutSecret(ctx, sec.OrganizationID, k, v); err != nil {
|
|
t.Fatalf("failed to populate secrets: %v", err)
|
|
}
|
|
}
|
|
}
|
|
return s, func() {
|
|
defer vaultC.Terminate(ctx, t)
|
|
}
|
|
}
|
|
|
|
func TestSecretService(t *testing.T) {
|
|
influxdbtesting.SecretService(initSecretService, t)
|
|
}
|