1 files changed, 7 insertions, 4 deletions
diff --git a/internal/net/server_test.go b/internal/net/server_test.go
index b106c53..c91c701 100644
--- a/internal/net/server_test.go
+++ b/internal/net/server_test.go
@@ -3,6 +3,7 @@ package net
 import (
 	"sync"
 	"testing"
+	"time"
 )
 
 func TestHubConcurrency(t *testing.T) {
@@ -38,10 +39,12 @@ func TestHubConcurrency(t *testing.T) {
 
 type testConn struct{ data []byte }
 
-func (c *testConn) ReadMessage() (string, error) { return "", nil }
-func (c *testConn) Write(b []byte) (int, error)  { c.data = append(c.data, b...); return len(b), nil }
-func (c *testConn) Close() error                 { return nil }
-func (c *testConn) SetEcho(bool) error           { return nil }
+func (c *testConn) ReadMessage() (string, error)     { return "", nil }
+func (c *testConn) Write(b []byte) (int, error)      { c.data = append(c.data, b...); return len(b), nil }
+func (c *testConn) SetWriteDeadline(time.Time) error { return nil }
+
+func (c *testConn) Close() error       { return nil }
+func (c *testConn) SetEcho(bool) error { return nil }
 
 func TestSessionWriteConcurrency(t *testing.T) {
 	conn := &testConn{}