From 1603f70b1f3e883aa420f0ee80cc7fde2c9ac454 Mon Sep 17 00:00:00 2001
From: Henk <henk@spaaij.net>
Date: Sat, 31 Dec 2022 13:05:34 -0800
Subject: [PATCH] Add testcase for unknown method

---
 server/server_test.go | 62 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)

diff --git a/server/server_test.go b/server/server_test.go
index b9578624..8444fc84 100644
--- a/server/server_test.go
+++ b/server/server_test.go
@@ -3,12 +3,17 @@ package server_test
 import (
 	"context"
 	"errors"
+	"net"
 	"strings"
 	"sync"
+	"sync/atomic"
 	"testing"
 
+	"github.com/stretchr/testify/require"
+
 	"capnproto.org/go/capnp/v3"
 	air "capnproto.org/go/capnp/v3/internal/aircraftlib"
+	"capnproto.org/go/capnp/v3/rpc"
 	"capnproto.org/go/capnp/v3/server"
 
 	"github.com/stretchr/testify/assert"
@@ -91,6 +96,63 @@ func TestServerCall(t *testing.T) {
 			}
 		}
 	})
+	t.Run("Unimplemented hook", func(t *testing.T) {
+		t.Parallel()
+		var echoText = "are you there?"
+		var proxyReceived atomic.Value
+
+		// start a proxy server with hook
+		srv := server.New(nil, nil, nil)
+		srv.HandleUnknownMethod = func(method capnp.Method) *server.Method {
+			sm := server.Method{
+				Method: method,
+				Impl:   nil,
+			}
+			sm.Impl = func(ctx context.Context, call *server.Call) error {
+				t.Log("welcome to blank")
+				echoArgs := air.Echo_echo_Params(call.Args())
+				inText, err := echoArgs.In()
+				assert.NoError(t, err)
+				proxyReceived.Store(inText)
+				// pretend we received an answer
+				echo := air.Echo_echo{Call: call}
+				resp, err := echo.AllocResults()
+				err = resp.SetOut(inText)
+				return err
+			}
+			return &sm
+		}
+		blankBoot := capnp.NewClient(srv)
+		lis, err := net.Listen("tcp", ":0")
+		defer lis.Close()
+		require.NoError(t, err)
+		go rpc.Serve(lis, blankBoot)
+
+		// invoke the proxy server with the echo client
+		addr := lis.Addr().String()
+		conn, err := net.Dial("tcp", addr)
+		assert.NoError(t, err)
+		transport := rpc.NewStreamTransport(conn)
+		rpcConn := rpc.NewConn(transport, nil)
+		defer rpcConn.Close()
+		blankClient := rpcConn.Bootstrap(context.Background())
+		echoClient := air.Echo(blankClient)
+		defer echoClient.Release()
+
+		ans, finish := echoClient.Echo(context.Background(), func(p air.Echo_echo_Params) error {
+			err := p.SetIn(echoText)
+			return err
+		})
+		defer finish()
+		resp, err := ans.Struct()
+		answerOut, _ := resp.Out()
+		rxValue := proxyReceived.Load()
+		assert.Equal(t, echoText, rxValue)
+		assert.Equal(t, echoText, answerOut)
+		if err != nil {
+			t.Error("echo.Echo() error != <nil>; want success")
+		}
+	})
 }
 
 type callSeq uint32