From c6a2c8de6c2bcea52ef123d518f4eb071eb81571 Mon Sep 17 00:00:00 2001
From: Asim Aslam <asim@aslam.me>
Date: Wed, 14 Nov 2018 19:45:46 +0000
Subject: [PATCH] add local/remote to testsocket

---
 server/rpc_codec_test.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/server/rpc_codec_test.go b/server/rpc_codec_test.go
index 76f5094e..9ef8e7fa 100644
--- a/server/rpc_codec_test.go
+++ b/server/rpc_codec_test.go
@@ -15,6 +15,8 @@ type testCodec struct {
 }
 
 type testSocket struct {
+	local  string
+	remote string
 }
 
 // TestCodecWriteError simulates what happens when a codec is unable
@@ -87,6 +89,14 @@ func (c *testCodec) String() string {
 	return "string"
 }
 
+func (s testSocket) Local() string {
+	return s.local
+}
+
+func (s testSocket) Remote() string {
+	return s.remote
+}
+
 func (s testSocket) Recv(message *transport.Message) error {
 	return nil
 }