1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148
|
package config
import (
"context"
"fmt"
"testing"
"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/credentials"
"github.com/google/go-cmp/cmp"
)
func TestConfigs_SharedConfigOptions(t *testing.T) {
var options LoadOptions
optFns := []func(*LoadOptions) error{
WithSharedConfigProfile("profile-name"),
WithSharedConfigFiles([]string{"creds-file"}),
}
for _, optFn := range optFns {
optFn(&options)
}
_, err := configs{options}.AppendFromLoaders(context.TODO(), []loader{
func(ctx context.Context, configs configs) (Config, error) {
var profile string
var found bool
var files []string
var err error
for _, cfg := range configs {
if p, ok := cfg.(sharedConfigProfileProvider); ok {
profile, found, err = p.getSharedConfigProfile(ctx)
if err != nil || !found {
return nil, err
}
}
if p, ok := cfg.(sharedConfigFilesProvider); ok {
files, found, err = p.getSharedConfigFiles(ctx)
if err != nil || !found {
return nil, err
}
}
}
if e, a := "profile-name", profile; e != a {
t.Errorf("expect %v profile, got %v", e, a)
}
if diff := cmp.Diff([]string{"creds-file"}, files); len(diff) != 0 {
t.Errorf("expect resolved shared config match, got diff: \n %s", diff)
}
return nil, nil
},
})
if err != nil {
t.Fatalf("expect no error, got %v", err)
}
}
func TestConfigs_AppendFromLoaders(t *testing.T) {
var options LoadOptions
err := WithRegion("mock-region")(&options)
if err != nil {
t.Fatalf("expect not error, got %v", err)
}
cfgs, err := configs{}.AppendFromLoaders(
context.TODO(), []loader{
func(ctx context.Context, configs configs) (Config, error) {
if e, a := 0, len(configs); e != a {
t.Errorf("expect %v configs, got %v", e, a)
}
return options, nil
},
})
if err != nil {
t.Fatalf("expect no error, got %v", err)
}
if e, a := 1, len(cfgs); e != a {
t.Errorf("expect %v configs, got %v", e, a)
}
if diff := cmp.Diff(options, cfgs[0]); len(diff) != 0 {
t.Errorf("expect config match, got diff: \n %s", diff)
}
}
func TestConfigs_ResolveAWSConfig(t *testing.T) {
var options LoadOptions
optFns := []func(*LoadOptions) error{
WithRegion("mock-region"),
WithCredentialsProvider(credentials.StaticCredentialsProvider{
Value: aws.Credentials{
AccessKeyID: "AKID", SecretAccessKey: "SECRET",
Source: "provider",
},
}),
}
for _, optFn := range optFns {
optFn(&options)
}
config := configs{options}
cfg, err := config.ResolveAWSConfig(context.TODO(), []awsConfigResolver{
resolveRegion,
resolveCredentials,
})
if err != nil {
t.Fatalf("expect no error, got %v", err)
}
if e, a := "mock-region", cfg.Region; e != a {
t.Errorf("expect %v region, got %v", e, a)
}
creds, err := cfg.Credentials.Retrieve(context.TODO())
if err != nil {
t.Fatalf("expect no error, got %v", err)
}
if e, a := "provider", creds.Source; e != a {
t.Errorf("expect %v provider, got %v", e, a)
}
var expectedSources []interface{}
for _, s := range cfg.ConfigSources {
expectedSources = append(expectedSources, s)
}
if diff := cmp.Diff(expectedSources, cfg.ConfigSources); len(diff) != 0 {
t.Errorf("expect config sources match, got diff: \n %s", diff)
}
}
func TestLoadDefaultConfig(t *testing.T) {
optWithErr := func(_ *LoadOptions) error {
return fmt.Errorf("some error")
}
_, err := LoadDefaultConfig(context.TODO(), optWithErr)
if err == nil {
t.Fatal("expect error when optFn returns error, got nil")
}
}
|