profile_writer_test.go

package main

import (
	"fmt"
	"github.com/aws/aws-sdk-go-v2/aws"
	"github.com/aws/aws-sdk-go-v2/service/sts/types"
	"github.com/stretchr/testify/assert"
	"io/ioutil"
	"os"
	"os/user"
	"path"
	"path/filepath"
	"testing"
)

func TestProfileWriter_NewProfileWriterWithDefaults(t *testing.T) {
	pw, err := NewProfileWriter()
	assert.NoError(t, err)
	assert.NotNil(t, pw)

	usr, _ := user.Current()
	awsPath := filepath.Join(usr.HomeDir, ".aws")
	credPath := filepath.Join(awsPath, "credentials")

	assert.Equal(t, awsPath+"/", pw.awsPath)
	assert.Equal(t, credPath, pw.credentialsPath)
}

func TestProfileWriter_NewProfileWriterWithEnvironment(t *testing.T) {
	awsPath := "/tmp/foo/"
	credPath := "/tmp/foo/.aws-secrets"
	os.Setenv("AWS_SHARED_CREDENTIALS_FILE", credPath)
	defer os.Clearenv()

	pw, err := NewProfileWriter()
	assert.NoError(t, err)
	assert.NotNil(t, pw)

	assert.Equal(t, awsPath, pw.awsPath)
	assert.Equal(t, credPath, pw.credentialsPath)
}

func TestProfileWriter_WriteProfile(t *testing.T) {
	credPath := path.Join(os.TempDir(), "swamp-test.ini")
	os.Remove(credPath)

	os.Setenv("AWS_SHARED_CREDENTIALS_FILE", credPath)
	defer os.Clearenv()
	defer os.Remove(credPath)

	profileName := "some-profile"
	region := "some-region"
	creds := &types.Credentials{
		AccessKeyId: aws.String("some-access-key"),
		SecretAccessKey: aws.String("some-secret-access-key"),
		SessionToken: aws.String("some-session-token"),
	}

	pw, _ := NewProfileWriter()
	pw.WriteProfile(creds, profileName, region)

	b, err := ioutil.ReadFile(credPath)
	assert.NoError(t, err)

	content := string(b)

	assert.Regexp(t, `^\[some-profile\]\n.*`, content)
	assertKeyValue(t, "region", "some-region", content)
	assertKeyValue(t, "aws_access_key_id", "some-access-key", content)
	assertKeyValue(t, "aws_secret_access_key", "some-secret-access-key", content)
	assertKeyValue(t, "aws_session_token", "some-session-token", content)
}

func TestProfileWriter_WriteProfileWoRegion(t *testing.T) {
	credPath := path.Join(os.TempDir(), "swamp-test.ini")
	os.Remove(credPath)

	os.Setenv("AWS_SHARED_CREDENTIALS_FILE", credPath)
	defer os.Clearenv()
	defer os.Remove(credPath)

	profileName := "some-profile"
	region := ""
	creds := &types.Credentials{
		AccessKeyId: aws.String("some-access-key"),
		SecretAccessKey: aws.String("some-secret-access-key"),
		SessionToken: aws.String("some-session-token"),
	}

	pw, _ := NewProfileWriter()
	pw.WriteProfile(creds, profileName, region)

	b, err := ioutil.ReadFile(credPath)
	assert.NoError(t, err)

	content := string(b)

	assert.Regexp(t, `^\[some-profile\]\n.*`, content)
	assert.NotRegexp(t, `region`, content)
	assertKeyValue(t, "aws_access_key_id", "some-access-key", content)
	assertKeyValue(t, "aws_secret_access_key", "some-secret-access-key", content)
	assertKeyValue(t, "aws_session_token", "some-session-token", content)
}

func assertKeyValue(t *testing.T, key, value, content string) {
	assert.Regexp(t, fmt.Sprintf(`\n%s\s*=\s*%s\n.*`, key, value), content)
}