refactor config setters

aliyun · Sep 14, 2024 · e3948a9 · e3948a9
1 parent 1d5d242
commit e3948a9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/credentials/credential.go b/credentials/credential.go
@@ -198,6 +198,11 @@ func (s *Config) SetSTSEndpoint(v string) *Config {
 	return s
 }
 
+func (s *Config) SetExternalId(v string) *Config {
+	s.ExternalId = &v
+	return s
+}
+
 // NewCredential return a credential according to the type in config.
 // if config is nil, the function will use default provider chain to get credentials.
 // please see README.md for detail.

diff --git a/credentials/credential_test.go b/credentials/credential_test.go
@@ -232,6 +232,15 @@ func TestNewCredentialWithRAMRoleARN(t *testing.T) {
 	cred, err = NewCredential(config)
 	assert.Nil(t, err)
 	assert.NotNil(t, cred)
+
+	config.SetExternalId("externalId")
+	config.SetPolicy("policy")
+	config.SetRoleSessionExpiration(3600)
+	config.SetRoleSessionName("roleSessionName")
+	cred, err = NewCredential(config)
+	assert.Nil(t, err)
+	assert.NotNil(t, cred)
+
 }
 
 func TestNewCredentialWithBearerToken(t *testing.T) {