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
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package provisionersdk_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/goleak"
|
|
"storj.io/drpc/drpcconn"
|
|
"storj.io/drpc/drpcerr"
|
|
|
|
"github.com/coder/coder/provisionersdk"
|
|
"github.com/coder/coder/provisionersdk/proto"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
goleak.VerifyTestMain(m)
|
|
}
|
|
|
|
func TestProvisionerSDK(t *testing.T) {
|
|
t.Run("Serve", func(t *testing.T) {
|
|
client, server := provisionersdk.TransportPipe()
|
|
defer client.Close()
|
|
defer server.Close()
|
|
|
|
ctx, cancelFunc := context.WithCancel(context.Background())
|
|
defer cancelFunc()
|
|
go func() {
|
|
err := provisionersdk.Serve(ctx, &proto.DRPCProvisionerUnimplementedServer{}, &provisionersdk.ServeOptions{
|
|
Transport: server,
|
|
})
|
|
require.NoError(t, err)
|
|
}()
|
|
|
|
api := proto.NewDRPCProvisionerClient(drpcconn.New(client))
|
|
_, err := api.Parse(context.Background(), &proto.Parse_Request{})
|
|
require.Equal(t, drpcerr.Unimplemented, int(drpcerr.Code(err)))
|
|
})
|
|
t.Run("ServeClosedPipe", func(t *testing.T) {
|
|
client, server := provisionersdk.TransportPipe()
|
|
_ = client.Close()
|
|
_ = server.Close()
|
|
|
|
err := provisionersdk.Serve(context.Background(), &proto.DRPCProvisionerUnimplementedServer{}, &provisionersdk.ServeOptions{
|
|
Transport: server,
|
|
})
|
|
require.NoError(t, err)
|
|
})
|
|
}
|