mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-12-11 17:04:22 -05:00
7cada2c9e8
* Run goleak as part of all tests We are already using goleak in various tests. This commit adds a TestMain to all remaining tests and calls goleak.VerifyTestMain in them. * Add goleak to debugd/deploy package and fix bug. * Run go mod tidy * Fix integration tests * Move goleak invocation for mount integration test * Ignore leak in state integration tests Co-authored-by: Fabian Kammel <fk@edgelss.systems>
59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/edgelesssys/constellation/internal/cloud/cloudprovider"
|
|
"github.com/edgelesssys/constellation/internal/config"
|
|
"github.com/edgelesssys/constellation/internal/state"
|
|
"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
|
|
state state.ConstellationState
|
|
createErr error
|
|
}
|
|
|
|
func (c *stubCloudCreator) Create(
|
|
ctx context.Context,
|
|
provider cloudprovider.Provider,
|
|
config *config.Config,
|
|
name, insType string,
|
|
coordCount, nodeCount int,
|
|
) (state.ConstellationState, error) {
|
|
c.createCalled = true
|
|
return c.state, c.createErr
|
|
}
|
|
|
|
type stubCloudTerminator struct {
|
|
called bool
|
|
terminateErr error
|
|
}
|
|
|
|
func (c *stubCloudTerminator) Terminate(context.Context, state.ConstellationState) error {
|
|
c.called = true
|
|
return c.terminateErr
|
|
}
|
|
|
|
func (c *stubCloudTerminator) Called() bool {
|
|
return c.called
|
|
}
|
|
|
|
type stubServiceAccountCreator struct {
|
|
cloudServiceAccountURI string
|
|
createErr error
|
|
}
|
|
|
|
func (c *stubServiceAccountCreator) Create(ctx context.Context, stat state.ConstellationState, config *config.Config) (string, state.ConstellationState, error) {
|
|
return c.cloudServiceAccountURI, stat, c.createErr
|
|
}
|