derohe-miniblock-mod/vendor/github.com/creachadair/jrpc2/server/run_test.go

// Copyright (C) 2017 Michael J. Fromberger. All Rights Reserved.

package server_test

import (
	"context"
	"testing"

	"github.com/creachadair/jrpc2"
	"github.com/creachadair/jrpc2/channel"
	"github.com/creachadair/jrpc2/handler"
	"github.com/creachadair/jrpc2/server"
)

type testService struct {
	assigner                   jrpc2.Assigner
	assignCalled, finishCalled bool
	stat                       jrpc2.ServerStatus
}

func (t *testService) Assigner() (jrpc2.Assigner, error) {
	t.assignCalled = true
	return t.assigner, nil
}

func (t *testService) Finish(_ jrpc2.Assigner, stat jrpc2.ServerStatus) {
	t.finishCalled = true
	t.stat = stat
}

func TestRun(t *testing.T) {
	svc := &testService{assigner: handler.Map{
		"Test": handler.New(func(ctx context.Context) string {
			return "OK"
		}),
	}}
	cpipe, spipe := channel.Direct()
	cdone := make(chan struct{})
	var result string
	go func() {
		defer close(cdone)
		cli := jrpc2.NewClient(cpipe, nil)
		defer cli.Close()
		if err := cli.CallResult(context.Background(), "Test", nil, &result); err != nil {
			t.Errorf("Call Test failed: %v", err)
		}
	}()

	if err := server.Run(spipe, svc, nil); err != nil {
		t.Errorf("Server failed: %v", err)
	}
	if result != "OK" {
		t.Errorf("Call result: got %q, want %q", result, "OK")
	}
	if !svc.assignCalled {
		t.Error("Assigner was not called")
	}
	if !svc.finishCalled {
		t.Error("Finish was not called")
	}
	if svc.stat.Err != nil {
		t.Errorf("Server status: unexpected error: %+v", svc.stat)
	}
}