From f917dc6722d3a16a6cbb3e4ba9538152c22c3218 Mon Sep 17 00:00:00 2001
From: bvn13 <mail4bvn@gmail.com>
Date: Fri, 28 Jan 2022 15:22:57 +0300
Subject: [PATCH] repeatable test echo client

---
 src/test/java/me/bvn13/sewy/EchoClientListener.java | 4 +++-
 src/test/java/me/bvn13/sewy/ServerTest.java         | 7 +++++--
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/test/java/me/bvn13/sewy/EchoClientListener.java b/src/test/java/me/bvn13/sewy/EchoClientListener.java
index 3ad4a45..e3909e4 100644
--- a/src/test/java/me/bvn13/sewy/EchoClientListener.java
+++ b/src/test/java/me/bvn13/sewy/EchoClientListener.java
@@ -9,6 +9,8 @@ public class EchoClientListener extends AbstractClientListener {
 
     @Override
     public void run() {
-        writeLine(readLine());
+        while (socket.isConnected()) {
+            writeLine(readLine());
+        }
     }
 }
diff --git a/src/test/java/me/bvn13/sewy/ServerTest.java b/src/test/java/me/bvn13/sewy/ServerTest.java
index f51c17e..097151d 100644
--- a/src/test/java/me/bvn13/sewy/ServerTest.java
+++ b/src/test/java/me/bvn13/sewy/ServerTest.java
@@ -57,8 +57,11 @@ public class ServerTest {
         new Server("localhost", port, EchoClientListener.class);
         Client client = new Client("localhost", port);
         client.writeLine("hello");
-        String response = client.readLine();
-        Assertions.assertEquals("hello", response);
+        String response1 = client.readLine();
+        Assertions.assertEquals("hello", response1);
+        client.writeLine("olleh");
+        String response2 = client.readLine();
+        Assertions.assertEquals("olleh", response2);
     }
 
 }