github.com/versent/saml2aws@v2.17.0+incompatible/pkg/shell/env_test.go (about)

     1  package shell
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/assert"
     7  	"github.com/versent/saml2aws/pkg/awsconfig"
     8  	"github.com/versent/saml2aws/pkg/cfg"
     9  	"github.com/versent/saml2aws/pkg/flags"
    10  )
    11  
    12  func TestBuildEnvVars(t *testing.T) {
    13  
    14  	expectedArray := []string{
    15  		"AWS_SESSION_TOKEN=567",
    16  		"AWS_SECURITY_TOKEN=567",
    17  		"EC2_SECURITY_TOKEN=567",
    18  		"AWS_ACCESS_KEY_ID=123",
    19  		"AWS_SECRET_ACCESS_KEY=345",
    20  		"AWS_PROFILE=saml",
    21  		"AWS_DEFAULT_PROFILE=saml",
    22  	}
    23  
    24  	account := &cfg.IDPAccount{
    25  		Profile: "saml",
    26  	}
    27  
    28  	awsCreds := &awsconfig.AWSCredentials{
    29  		AWSAccessKey:     "123",
    30  		AWSSecretKey:     "345",
    31  		AWSSecurityToken: "567",
    32  		AWSSessionToken:  "567",
    33  	}
    34  
    35  	flags := new(flags.LoginExecFlags)
    36  
    37  	assert.Equal(t, expectedArray, BuildEnvVars(awsCreds, account, flags))
    38  }