AttachUserPolicy_test.go
// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. // SPDX-License-Identifier: MIT-0
package main
import (
"errors"
"testing"
"time"
"github.com/aws/aws-sdk-go/service/iam"
"github.com/aws/aws-sdk-go/service/iam/iamiface"
)
// Define a mock struct to use in unit tests
type mockIAMClient struct {
iamiface.IAMAPI
}
func (m *mockIAMClient) AttachRolePolicy(input *iam.AttachRolePolicyInput) (*iam.AttachRolePolicyOutput, error) {
// Check that required inputs exist
if input.PolicyArn == nil || *input.PolicyArn == "" || input.RoleName == nil || *input.RoleName == "" {
return nil, errors.New("AttacheRolePolicyInput.PolicyArn or AttachRolePolicyInput.RoleName is nil or an empty string")
}
resp := iam.AttachRolePolicyOutput{}
return &resp, nil
}
func TestAttachRolePolicy(t *testing.T) {
thisTime := time.Now()
nowString := thisTime.Format("2006-01-02 15:04:05 Monday")
t.Log("Starting unit test at " + nowString)
// mock resource
roleName := "test-role"
mockSvc := &mockIAMClient{}
err := AttachDynamoFullPolicy(mockSvc, &roleName)
if err != nil {
t.Fatal(err)
}
t.Log("Attached DynamoDB full-access policy to role " + roleName)
}