constellation/cli/internal/cmd/cloud_test.go
Daniel Weiße 21c80e7bf3 Remove iamid package
Signed-off-by: Daniel Weiße <dw@edgeless.systems>
2023-08-09 15:42:24 +02:00

90 lines
2.1 KiB
Go

/*
Copyright (c) Edgeless Systems GmbH
SPDX-License-Identifier: AGPL-3.0-only
*/
package cmd
import (
"context"
"testing"
"github.com/edgelesssys/constellation/v2/cli/internal/cloudcmd"
"github.com/edgelesssys/constellation/v2/cli/internal/clusterid"
"github.com/edgelesssys/constellation/v2/cli/internal/terraform"
"github.com/edgelesssys/constellation/v2/internal/cloud/cloudprovider"
"github.com/edgelesssys/constellation/v2/internal/cloud/gcpshared"
"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 stubCloudCreator struct {
createCalled bool
id clusterid.File
createErr error
}
func (c *stubCloudCreator) Create(
_ context.Context,
opts cloudcmd.CreateOptions,
) (clusterid.File, error) {
c.createCalled = true
c.id.CloudProvider = opts.Provider
return c.id, c.createErr
}
type stubCloudTerminator struct {
called bool
terminateErr error
}
func (c *stubCloudTerminator) Terminate(_ context.Context, _ string, _ terraform.LogLevel) error {
c.called = true
return c.terminateErr
}
func (c *stubCloudTerminator) Called() bool {
return c.called
}
type stubIAMCreator struct {
createCalled bool
id cloudcmd.IAMOutput
createErr error
}
func (c *stubIAMCreator) Create(
_ context.Context,
provider cloudprovider.Provider,
_ *cloudcmd.IAMConfigOptions,
) (cloudcmd.IAMOutput, error) {
c.createCalled = true
c.id.CloudProvider = provider
return c.id, c.createErr
}
type stubIAMDestroyer struct {
destroyCalled bool
getTfStateKeyCalled bool
gcpSaKey gcpshared.ServiceAccountKey
destroyErr error
getTfStateKeyErr error
}
func (d *stubIAMDestroyer) DestroyIAMConfiguration(_ context.Context, _ string, _ terraform.LogLevel) error {
d.destroyCalled = true
return d.destroyErr
}
func (d *stubIAMDestroyer) GetTfStateServiceAccountKey(_ context.Context, _ string) (gcpshared.ServiceAccountKey, error) {
d.getTfStateKeyCalled = true
return d.gcpSaKey, d.getTfStateKeyErr
}