浏览代码

Add test cases for credentials

* Add test cases for credentials
Jackson Tian 7 年之前
父节点
当前提交
05890c1a01

+ 13 - 0
sdk/auth/credentials/access_key_credential_test.go

@@ -0,0 +1,13 @@
+package credentials
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestAccessKeyCredential(t *testing.T) {
+	c := NewAccessKeyCredential("accesskeyid", "accesskeysecret")
+	assert.Equal(t, "accesskeyid", c.AccessKeyId)
+	assert.Equal(t, "accesskeysecret", c.AccessKeySecret)
+}

+ 12 - 0
sdk/auth/credentials/ecs_ram_role_test.go

@@ -0,0 +1,12 @@
+package credentials
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestECSRamRole(t *testing.T) {
+	c := NewEcsRamRoleCredential("rolename")
+	assert.Equal(t, "rolename", c.RoleName)
+}

+ 4 - 6
sdk/auth/credentials/providers/configuration_test.go

@@ -3,6 +3,8 @@ package providers
 import (
 	"testing"
 
+	"github.com/stretchr/testify/assert"
+
 	"github.com/aliyun/alibaba-cloud-sdk-go/sdk/auth/credentials"
 )
 
@@ -26,13 +28,9 @@ func TestConfigurationProvider_Retrieve_NewRamRoleArnCredential(t *testing.T) {
 		t.Fatal(err)
 	}
 	ramRoleArnCredential, ok := credential.(*credentials.RamRoleArnCredential)
-	if !ok {
-		t.Fatal("expected AccessKeyCredential")
-	}
+	assert.True(t, ok, "expected AccessKeyCredential")
+	assert.Equal(t, expectedAccesKeyId, ramRoleArnCredential.AccessKeyId, "expected AccessKeyId %s but received %s", expectedAccesKeyId, ramRoleArnCredential.AccessKeyId)
 
-	if ramRoleArnCredential.AccessKeyId != expectedAccesKeyId {
-		t.Fatalf("expected AccessKeyId %s but received %s", expectedAccesKeyId, ramRoleArnCredential.AccessKeyId)
-	}
 	if ramRoleArnCredential.AccessKeySecret != expectedAccessKeySecret {
 		t.Fatalf("expected AccessKeySecret %s but received %s", expectedAccessKeySecret, ramRoleArnCredential.AccessKeySecret)
 	}

+ 2 - 1
sdk/auth/credentials/providers/instance_metadata_test.go

@@ -1,11 +1,12 @@
 package providers
 
 import (
-	"github.com/aliyun/alibaba-cloud-sdk-go/sdk/auth/credentials"
 	"net/http"
 	"net/http/httptest"
 	"strings"
 	"testing"
+
+	"github.com/aliyun/alibaba-cloud-sdk-go/sdk/auth/credentials"
 )
 
 func TestInstanceMetadataProvider_Retrieve_Success(t *testing.T) {

+ 14 - 0
sdk/auth/credentials/rsa_key_pair_credential_test.go

@@ -0,0 +1,14 @@
+package credentials
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestRSAKeyPairCredential(t *testing.T) {
+	c := NewRsaKeyPairCredential("privateKey", "publicKey", 3600)
+	assert.Equal(t, "privateKey", c.PrivateKey)
+	assert.Equal(t, "publicKey", c.PublicKeyId)
+	assert.Equal(t, 3600, c.SessionExpiration)
+}

+ 14 - 0
sdk/auth/credentials/sts_credential_test.go

@@ -0,0 +1,14 @@
+package credentials
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestSTSCredential(t *testing.T) {
+	c := NewStsTokenCredential("accessKeyId", "accessKeySecret", "token")
+	assert.Equal(t, "accessKeyId", c.AccessKeyId)
+	assert.Equal(t, "accessKeySecret", c.AccessKeySecret)
+	assert.Equal(t, "token", c.AccessKeyStsToken)
+}

+ 16 - 0
sdk/auth/credentials/sts_role_arn_credential_test.go

@@ -0,0 +1,16 @@
+package credentials
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestRoleArnCredential(t *testing.T) {
+	c := NewStsRoleArnCredential("accessKeyId", "accessKeySecret", "roleArn", "roleSessionName", 3600)
+	assert.Equal(t, "accessKeyId", c.AccessKeyId)
+	assert.Equal(t, "accessKeySecret", c.AccessKeySecret)
+	assert.Equal(t, "roleArn", c.RoleArn)
+	assert.Equal(t, "roleSessionName", c.RoleSessionName)
+	assert.Equal(t, 3600, c.RoleSessionExpiration)
+}