go.mondoo.com/cnquery@v0.0.0-20231005093811-59568235f6ea/providers-sdk/v1/vault/awsparameterstore/parameterstore_test.go (about)

     1  // Copyright (c) Mondoo, Inc.
     2  // SPDX-License-Identifier: BUSL-1.1
     3  
     4  //go:build debugtest
     5  // +build debugtest
     6  
     7  package awsparameterstore
     8  
     9  import (
    10  	"context"
    11  	"os"
    12  	"testing"
    13  
    14  	"github.com/aws/aws-sdk-go-v2/config"
    15  	"github.com/stretchr/testify/assert"
    16  	"github.com/stretchr/testify/require"
    17  	"go.mondoo.com/cnquery/providers-sdk/v1/vault"
    18  )
    19  
    20  func TestAwsParameterStore(t *testing.T) {
    21  	os.Setenv("AWS_PROFILE", "mondoo-dev")
    22  	os.Setenv("AWS_REGION", "us-east-1")
    23  	cfg, err := config.LoadDefaultAWSConfig()
    24  	require.NoError(t, err)
    25  	v := New(cfg)
    26  	ctx := context.Background()
    27  
    28  	key := "mondoo-test-secret-key"
    29  	cred := &vault.Credential{
    30  		Key: key,
    31  		Fields: map[string]string{
    32  			"key":  "value1",
    33  			"key2": "value2",
    34  		},
    35  	}
    36  
    37  	id, err := v.Set(ctx, cred)
    38  	require.NoError(t, err)
    39  
    40  	newCred, err := v.Get(ctx, id)
    41  	require.NoError(t, err)
    42  	assert.Equal(t, cred, newCred)
    43  }