mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-10-01 01:36:09 -04:00
75 lines
1.7 KiB
Go
75 lines
1.7 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
package cloudcmd
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/edgelesssys/constellation/v2/cli/internal/terraform"
|
|
"github.com/edgelesssys/constellation/v2/internal/cloud/cloudprovider"
|
|
|
|
"go.uber.org/goleak"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
goleak.VerifyTestMain(m,
|
|
// https://github.com/census-instrumentation/opencensus-go/issues/1262
|
|
goleak.IgnoreTopFunction("go.opencensus.io/stats/view.(*worker).start"),
|
|
)
|
|
}
|
|
|
|
type stubTerraformClient struct {
|
|
ip string
|
|
cleanUpWorkspaceCalled bool
|
|
removeInstallerCalled bool
|
|
destroyClusterCalled bool
|
|
createClusterErr error
|
|
destroyClusterErr error
|
|
prepareWorkspaceErr error
|
|
cleanUpWorkspaceErr error
|
|
}
|
|
|
|
func (c *stubTerraformClient) CreateCluster(ctx context.Context) (string, error) {
|
|
return c.ip, c.createClusterErr
|
|
}
|
|
|
|
func (c *stubTerraformClient) PrepareWorkspace(provider cloudprovider.Provider, input terraform.Variables) error {
|
|
return c.prepareWorkspaceErr
|
|
}
|
|
|
|
func (c *stubTerraformClient) DestroyCluster(ctx context.Context) error {
|
|
c.destroyClusterCalled = true
|
|
return c.destroyClusterErr
|
|
}
|
|
|
|
func (c *stubTerraformClient) CleanUpWorkspace() error {
|
|
c.cleanUpWorkspaceCalled = true
|
|
return c.cleanUpWorkspaceErr
|
|
}
|
|
|
|
func (c *stubTerraformClient) RemoveInstaller() {
|
|
c.removeInstallerCalled = true
|
|
}
|
|
|
|
type stubLibvirtRunner struct {
|
|
startCalled bool
|
|
stopCalled bool
|
|
startErr error
|
|
stopErr error
|
|
}
|
|
|
|
func (r *stubLibvirtRunner) Start(_ context.Context, _, _ string) error {
|
|
r.startCalled = true
|
|
return r.startErr
|
|
}
|
|
|
|
func (r *stubLibvirtRunner) Stop(context.Context) error {
|
|
r.stopCalled = true
|
|
return r.stopErr
|
|
}
|