1 package credentials
2
3 import (
4 "os"
5 "testing"
6
7 "github.com/alibabacloud-go/tea/tea"
8 "github.com/stretchr/testify/assert"
9 )
10
11 func TestInstanceCredentialsProvider(t *testing.T) {
12 p := newInstanceCredentialsProvider()
13 originEcsMetadata := os.Getenv(ENVEcsMetadata)
14 os.Setenv(ENVEcsMetadata, "")
15 defer func() {
16 os.Setenv(ENVEcsMetadata, originEcsMetadata)
17 }()
18 c, err := p.resolve()
19 assert.NotNil(t, c)
20 assert.Nil(t, err)
21
22 os.Setenv(ENVEcsMetadata, "role_name")
23 c, err = p.resolve()
24 assert.Nil(t, err)
25 assert.Equal(t, "role_name", tea.StringValue(c.RoleName))
26 assert.Equal(t, "ecs_ram_role", tea.StringValue(c.Type))
27
28 os.Unsetenv(ENVEcsMetadata)
29 c, err = p.resolve()
30 assert.Nil(t, c)
31 assert.Nil(t, err)
32 }
33
View as plain text