mirror of
https://github.com/golang/go
synced 2024-11-05 22:46:12 -07:00
434f7a8fef
The previous implementation was exposing the details of the wire format and resulted in non idomatic go, detecting the presence of absence of values in fields to deterimine the message type. Now the messages are distinct types and we use type switches instead. Request still exists as an interface to expose the shared behaviour of Call and Notification, as this is the type accepted by handlers. The set of messages is deliberately closed by using a private methods on the interfaces. Change-Id: I2cf15ee3923ef4688670c62896f81f760c77fe04 Reviewed-on: https://go-review.googlesource.com/c/tools/+/228719 Run-TryBot: Ian Cottrell <iancottrell@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Robert Findley <rfindley@google.com>
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
// Copyright 2020 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package servertest
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"golang.org/x/tools/internal/jsonrpc2"
|
|
)
|
|
|
|
type msg struct {
|
|
Msg string
|
|
}
|
|
|
|
func fakeHandler(ctx context.Context, reply jsonrpc2.Replier, req jsonrpc2.Request) error {
|
|
return reply(ctx, &msg{"pong"}, nil)
|
|
}
|
|
|
|
func TestTestServer(t *testing.T) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
server := jsonrpc2.HandlerServer(fakeHandler)
|
|
tcpTS := NewTCPServer(ctx, server)
|
|
defer tcpTS.Close()
|
|
pipeTS := NewPipeServer(ctx, server)
|
|
defer pipeTS.Close()
|
|
|
|
tests := []struct {
|
|
name string
|
|
connector Connector
|
|
}{
|
|
{"tcp", tcpTS},
|
|
{"pipe", pipeTS},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
conn := test.connector.Connect(ctx)
|
|
go conn.Run(ctx, jsonrpc2.MethodNotFound)
|
|
var got msg
|
|
if _, err := conn.Call(ctx, "ping", &msg{"ping"}, &got); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if want := "pong"; got.Msg != want {
|
|
t.Errorf("conn.Call(...): returned %q, want %q", got, want)
|
|
}
|
|
})
|
|
}
|
|
}
|