83 lines
2.7 KiB
Go
83 lines
2.7 KiB
Go
|
package provisioning
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
capi "github.com/hashicorp/consul/api"
|
||
|
|
napi "github.com/hashicorp/nomad/api"
|
||
|
|
vapi "github.com/hashicorp/vault/api"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewProvisioner(config ProvisionerConfig) Provisioner {
|
||
|
|
if config.IsLocal {
|
||
|
|
return DefaultProvisioner
|
||
|
|
}
|
||
|
|
if config.VagrantBox != "" {
|
||
|
|
return PreProvisioner(ProvisionerConfigVagrant(config))
|
||
|
|
}
|
||
|
|
if config.TerraformConfig != "" {
|
||
|
|
return PreProvisioner(ProvisionerConfigTerraform(config))
|
||
|
|
}
|
||
|
|
return DefaultProvisioner
|
||
|
|
}
|
||
|
|
|
||
|
|
// ProvisionerConfig defines options for the entire lifecycle of the provisioner.
|
||
|
|
type ProvisionerConfig struct {
|
||
|
|
IsLocal bool
|
||
|
|
VagrantBox string
|
||
|
|
TerraformConfig string
|
||
|
|
|
||
|
|
NomadSha string
|
||
|
|
NomadVersion string
|
||
|
|
NomadLocalBinary string
|
||
|
|
}
|
||
|
|
|
||
|
|
// Provisioner interface is used by the test framework to provision Nomad.
|
||
|
|
// The Setup* methods should be used to create a Nomad cluster at the
|
||
|
|
// appropriate stage. The returned ClusterInfo handle helps TestCases
|
||
|
|
// isolate test state by using the ClusterInfo.ID as part of job IDs.
|
||
|
|
type Provisioner interface {
|
||
|
|
// SetupTestRun is called at the start of the entire test run.
|
||
|
|
SetupTestRun(t *testing.T, opts SetupOptions) (*ClusterInfo, error)
|
||
|
|
|
||
|
|
// SetupTestSuite is called at the start of each TestSuite.
|
||
|
|
// TODO: no current provisioner implementation uses this, but we
|
||
|
|
// could use it to provide each TestSuite with an entirely separate
|
||
|
|
// Nomad cluster.
|
||
|
|
SetupTestSuite(t *testing.T, opts SetupOptions) (*ClusterInfo, error)
|
||
|
|
|
||
|
|
// SetupTestCase is called at the start of each TestCase in every TestSuite.
|
||
|
|
SetupTestCase(t *testing.T, opts SetupOptions) (*ClusterInfo, error)
|
||
|
|
|
||
|
|
// TODO: no current provisioner implementation uses any of these,
|
||
|
|
// but it's the obvious need if we setup/teardown after each TestSuite
|
||
|
|
// or TestCase.
|
||
|
|
|
||
|
|
// TearDownTestCase is called after each TestCase in every TestSuite.
|
||
|
|
TearDownTestCase(t *testing.T, clusterID string) error
|
||
|
|
|
||
|
|
// TearDownTestSuite is called after every TestSuite.
|
||
|
|
TearDownTestSuite(t *testing.T, clusterID string) error
|
||
|
|
|
||
|
|
// TearDownTestRun is called at the end of the entire test run.
|
||
|
|
TearDownTestRun(t *testing.T, clusterID string) error
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetupOptions defines options to be given to the Provisioner when
|
||
|
|
// calling Setup* methods.
|
||
|
|
type SetupOptions struct {
|
||
|
|
Name string
|
||
|
|
ExpectConsul bool // If true, fails if a Consul client can't be configured
|
||
|
|
ExpectVault bool // If true, fails if a Vault client can't be configured
|
||
|
|
}
|
||
|
|
|
||
|
|
// ClusterInfo is a handle to a provisioned cluster, along with clients
|
||
|
|
// a test run can use to connect to the cluster.
|
||
|
|
type ClusterInfo struct {
|
||
|
|
ID string
|
||
|
|
Name string
|
||
|
|
NomadClient *napi.Client
|
||
|
|
ConsulClient *capi.Client
|
||
|
|
VaultClient *vapi.Client
|
||
|
|
}
|