runtimes/google/ipc/signature_test.go

package ipc_test

import (
	"fmt"
	"reflect"
	"testing"

	"veyron.io/veyron/veyron2"
	"veyron.io/veyron/veyron2/ipc"
	"veyron.io/veyron/veyron2/ipc/reserved"
	"veyron.io/veyron/veyron2/naming"
	"veyron.io/veyron/veyron2/rt"
	"veyron.io/veyron/veyron2/vdl"
	"veyron.io/veyron/veyron2/vdl/vdlroot/src/signature"

	"veyron.io/veyron/veyron/lib/testutil"
	"veyron.io/veyron/veyron/profiles"
)

func init() { testutil.Init() }

func startSigServer(runtime veyron2.Runtime, sig sigImpl) (string, func(), error) {
	server, err := runtime.NewServer()
	if err != nil {
		return "", nil, fmt.Errorf("failed to start sig server: %v", err)
	}
	ep, err := server.Listen(profiles.LocalListenSpec)
	if err != nil {
		return "", nil, fmt.Errorf("failed to listen: %v", err)
	}
	if err := server.Serve("", sig, nil); err != nil {
		return "", nil, err
	}
	return ep.String(), func() { server.Stop() }, nil
}

type sigImpl struct{}

func (sigImpl) NonStreaming0(ipc.ServerContext)                          { panic("X") }
func (sigImpl) NonStreaming1(_ ipc.ServerContext, _ string) error        { panic("X") }
func (sigImpl) Streaming0(_ *streamStringBool)                           { panic("X") }
func (sigImpl) Streaming1(_ *streamStringBool, _ int64) (float64, error) { panic("X") }

type streamStringBool struct{ ipc.ServerCall }

func (*streamStringBool) Init(ipc.ServerCall) { panic("X") }
func (*streamStringBool) RecvStream() interface {
	Advance() bool
	Value() string
	Err() error
} {
	panic("X")
}
func (*streamStringBool) SendStream() interface {
	Send(_ bool) error
} {
	panic("X")
}

// TODO(toddw): This test only works with vom2, since we need to send *vdl.Type
// over the wire for the new Signature format.  Re-enable this test when the
// vom2 transition is done.
func disabledTestMethodSignature(t *testing.T) {
	runtime, err := rt.New()
	if err != nil {
		t.Fatalf("Couldn't initialize runtime: %s", err)
	}
	defer runtime.Cleanup()

	ep, stop, err := startSigServer(runtime, sigImpl{})
	if err != nil {
		t.Fatalf("startSigServer: %v", err)
	}
	defer stop()
	name := naming.JoinAddressName(ep, "")

	tests := []struct {
		Method string
		Want   signature.Method
	}{
		{"NonStreaming0", signature.Method{
			Name: "NonStreaming0",
		}},
		{"NonStreaming1", signature.Method{
			Name:    "NonStreaming1",
			InArgs:  []signature.Arg{{Type: vdl.StringType}},
			OutArgs: []signature.Arg{{Type: vdl.ErrorType}},
		}},
		{"Streaming0", signature.Method{
			Name:      "Streaming0",
			InStream:  &signature.Arg{Type: vdl.StringType},
			OutStream: &signature.Arg{Type: vdl.BoolType},
		}},
		{"Streaming1", signature.Method{
			Name:      "Streaming1",
			InArgs:    []signature.Arg{{Type: vdl.Int64Type}},
			OutArgs:   []signature.Arg{{Type: vdl.Float64Type}, {Type: vdl.ErrorType}},
			InStream:  &signature.Arg{Type: vdl.StringType},
			OutStream: &signature.Arg{Type: vdl.BoolType},
		}},
	}
	for _, test := range tests {
		sig, err := reserved.MethodSignature(runtime.NewContext(), nil, name, test.Method)
		if err != nil {
			t.Errorf("call failed: %v", err)
		}
		if got, want := sig, test.Want; !reflect.DeepEqual(got, want) {
			t.Errorf("%s got %#v, want %#v", test.Method, got, want)
		}
	}
}

// TODO(toddw): This test only works with vom2, since we need to send *vdl.Type
// over the wire for the new Signature format.  Re-enable this test when the
// vom2 transition is done.
func disabledTestSignature(t *testing.T) {
	runtime, err := rt.New()
	if err != nil {
		t.Fatalf("Couldn't initialize runtime: %s", err)
	}
	defer runtime.Cleanup()

	ep, stop, err := startSigServer(runtime, sigImpl{})
	if err != nil {
		t.Fatalf("startSigServer: %v", err)
	}
	defer stop()
	name := naming.JoinAddressName(ep, "")

	want := []signature.Interface{
		signature.Interface{
			Doc: "The empty interface contains methods not attached to any interface.",
			Methods: []signature.Method{
				{
					Name: "NonStreaming0",
				},
				{
					Name:    "NonStreaming1",
					InArgs:  []signature.Arg{{Type: vdl.StringType}},
					OutArgs: []signature.Arg{{Type: vdl.ErrorType}},
				},
				{
					Name:      "Streaming0",
					InStream:  &signature.Arg{Type: vdl.StringType},
					OutStream: &signature.Arg{Type: vdl.BoolType},
				},
				{
					Name:      "Streaming1",
					InArgs:    []signature.Arg{{Type: vdl.Int64Type}},
					OutArgs:   []signature.Arg{{Type: vdl.Float64Type}, {Type: vdl.ErrorType}},
					InStream:  &signature.Arg{Type: vdl.StringType},
					OutStream: &signature.Arg{Type: vdl.BoolType},
				},
			},
		},
	}

	sig, err := reserved.Signature(runtime.NewContext(), nil, name)
	if err != nil {
		t.Errorf("call failed: %v", err)
	}
	if got, want := sig, want; !reflect.DeepEqual(got, want) {
		t.Errorf("got %#v, want %#v", got, want)
	}
}