Start tests for add-profile

Author: Chris Gilmer

cmd/add_profile_test.go

@@ -0,0 +1,58 @@
+package main
+
+import (
+	"log"
+	"os"
+	"testing"
+
+	"github.com/99designs/aws-vault/vault"
+	"github.com/stretchr/testify/assert"
+)
+
+var defaultConfigAddProfile = []byte(`[profile test-id-base]
+region=us-west-2
+output=json
+
+[profile test-id]
+source_profile=test-id-base
+mfa_serial=arn:aws:iam::111111111111:mfa/test-user
+role_arn=arn:aws:iam::222222222222:role/engineer
+region=us-west-2
+output=json
+`)
+
+func TestAddProfile(t *testing.T) {
+
+	// Test logger
+	logger := log.New(os.Stdout, "", log.LstdFlags)
+	logger.SetFlags(0)
+
+	f := newConfigFile(t, defaultConfigAddProfile)
+	defer func() {
+		errRemove := os.Remove(f)
+		assert.NoError(t, errRemove)
+	}()
+
+	config, err := vault.LoadConfig(f)
+	assert.NoError(t, err)
+
+	addProfileConfig := AddProfileConfig{
+		// Config
+		Logger: logger,
+		Config: config,
+
+		// Profile Inputs
+		IAMRole:   "test-role",
+		Region:    "us-west-2",
+		Partition: "aws",
+		Output:    "json",
+
+		// Profiles
+		AWSProfileAccounts: []string{"test-id:123456789012"},
+		AWSProfileName:     "test-id",
+	}
+	err = addProfileConfig.AddProfile()
+	assert.NoError(t, err)
+
+	// TODO: Check contents of file
+}

cmd/setup_test.go

@@ -36,9 +36,12 @@ output=json
 		assert.NoError(t, errRemove)
 	}()
 
-	config, _ := vault.LoadConfig(f)
+	config, err := vault.LoadConfig(f)
+	assert.NoError(t, err)
+
 	keyring, err := getKeyring("test")
 	assert.NoError(t, err)
+
 	setupConfig := SetupConfig{
 		// Config
 		Logger:     logger,
@@ -59,6 +62,8 @@ output=json
 	}
 	err = setupConfig.UpdateAWSConfigFile()
 	assert.NoError(t, err)
+
+	// TODO: Check contents of file
 }
 
 func TestGenerateQrCode(t *testing.T) {