Started reorganization to integration aws_ssm with checkenv

Author: zomglings

aws_ssm/aws_ssm.go

@@ -0,0 +1,72 @@
+/*
+Based on: https://github.com/awsdocs/aws-doc-sdk-examples/blob/main/gov2/ssm/GetParameter/GetParameterv2.go
+*/
+package aws_ssm
+
+import (
+	"context"
+	"log"
+
+	"github.com/aws/aws-sdk-go-v2/config"
+	"github.com/aws/aws-sdk-go-v2/service/ssm"
+)
+
+// SSMGetParametersAPI and SSMDescribeParametersAPI defines the interface
+// for the GetParameters and DescribeParameters function.
+// We use this interface to test the function using a mocked service
+type SSMGetParametersAPI interface {
+	GetParameters(
+		ctx context.Context,
+		params *ssm.GetParametersInput,
+		optFns ...func(*ssm.Options),
+	) (*ssm.GetParametersOutput, error)
+}
+
+type SSMDescribeParametersAPI interface {
+	DescribeParameters(
+		ctx context.Context,
+		params *ssm.DescribeParametersInput,
+		optFns ...func(*ssm.Options),
+	) (*ssm.DescribeParametersOutput, error)
+}
+
+type AWSSystemsManagerParameterStore interface {
+	GetParameters(
+		ctx context.Context,
+		params *ssm.GetParametersInput,
+		optFns ...func(*ssm.Options),
+	) (*ssm.GetParametersOutput, error)
+
+	DescribeParameters(
+		ctx context.Context,
+		params *ssm.DescribeParametersInput,
+		optFns ...func(*ssm.Options),
+	) (*ssm.DescribeParametersOutput, error)
+}
+
+// ExecGetParameters and ExecDescribeParameters retrieves an AWS Systems Manager string parameter
+// Inputs:
+// 		c: is the context of the method call, which includes the AWS Region
+// 		api: is the interface that defines the method call
+// 		input: defines the input arguments to the service call
+// Output:
+// 		If success, a GetParametersOutput object containing the result of the service call and nil
+// 		Otherwise, nil and an error from the call to GetParameters
+func ExecGetParameters(c context.Context, api SSMGetParametersAPI, input *ssm.GetParametersInput) (*ssm.GetParametersOutput, error) {
+	return api.GetParameters(c, input)
+}
+
+func ExecDescribeParameters(c context.Context, api SSMDescribeParametersAPI, input *ssm.DescribeParametersInput) (*ssm.DescribeParametersOutput, error) {
+	return api.DescribeParameters(c, input)
+}
+
+// Load the Shared AWS Configuration (~/.aws/config)
+func InitAWSClient(ctx context.Context) *ssm.Client {
+	cfg, err := config.LoadDefaultConfig(ctx)
+	if err != nil {
+		log.Fatalln("Failed loading AWS Configuration", err)
+	}
+	client := ssm.NewFromConfig(cfg)
+
+	return client
+}

aws_ssm/aws_ssm_test.go

@@ -0,0 +1,130 @@
+package aws_ssm
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"testing"
+
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/service/ssm"
+	"github.com/aws/aws-sdk-go-v2/service/ssm/types"
+)
+
+// Fill fake output data
+type SSMGetParametersImpl struct{}
+type SSMDescribeParametersImpl struct{}
+
+func (dt SSMGetParametersImpl) GetParameters(
+	ctx context.Context,
+	params *ssm.GetParametersInput,
+	optFns ...func(*ssm.Options),
+) (*ssm.GetParametersOutput, error) {
+	parameters := []types.Parameter{}
+	for _, d := range globalData {
+		parameters = append(parameters, types.Parameter{
+			Name:  aws.String(d.Name),
+			Value: aws.String(d.Value),
+		})
+	}
+	output := &ssm.GetParametersOutput{
+		Parameters: parameters,
+	}
+
+	return output, nil
+}
+func (dt SSMDescribeParametersImpl) DescribeParameters(
+	ctx context.Context,
+	params *ssm.DescribeParametersInput,
+	optFns ...func(*ssm.Options),
+) (*ssm.DescribeParametersOutput, error) {
+
+	// TODO(kompotkot): How to test filters?
+	parameters := []types.ParameterMetadata{
+		{Name: aws.String("/test/dev/t1")},
+		{Name: aws.String("/test/dev/t2")},
+	}
+	output := &ssm.DescribeParametersOutput{
+		Parameters: parameters,
+	}
+
+	return output, nil
+}
+
+type DataTags struct {
+	Product string `json:"Product"`
+}
+
+type Data struct {
+	Name  string     `json:"Name"`
+	Value string     `json:"Value"`
+	Tags  []DataTags `json:"Tags"`
+}
+
+var globalData []Data
+
+var globalParameterKeys []string
+
+func populateData(t *testing.T) error {
+	content, err := ioutil.ReadFile("../data.json")
+	if err != nil {
+		return err
+	}
+
+	contentStr := string(content)
+	err = json.Unmarshal([]byte(contentStr), &globalData)
+	if err != nil {
+		return nil
+	}
+
+	return nil
+}
+
+func TestDescribeParameters(t *testing.T) {
+	err := populateData(t)
+	if err != nil {
+		t.Fatal("Failed to populate data")
+	}
+
+	api := &SSMDescribeParametersImpl{}
+
+	flags := Flags{FilterTags: []FilterTag{{Name: "Product", Value: "test"}}}
+
+	// Test DescribeParameters
+	parameterKeys := FetchKeysOfParameters(
+		context.Background(),
+		*api,
+		flags,
+	)
+	if len(parameterKeys) != 2 {
+		// TODO(kompotkot): Extract length of parameters from data.json
+		t.Logf("Length of parameter keys should be 2, but got %d", len(parameterKeys))
+		t.Fail()
+	}
+
+	for _, p := range parameterKeys {
+		globalParameterKeys = append(globalParameterKeys, p)
+	}
+}
+
+func TestGetParameters(t *testing.T) {
+	parameterKeyChunks := GenerateChunks(globalParameterKeys, 10)
+
+	api := &SSMGetParametersImpl{}
+
+	flags := Flags{Export: false}
+
+	parameters := FetchParameters(
+		context.Background(),
+		*api,
+		parameterKeyChunks,
+		flags,
+	)
+	if len(parameters) != 2 {
+		// TODO(kompotkot): Extract length of parameters from data.json
+		t.Logf("Length of parameters should be 2, but got %d", len(parameters))
+		t.Fail()
+	}
+	fmt.Println(parameters)
+}

aws_ssm/data.go

@@ -0,0 +1,23 @@
+package aws_ssm
+
+// Parameter structure for storing final result from AWS SSM
+type Parameter struct {
+	Name   string
+	Value  string
+	Export string
+}
+
+// Tags for filter defined by user
+type FilterTag struct {
+	Name  string
+	Value string
+}
+
+// Contains command-line flags defined by user
+type Flags struct {
+	Export     bool
+	MaxResults int
+	Outfile    string
+	FilterTags []FilterTag
+	Update     bool
+}