mirror of
https://github.com/hashicorp/vault.git
synced 2025-11-22 03:01:10 +01:00
87 lines
2.2 KiB
Go
87 lines
2.2 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package docker
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/vault/sdk/helper/logging"
|
|
"github.com/hashicorp/vault/sdk/helper/testcluster"
|
|
)
|
|
|
|
type ReplicationDockerOptions struct {
|
|
NumCores int
|
|
ClusterName string
|
|
}
|
|
|
|
func NewReplicationSetDocker(t *testing.T, opt *ReplicationDockerOptions) (*testcluster.ReplicationSet, error) {
|
|
binary := os.Getenv("VAULT_BINARY")
|
|
if binary == "" {
|
|
t.Skip("only running docker test when $VAULT_BINARY present")
|
|
}
|
|
|
|
r := &testcluster.ReplicationSet{
|
|
Clusters: map[string]testcluster.VaultCluster{},
|
|
Logger: logging.NewVaultLogger(hclog.Trace).Named(t.Name()),
|
|
}
|
|
|
|
if opt == nil {
|
|
opt = &ReplicationDockerOptions{}
|
|
}
|
|
|
|
var nc int
|
|
if opt.NumCores > 0 {
|
|
nc = opt.NumCores
|
|
}
|
|
|
|
clusterName := t.Name()
|
|
if opt.ClusterName != "" {
|
|
clusterName = opt.ClusterName
|
|
}
|
|
// clusterName is used for container name as well.
|
|
// A container name should not exceed 64 chars.
|
|
// There are additional chars that are added to the name as well
|
|
// like "-A-core0". So, setting a max limit for a cluster name.
|
|
if len(clusterName) > MaxClusterNameLength {
|
|
return nil, fmt.Errorf("cluster name length exceeded the maximum allowed length of %v", MaxClusterNameLength)
|
|
}
|
|
|
|
r.Builder = func(ctx context.Context, name string, baseLogger hclog.Logger) (testcluster.VaultCluster, error) {
|
|
cluster := NewTestDockerCluster(t, &DockerClusterOptions{
|
|
ImageRepo: "hashicorp/vault",
|
|
ImageTag: "latest",
|
|
VaultBinary: os.Getenv("VAULT_BINARY"),
|
|
ClusterOptions: testcluster.ClusterOptions{
|
|
ClusterName: strings.ReplaceAll(clusterName+"-"+name, "/", "-"),
|
|
Logger: baseLogger.Named(name),
|
|
VaultNodeConfig: &testcluster.VaultNodeConfig{
|
|
LogLevel: "TRACE",
|
|
// If you want the test to run faster locally, you could
|
|
// uncomment this performance_multiplier change.
|
|
//StorageOptions: map[string]string{
|
|
// "performance_multiplier": "1",
|
|
//},
|
|
},
|
|
NumCores: nc,
|
|
},
|
|
CA: r.CA,
|
|
})
|
|
return cluster, nil
|
|
}
|
|
|
|
a, err := r.Builder(context.TODO(), "A", r.Logger)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
r.Clusters["A"] = a
|
|
r.CA = a.(*DockerCluster).CA
|
|
|
|
return r, err
|
|
}
|