mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
fix: don't use yamux for in-memory provisioner{,d} streams (#5136)
This commit is contained in:
@@ -23,7 +23,7 @@ func TestEcho(t *testing.T) {
|
||||
|
||||
fs := afero.NewMemMapFs()
|
||||
// Create an in-memory provisioner to communicate with.
|
||||
client, server := provisionersdk.TransportPipe()
|
||||
client, server := provisionersdk.MemTransportPipe()
|
||||
ctx, cancelFunc := context.WithCancel(context.Background())
|
||||
t.Cleanup(func() {
|
||||
_ = client.Close()
|
||||
@@ -36,7 +36,7 @@ func TestEcho(t *testing.T) {
|
||||
})
|
||||
assert.NoError(t, err)
|
||||
}()
|
||||
api := proto.NewDRPCProvisionerClient(provisionersdk.Conn(client))
|
||||
api := proto.NewDRPCProvisionerClient(client)
|
||||
|
||||
t.Run("Parse", func(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
@@ -36,7 +36,7 @@ func setupProvisioner(t *testing.T, opts *provisionerServeOptions) (context.Cont
|
||||
opts = &provisionerServeOptions{}
|
||||
}
|
||||
cachePath := t.TempDir()
|
||||
client, server := provisionersdk.TransportPipe()
|
||||
client, server := provisionersdk.MemTransportPipe()
|
||||
ctx, cancelFunc := context.WithCancel(context.Background())
|
||||
serverErr := make(chan error, 1)
|
||||
t.Cleanup(func() {
|
||||
@@ -59,7 +59,7 @@ func setupProvisioner(t *testing.T, opts *provisionerServeOptions) (context.Cont
|
||||
ExitTimeout: opts.exitTimeout,
|
||||
})
|
||||
}()
|
||||
api := proto.NewDRPCProvisionerClient(provisionersdk.Conn(client))
|
||||
api := proto.NewDRPCProvisionerClient(client)
|
||||
return ctx, api
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user