provision_test.go

package sparta

import (
	"context"
	"testing"

	gocf "github.com/mweagle/go-cloudformation"
	"github.com/pkg/errors"
	"github.com/sirupsen/logrus"
)

type cloudFormationProvisionTestResource struct {
	gocf.CloudFormationCustomResource
	ServiceToken string
	TestKey      interface{}
}

func customResourceTestProvider(resourceType string) gocf.ResourceProperties {
	switch resourceType {
	case "Custom::ProvisionTestEmpty":
		{
			return &cloudFormationProvisionTestResource{}
		}
	default:
		return nil
	}
}

func init() {
	gocf.RegisterCustomResourceProvider(customResourceTestProvider)
}

func TestProvision(t *testing.T) {
	testProvision(t, testLambdaData(), nil)
}

func templateDecorator(ctx context.Context,
	serviceName string,
	lambdaResourceName string,
	lambdaResource gocf.LambdaFunction,
	resourceMetadata map[string]interface{},
	lambdaFunctionCode *gocf.LambdaFunctionCode,
	buildID string,
	cfTemplate *gocf.Template,
	logger *logrus.Logger) (context.Context, error) {

	// Add an empty resource
	newResource, err := newCloudFormationResource("Custom::ProvisionTestEmpty", logger)
	if nil != err {
		return ctx, errors.Wrapf(err, "Failed to create test resource")
	}
	customResource := newResource.(*cloudFormationProvisionTestResource)
	customResource.ServiceToken = "arn:aws:sns:us-east-1:84969EXAMPLE:CRTest"
	customResource.TestKey = "Hello World"
	cfTemplate.AddResource("ProvisionTestResource", customResource)

	// Add an output
	cfTemplate.Outputs["OutputDecorationTest"] = &gocf.Output{
		Description: "Information about the value",
		Value:       gocf.String("My key"),
	}
	return ctx, nil
}

func TestDecorateProvision(t *testing.T) {

	lambdas := testLambdaData()
	lambdas[0].Decorator = templateDecorator
	testProvision(t, lambdas, nil)
}