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

     1  package awsconfig
     2  
     3  import (
     4  	"os"
     5  	"testing"
     6  
     7  	"github.com/sirupsen/logrus"
     8  
     9  	"github.com/stretchr/testify/assert"
    10  )
    11  
    12  func TestUpdateSamlConfig(t *testing.T) {
    13  	os.Remove(".credentials")
    14  
    15  	logrus.SetLevel(logrus.DebugLevel)
    16  
    17  	sharedCreds := &CredentialsProvider{".credentials", "saml"}
    18  
    19  	exist, err := sharedCreds.CredsExists()
    20  	assert.Nil(t, err)
    21  	assert.True(t, exist)
    22  
    23  	awsCreds := &AWSCredentials{
    24  		AWSAccessKey:     "testid",
    25  		AWSSecretKey:     "testsecret",
    26  		AWSSessionToken:  "testtoken",
    27  		AWSSecurityToken: "testtoken",
    28  	}
    29  
    30  	err = sharedCreds.Save(awsCreds)
    31  	assert.Nil(t, err)
    32  
    33  	awsCreds, err = sharedCreds.Load()
    34  	assert.Nil(t, err)
    35  	assert.Equal(t, "testid", awsCreds.AWSAccessKey)
    36  	assert.Equal(t, "testsecret", awsCreds.AWSSecretKey)
    37  	assert.Equal(t, "testtoken", awsCreds.AWSSessionToken)
    38  
    39  	os.Remove(".credentials")
    40  }