mirror of
https://github.com/coder/coder.git
synced 2025-07-12 00:14:10 +00:00
7c260f88d1
* feat: Create provisioner abstraction Creates a provisioner abstraction that takes prior art from the Terraform plugin system. It's safe to assume this code will change a lot when it becomes integrated with provisionerd. Closes #10. * Ignore generated files in diff view * Check for unstaged file changes * Install protoc-gen-go * Use proper drpc plugin version * Fix serve closed pipe * Install sqlc with curl for speed * Fix install command * Format CI action * Add linguist-generated and closed pipe test * Cleanup code from comments * Add dRPC comment * Add Terraform installer for cross-platform * Build provisioner tests on Linux only
130 lines
2.8 KiB
Go
130 lines
2.8 KiB
Go
//go:build linux
|
|
|
|
package terraform
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/coder/coder/provisionersdk"
|
|
"github.com/coder/coder/provisionersdk/proto"
|
|
"github.com/hashicorp/go-version"
|
|
"github.com/stretchr/testify/require"
|
|
"storj.io/drpc/drpcconn"
|
|
|
|
"github.com/hashicorp/hc-install/product"
|
|
"github.com/hashicorp/hc-install/releases"
|
|
)
|
|
|
|
func TestProvision(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
installer := &releases.ExactVersion{
|
|
Product: product.Terraform,
|
|
Version: version.Must(version.NewVersion("1.1.2")),
|
|
}
|
|
execPath, err := installer.Install(context.Background())
|
|
require.NoError(t, err)
|
|
|
|
client, server := provisionersdk.TransportPipe()
|
|
ctx, cancelFunc := context.WithCancel(context.Background())
|
|
t.Cleanup(func() {
|
|
_ = client.Close()
|
|
_ = server.Close()
|
|
cancelFunc()
|
|
})
|
|
go func() {
|
|
err := Serve(ctx, &ServeOptions{
|
|
ServeOptions: &provisionersdk.ServeOptions{
|
|
Transport: server,
|
|
},
|
|
BinaryPath: execPath,
|
|
})
|
|
require.NoError(t, err)
|
|
}()
|
|
api := proto.NewDRPCProvisionerClient(drpcconn.New(client))
|
|
|
|
for _, tc := range []struct {
|
|
Name string
|
|
Files map[string]string
|
|
Request *proto.Provision_Request
|
|
Response *proto.Provision_Response
|
|
Error bool
|
|
}{{
|
|
Name: "single-variable",
|
|
Files: map[string]string{
|
|
"main.tf": `variable "A" {
|
|
description = "Testing!"
|
|
}`,
|
|
},
|
|
Request: &proto.Provision_Request{
|
|
ParameterValues: []*proto.ParameterValue{{
|
|
Name: "A",
|
|
Value: "example",
|
|
}},
|
|
},
|
|
Response: &proto.Provision_Response{},
|
|
}, {
|
|
Name: "missing-variable",
|
|
Files: map[string]string{
|
|
"main.tf": `variable "A" {
|
|
}`,
|
|
},
|
|
Error: true,
|
|
}, {
|
|
Name: "single-resource",
|
|
Files: map[string]string{
|
|
"main.tf": `resource "null_resource" "A" {}`,
|
|
},
|
|
Response: &proto.Provision_Response{
|
|
Resources: []*proto.Resource{{
|
|
Name: "A",
|
|
Type: "null_resource",
|
|
}},
|
|
},
|
|
}, {
|
|
Name: "invalid-sourcecode",
|
|
Files: map[string]string{
|
|
"main.tf": `a`,
|
|
},
|
|
Error: true,
|
|
}} {
|
|
tc := tc
|
|
t.Run(tc.Name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
directory := t.TempDir()
|
|
for path, content := range tc.Files {
|
|
err := os.WriteFile(filepath.Join(directory, path), []byte(content), 0644)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
request := &proto.Provision_Request{
|
|
Directory: directory,
|
|
}
|
|
if tc.Request != nil {
|
|
request.ParameterValues = tc.Request.ParameterValues
|
|
request.State = tc.Request.State
|
|
}
|
|
response, err := api.Provision(ctx, request)
|
|
if tc.Error {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
require.NoError(t, err)
|
|
require.Greater(t, len(response.State), 0)
|
|
|
|
resourcesGot, err := json.Marshal(response.Resources)
|
|
require.NoError(t, err)
|
|
|
|
resourcesWant, err := json.Marshal(tc.Response.Resources)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, string(resourcesWant), string(resourcesGot))
|
|
})
|
|
}
|
|
}
|