sts_role_arn_credential_test.go 1.5 KB

1234567891011121314151617181920212223242526272829303132333435
  1. package credentials
  2. import (
  3. "testing"
  4. "github.com/stretchr/testify/assert"
  5. )
  6. func TestRoleArnCredential(t *testing.T) {
  7. c := NewRamRoleArnCredential("accessKeyId", "accessKeySecret", "roleArn", "roleSessionName", 3600)
  8. assert.Equal(t, "accessKeyId", c.AccessKeyId)
  9. assert.Equal(t, "accessKeySecret", c.AccessKeySecret)
  10. assert.Equal(t, "roleArn", c.RoleArn)
  11. assert.Equal(t, "roleSessionName", c.RoleSessionName)
  12. assert.Equal(t, 3600, c.RoleSessionExpiration)
  13. s := NewStsRoleArnCredential("accessKeyId", "accessKeySecret", "roleArn", "roleSessionName", 3600)
  14. assert.Equal(t, "accessKeyId", s.AccessKeyId)
  15. assert.Equal(t, "accessKeySecret", s.AccessKeySecret)
  16. assert.Equal(t, "roleArn", s.RoleArn)
  17. assert.Equal(t, "roleSessionName", s.RoleSessionName)
  18. assert.Equal(t, 3600, s.RoleSessionExpiration)
  19. r := s.ToRamRoleArnCredential()
  20. assert.Equal(t, "accessKeyId", r.AccessKeyId)
  21. assert.Equal(t, "accessKeySecret", r.AccessKeySecret)
  22. assert.Equal(t, "roleArn", r.RoleArn)
  23. assert.Equal(t, "roleSessionName", r.RoleSessionName)
  24. assert.Equal(t, 3600, r.RoleSessionExpiration)
  25. p := NewRamRoleArnWithPolicyCredential("accessKeyId", "accessKeySecret", "roleArn", "roleSessionName", "test", 3600)
  26. assert.Equal(t, "accessKeyId", p.AccessKeyId)
  27. assert.Equal(t, "accessKeySecret", p.AccessKeySecret)
  28. assert.Equal(t, "roleArn", p.RoleArn)
  29. assert.Equal(t, "test", p.Policy)
  30. assert.Equal(t, "roleSessionName", p.RoleSessionName)
  31. assert.Equal(t, 3600, p.RoleSessionExpiration)
  32. }