From 44565b398e733d785cab0cc305ddbeb6f8e8c91b Mon Sep 17 00:00:00 2001
From: cheatfate <ka@hardcore.kiev.ua>
Date: Mon, 6 Aug 2018 22:33:52 +0300
Subject: [PATCH] Improved test for #6.

---
 tests/testbugs.nim | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/tests/testbugs.nim b/tests/testbugs.nim
index c6122e6..dedd0a4 100644
--- a/tests/testbugs.nim
+++ b/tests/testbugs.nim
@@ -20,7 +20,12 @@ type
 proc udp4DataAvailable(transp: DatagramTransport,
                      remote: TransportAddress): Future[void] {.async, gcsafe.} =
   var udata = getUserData[CustomData](transp)
-  if udata.test == "CHECK":
+  var expect = TEST_MSG
+  var data: seq[byte]
+  var datalen: int
+  transp.peekMessage(data, datalen)
+  if udata.test == "CHECK" and datalen == MSG_LEN and
+     equalMem(addr data[0], addr expect[0], datalen):
     udata.test = "OK"
   transp.close()