Skip to content

Commit f00c5c4

Browse files
Adding to test for multiple receivers with single secret
Signed-off-by: Anand Rajagopal <[email protected]>
1 parent fa700a9 commit f00c5c4

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed

secrets/providers/aws_secrets_manager_test.go

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -428,6 +428,37 @@ func TestRegisterSecret(t *testing.T) {
428428
provider.Stop()
429429
}
430430

431+
func TestRegisterMultipleSecretsNoRefreshInterval(t *testing.T) {
432+
reg := prometheus.NewPedanticRegistry()
433+
ctx, cancel := context.WithCancel(context.Background())
434+
options := secrets.SecretProviderOptions{
435+
Logger: promslog.NewNopLogger(),
436+
Registerer: reg,
437+
Context: ctx,
438+
}
439+
provider := NewAWSSecretsManagerProvider(options)
440+
secretOne := secrets.GenericSecret{
441+
AWSSecretsManagerConfig: secrets.AWSSecretsManagerConfig{
442+
SecretARN: "arn:aws:secretsmanager:us-west-2:123456789:secret:receiver-pager-duty",
443+
SecretKey: "key1",
444+
},
445+
}
446+
secretOneCopy := secrets.GenericSecret{
447+
AWSSecretsManagerConfig: secrets.AWSSecretsManagerConfig{
448+
SecretARN: "arn:aws:secretsmanager:us-west-2:123456789:secret:receiver-pager-duty",
449+
SecretKey: "key2",
450+
},
451+
}
452+
provider.Register(secretOne)
453+
require.Equal(t, 1, provider.fetchersCount())
454+
455+
provider.Register(secretOneCopy)
456+
require.Equal(t, 1, provider.fetchersCount())
457+
458+
cancel()
459+
provider.Stop()
460+
}
461+
431462
type MockSecretsManagerClient struct {
432463
secretsmanager.Client
433464
secrets map[string]string

0 commit comments

Comments
 (0)