mirror of
https://github.com/opentofu/opentofu.git
synced 2024-12-28 01:41:48 -06:00
4fe9632f09
The main significant change here is that the package name for the proto definition is "tfplugin5", which is important because this name is part of the wire protocol for references to types defined in our package. Along with that, we also move the generated package into "internal" to make it explicit that importing the generated Go package from elsewhere is not the right approach for externally-implemented SDKs, which should instead vendor the proto definition they are using and generate their own stubs to ensure that the wire protocol is the only hard dependency between Terraform Core and plugins. After this is merged, any provider binaries built against our helper/schema package will need to be rebuilt so that they use the new "tfplugin5" package name instead of "proto". In a future commit we will include more elaborate and organized documentation on how an external codebase might make use of our RPC interface definition to implement an SDK, but the primary concern here is to ensure we have the right wire package name before release.
147 lines
3.3 KiB
Go
147 lines
3.3 KiB
Go
package plugin
|
|
|
|
import (
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/google/go-cmp/cmp/cmpopts"
|
|
"github.com/hashicorp/terraform/config/hcl2shim"
|
|
proto "github.com/hashicorp/terraform/internal/tfplugin5"
|
|
"github.com/hashicorp/terraform/provisioners"
|
|
"github.com/zclconf/go-cty/cty"
|
|
|
|
mockproto "github.com/hashicorp/terraform/plugin/mock_proto"
|
|
)
|
|
|
|
var _ provisioners.Interface = (*GRPCProvisioner)(nil)
|
|
|
|
var (
|
|
equateEmpty = cmpopts.EquateEmpty()
|
|
typeComparer = cmp.Comparer(cty.Type.Equals)
|
|
valueComparer = cmp.Comparer(cty.Value.RawEquals)
|
|
)
|
|
|
|
func mockProvisionerClient(t *testing.T) *mockproto.MockProvisionerClient {
|
|
ctrl := gomock.NewController(t)
|
|
client := mockproto.NewMockProvisionerClient(ctrl)
|
|
|
|
// we always need a GetSchema method
|
|
client.EXPECT().GetSchema(
|
|
gomock.Any(),
|
|
gomock.Any(),
|
|
).Return(provisionerProtoSchema(), nil)
|
|
|
|
return client
|
|
}
|
|
|
|
func provisionerProtoSchema() *proto.GetProvisionerSchema_Response {
|
|
return &proto.GetProvisionerSchema_Response{
|
|
Provisioner: &proto.Schema{
|
|
Block: &proto.Schema_Block{
|
|
Attributes: []*proto.Schema_Attribute{
|
|
{
|
|
Name: "attr",
|
|
Type: []byte(`"string"`),
|
|
Required: true,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func TestGRPCProvisioner_GetSchema(t *testing.T) {
|
|
p := &GRPCProvisioner{
|
|
client: mockProvisionerClient(t),
|
|
}
|
|
|
|
resp := p.GetSchema()
|
|
checkDiags(t, resp.Diagnostics)
|
|
}
|
|
|
|
func TestGRPCProvisioner_ValidateProvisionerConfig(t *testing.T) {
|
|
client := mockProvisionerClient(t)
|
|
p := &GRPCProvisioner{
|
|
client: client,
|
|
}
|
|
|
|
client.EXPECT().ValidateProvisionerConfig(
|
|
gomock.Any(),
|
|
gomock.Any(),
|
|
).Return(&proto.ValidateProvisionerConfig_Response{}, nil)
|
|
|
|
cfg := hcl2shim.HCL2ValueFromConfigValue(map[string]interface{}{"attr": "value"})
|
|
resp := p.ValidateProvisionerConfig(provisioners.ValidateProvisionerConfigRequest{Config: cfg})
|
|
checkDiags(t, resp.Diagnostics)
|
|
}
|
|
|
|
func TestGRPCProvisioner_ProvisionResource(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
client := mockproto.NewMockProvisionerClient(ctrl)
|
|
|
|
// we always need a GetSchema method
|
|
client.EXPECT().GetSchema(
|
|
gomock.Any(),
|
|
gomock.Any(),
|
|
).Return(provisionerProtoSchema(), nil)
|
|
|
|
stream := mockproto.NewMockProvisioner_ProvisionResourceClient(ctrl)
|
|
stream.EXPECT().Recv().Return(&proto.ProvisionResource_Response{
|
|
Output: "provisioned",
|
|
}, io.EOF)
|
|
|
|
client.EXPECT().ProvisionResource(
|
|
gomock.Any(),
|
|
gomock.Any(),
|
|
).Return(stream, nil)
|
|
|
|
p := &GRPCProvisioner{
|
|
client: client,
|
|
}
|
|
|
|
rec := &provisionRecorder{}
|
|
|
|
resp := p.ProvisionResource(provisioners.ProvisionResourceRequest{
|
|
Config: cty.ObjectVal(map[string]cty.Value{
|
|
"attr": cty.StringVal("value"),
|
|
}),
|
|
Connection: cty.EmptyObjectVal,
|
|
UIOutput: rec,
|
|
})
|
|
|
|
if resp.Diagnostics.HasErrors() {
|
|
t.Fatal(resp.Diagnostics.Err())
|
|
}
|
|
|
|
if len(rec.output) == 0 || rec.output[0] != "provisioned" {
|
|
t.Fatalf("expected %q, got %q", []string{"provisioned"}, rec.output)
|
|
}
|
|
}
|
|
|
|
type provisionRecorder struct {
|
|
output []string
|
|
}
|
|
|
|
func (r *provisionRecorder) Output(s string) {
|
|
r.output = append(r.output, s)
|
|
}
|
|
|
|
func TestGRPCProvisioner_Stop(t *testing.T) {
|
|
client := mockProvisionerClient(t)
|
|
p := &GRPCProvisioner{
|
|
client: client,
|
|
}
|
|
|
|
client.EXPECT().Stop(
|
|
gomock.Any(),
|
|
gomock.Any(),
|
|
).Return(&proto.Stop_Response{}, nil)
|
|
|
|
err := p.Stop()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|