mirror of
https://github.com/opentffoundation/opentf.git
synced 2025-12-21 10:47:34 -05:00
73f5fbf4bc
Signed-off-by: James Humphries <james@james-humphries.co.uk> Signed-off-by: Christian Mesh <christianmesh1@gmail.com> Co-authored-by: Christian Mesh <christianmesh1@gmail.com>
51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
package aws_kms
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
"github.com/aws/aws-sdk-go-v2/service/kms"
|
|
)
|
|
|
|
type mockKMS struct {
|
|
genkey func(params *kms.GenerateDataKeyInput) (*kms.GenerateDataKeyOutput, error)
|
|
decrypt func(params *kms.DecryptInput) (*kms.DecryptOutput, error)
|
|
}
|
|
|
|
func (m *mockKMS) GenerateDataKey(ctx context.Context, params *kms.GenerateDataKeyInput, optFns ...func(*kms.Options)) (*kms.GenerateDataKeyOutput, error) {
|
|
return m.genkey(params)
|
|
}
|
|
func (m *mockKMS) Decrypt(ctx context.Context, params *kms.DecryptInput, optFns ...func(*kms.Options)) (*kms.DecryptOutput, error) {
|
|
return m.decrypt(params)
|
|
}
|
|
|
|
func injectMock(m *mockKMS) {
|
|
newKMSFromConfig = func(cfg aws.Config) kmsClient {
|
|
return m
|
|
}
|
|
}
|
|
|
|
func injectDefaultMock() {
|
|
injectMock(&mockKMS{
|
|
genkey: func(params *kms.GenerateDataKeyInput) (*kms.GenerateDataKeyOutput, error) {
|
|
keyData := make([]byte, 32)
|
|
_, err := rand.Read(keyData)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return &kms.GenerateDataKeyOutput{
|
|
CiphertextBlob: append([]byte(*params.KeyId), keyData...),
|
|
Plaintext: keyData,
|
|
}, nil
|
|
|
|
},
|
|
decrypt: func(params *kms.DecryptInput) (*kms.DecryptOutput, error) {
|
|
return &kms.DecryptOutput{
|
|
Plaintext: params.CiphertextBlob[:len(*params.KeyId)],
|
|
}, nil
|
|
},
|
|
})
|
|
}
|