Validate the output of the config file after it is created

Author: Chris Gilmer

cmd/add_profile_test.go

@@ -36,6 +36,7 @@ output=json
 	config, err := vault.LoadConfig(f)
 	assert.NoError(t, err)
 
+	mfaSerial := "arn:aws:iam::111111111111:mfa/test-user"
 	addProfileConfig := AddProfileConfig{
 		// Config
 		Logger: logger,
@@ -48,11 +49,21 @@ output=json
 		Output:    "json",
 
 		// Profiles
-		AWSProfileAccounts: []string{"test-id:123456789012"},
+		AWSProfileAccounts: []string{"test-id-new:123456789012"},
 		AWSProfileName:     "test-id",
 	}
 	err = addProfileConfig.AddProfile()
 	assert.NoError(t, err)
 
-	// TODO: Check contents of file
+	// re-load the config file
+	config, err = vault.LoadConfig(f)
+	assert.NoError(t, err)
+
+	testSection, ok := config.ProfileSection("test-id-new")
+	assert.True(t, ok)
+	assert.Equal(t, testSection.SourceProfile, "test-id-base")
+	assert.Equal(t, testSection.MfaSerial, mfaSerial)
+	assert.Equal(t, testSection.RoleARN, "arn:aws:iam::123456789012:role/test-role")
+	assert.Equal(t, testSection.Region, "us-west-2")
+	// assert.Equal(t, testBaseSection.Output, "json")
 }

cmd/setup_test.go

@@ -44,6 +44,7 @@ output=json
 	keyring, err := getKeyring("test")
 	assert.NoError(t, err)
 
+	mfaSerial := "arn:aws:iam::111111111111:mfa/test-user"
 	setupConfig := SetupConfig{
 		// Config
 		Logger:     logger,
@@ -61,11 +62,28 @@ output=json
 		// Profiles
 		BaseProfileName:    "test-id-base",
 		AWSProfileAccounts: []string{"test-id:123456789012"},
+		MFASerial:          mfaSerial,
 	}
 	err = setupConfig.UpdateAWSConfigFile()
 	assert.NoError(t, err)
 
-	// TODO: Check contents of file
+	// re-load the config file
+	config, err = vault.LoadConfig(f)
+	assert.NoError(t, err)
+
+	testBaseSection, ok := config.ProfileSection("test-id-base")
+	assert.True(t, ok)
+	assert.Equal(t, len(testBaseSection.MfaSerial), 0)
+	assert.Equal(t, testBaseSection.Region, "us-west-2")
+	// assert.Equal(t, testBaseSection.Output, "json")
+
+	testSection, ok := config.ProfileSection("test-id")
+	assert.True(t, ok)
+	assert.Equal(t, testSection.SourceProfile, "test-id-base")
+	assert.Equal(t, testSection.MfaSerial, mfaSerial)
+	assert.Equal(t, testSection.RoleARN, "arn:aws:iam::123456789012:role/test-role")
+	assert.Equal(t, testSection.Region, "us-west-2")
+	// assert.Equal(t, testBaseSection.Output, "json")
 }
 
 func TestGenerateQrCode(t *testing.T) {