test(packet): test packet framer and packet encoding/decoding

1 files changed, 5 insertions, 3 deletions

diff --git a/internal/server/server.go b/internal/server/server.go
index 05af434..1b8dccb 100644
--- a/internal/server/server.go
+++ b/internal/server/server.go
@@ -152,7 +152,7 @@ func handleConnection(ctx context.Context, conn net.Conn, server *server) {
 	sess := session.NewSession(server, addr, id, pubKey)
 	server.AddSession(sess)
 	ctx = session.NewContext(ctx, sess)
-	framer := packet.NewFramer(ctx)
+	framer := packet.NewFramer()
 
 	defer func() {
 		conn.Close()
@@ -292,9 +292,11 @@ func processRequest(ctx context.Context, request packet.Payload) packet.Payload
 
 	switch request := request.(type) {
 	case *packet.SendMessage:
-		return timeout(100 * time.Millisecond, api.SendMessage, ctx, request)
+		return timeout(50 * time.Millisecond, api.SendMessage, ctx, request)
+	case *packet.GetMessagesRange:
+		return timeout(100 * time.Millisecond, api.GetMessages, ctx, request)
 	default:
-		return &packet.ErrorMessage{Error: "use of unsupported packet type"}
+		return &packet.ErrorMessage{Error: "use of disallowed packet type for request"}
 	}
 }