awsprofile_test.go

package awsprofile_test

import (
	"errors"
	"fmt"
	"log"
	"os"
	"testing"

	"github.com/youyo/awsprofile"
)

func ExampleAwsProfile_ProfileNames() {
	// if you use non-default configuration file path
	if err := os.Setenv("AWS_SHARED_CREDENTIALS_FILE", "./tests/.aws/credentials"); err != nil {
		log.Fatal(err)
	}
	if err := os.Setenv("AWS_CONFIG_FILE", "./tests/.aws/config"); err != nil {
		log.Fatal(err)
	}

	// New
	awsProfile := awsprofile.New()

	// Parse AWS_SHARED_CREDENTIALS_FILE and AWS_CONFIG_FILE
	if err := awsProfile.Parse(); err != nil {
		log.Fatal(err)
	}

	// Get profile names
	profiles, err := awsProfile.ProfileNames()
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println(profiles)
	// Output: [default foo foobar bar barbar]
}

func ExampleAwsProfile_GetCredentials() {
	// if you use non-default configuration file path
	if err := os.Setenv("AWS_SHARED_CREDENTIALS_FILE", "./tests/.aws/credentials"); err != nil {
		log.Fatal(err)
	}
	if err := os.Setenv("AWS_CONFIG_FILE", "./tests/.aws/config"); err != nil {
		log.Fatal(err)
	}

	// New
	awsProfile := awsprofile.New()

	// Parse AWS_SHARED_CREDENTIALS_FILE and AWS_CONFIG_FILE
	if err := awsProfile.Parse(); err != nil {
		log.Fatal(err)
	}

	credentials := awsProfile.GetCredentials()

	profiles, err := credentials.ProfileNames()
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println(profiles)
	// Output: [default foo foobar]
}

func ExampleAwsProfile_GetConfigs() {
	// if you use non-default configuration file path
	if err := os.Setenv("AWS_SHARED_CREDENTIALS_FILE", "./tests/.aws/credentials"); err != nil {
		log.Fatal(err)
	}
	if err := os.Setenv("AWS_CONFIG_FILE", "./tests/.aws/config"); err != nil {
		log.Fatal(err)
	}

	// New
	awsProfile := awsprofile.New()

	// Parse AWS_SHARED_CREDENTIALS_FILE and AWS_CONFIG_FILE
	if err := awsProfile.Parse(); err != nil {
		log.Fatal(err)
	}

	configs := awsProfile.GetConfigs()

	profiles, err := configs.ProfileNames()
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println(profiles)
	// Output: [default bar barbar]
}

func TestAwsProfile_Parse(t *testing.T) {
	// if you use non-default configuration file path
	if err := os.Setenv("AWS_SHARED_CREDENTIALS_FILE", "./tests/.aws/credentials"); err != nil {
		log.Fatal(err)
	}
	if err := os.Setenv("AWS_CONFIG_FILE", "./tests/.aws/config"); err != nil {
		log.Fatal(err)
	}

	// New
	awsProfile := awsprofile.New()

	// Parse AWS_SHARED_CREDENTIALS_FILE and AWS_CONFIG_FILE
	if err := awsProfile.Parse(); err != nil {
		log.Fatal(err)
	}
}

func TestAwsProfile_IsCredential(t *testing.T) {
	os.Setenv("AWS_SHARED_CREDENTIALS_FILE", "./tests/.aws/credentials")
	os.Setenv("AWS_CONFIG_FILE", "./tests/.aws/config")

	awsProfile := awsprofile.New()
	awsProfile.Parse()

	ok, _ := awsProfile.IsCredential("fooooo")
	if ok {
		log.Fatal(errors.New("Unexpected"))
	}

	ok, cred := awsProfile.IsCredential("foo")
	if !ok {
		log.Fatal(errors.New("Unmatched profile"))
	}

	if ok {
		if cred.GetAwsAccessKeyID() != "ACCESS-2-XXXXXXXXXXXXX" {
			t.Fatal(errors.New("Unmatched AwsAccessKeyID"))
		}
	}
}

func TestAwsProfile_IsConfig(t *testing.T) {
	os.Setenv("AWS_SHARED_CREDENTIALS_FILE", "./tests/.aws/credentials")
	os.Setenv("AWS_CONFIG_FILE", "./tests/.aws/config")

	awsProfile := awsprofile.New()
	awsProfile.Parse()

	ok, _ := awsProfile.IsCredential("fooooo")
	if ok {
		log.Fatal(errors.New("Unexpected"))
	}

	ok, config := awsProfile.IsConfig("bar")
	if !ok {
		log.Fatal(errors.New("Unmatched profile"))
	}

	if ok {
		if config.GetRoleArn() != "arn:aws:iam::xxxxxxxxxxxx:role/bar" {
			t.Fatal(errors.New("Unmatched RoleArn"))
		}
	}
}