github.com/opentofu/opentofu@v1.7.1/internal/encryption/keyprovider/gcp_kms/mock_test.go (about)

     1  package gcp_kms
     2  
     3  import (
     4  	"context"
     5  
     6  	"cloud.google.com/go/kms/apiv1/kmspb"
     7  	"github.com/googleapis/gax-go/v2"
     8  	"google.golang.org/api/option"
     9  )
    10  
    11  type mockKMC struct {
    12  	encrypt func(*kmspb.EncryptRequest) (*kmspb.EncryptResponse, error)
    13  	decrypt func(*kmspb.DecryptRequest) (*kmspb.DecryptResponse, error)
    14  }
    15  
    16  func (m *mockKMC) Encrypt(ctx context.Context, req *kmspb.EncryptRequest, opts ...gax.CallOption) (*kmspb.EncryptResponse, error) {
    17  	return m.encrypt(req)
    18  }
    19  func (m *mockKMC) Decrypt(ctx context.Context, req *kmspb.DecryptRequest, opts ...gax.CallOption) (*kmspb.DecryptResponse, error) {
    20  	return m.decrypt(req)
    21  }
    22  
    23  func injectMock(m *mockKMC) {
    24  	newKeyManagementClient = func(ctx context.Context, opts ...option.ClientOption) (keyManagementClient, error) {
    25  		return m, nil
    26  	}
    27  }