threshold-plugin-aws/plugin/service/storage/state_test.go

// Copyright IBM Corp. 2021, 2026
// SPDX-License-Identifier: MPL-2.0

package storage

import (
	"context"
	"testing"

	"repo.anhonesthost.net/CyberCoveLLC/threshold-plugin-aws/internal/credential"
	"github.com/hashicorp/go-secure-stdlib/awsutil/v2"
	"github.com/stretchr/testify/require"
)

func TestAwsStoragePersistedStateS3Client(t *testing.T) {
	cases := []struct {
		name             string
		creds            *awsutil.CredentialsConfig
		opts             []awsStoragePersistedStateOption
		awsOpts          []awsutil.Option
		s3Opts           []s3Option
		expectedRegion   string
		expectedEndpoint string
	}{
		{
			name: "static credentials",
			creds: func() *awsutil.CredentialsConfig {
				creds, err := awsutil.NewCredentialsConfig(
					awsutil.WithRegion("us-west-2"),
					awsutil.WithAccessKey("foobar"),
					awsutil.WithSecretKey("barfoo"),
				)
				require.NoError(t, err)
				return creds
			}(),
			awsOpts: []awsutil.Option{
				awsutil.WithIAMAPIFunc(
					awsutil.NewMockIAM(),
				),
			},
			opts: []awsStoragePersistedStateOption{
				withTestS3APIFunc(newTestMockS3(nil)),
			},
			s3Opts: []s3Option{
				WithEndpoint("0.0.0.0"),
			},
			expectedRegion:   "us-west-2",
			expectedEndpoint: "0.0.0.0",
		},
	}
	for _, tc := range cases {
		tc := tc
		t.Run(tc.name, func(t *testing.T) {
			require := require.New(t)

			cred, err := credential.NewAwsCredentialPersistedState(
				credential.WithCredentialsConfig(tc.creds),
				credential.WithStateTestOpts(tc.awsOpts),
			)
			require.NoError(err)

			state, err := newAwsStoragePersistedState(
				append(tc.opts, withCredentials(cred))...,
			)
			require.NoError(err)

			clientRaw, err := state.s3Client(context.Background(), tc.s3Opts...)
			require.NoError(err)
			require.NotNil(clientRaw)
			client, ok := clientRaw.(*testMockS3)
			require.True(ok)
			require.NotNil(client)
			require.Equal(tc.expectedRegion, client.Region)
		})
	}
}