From 049d57970559d4b1f798ef32833a74d576cd9663 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Miroslav=20=C5=A0ediv=C3=BD?= <sedivy.miro@gmail.com>
Date: Thu, 8 Dec 2022 21:56:32 +0100
Subject: [PATCH] Add tests for read/write from/to closed connection

Add tests for read/write from/to closed connection
---
 agent_udpmux_test.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/agent_udpmux_test.go b/agent_udpmux_test.go
index 7572c8b..a1841ca 100644
--- a/agent_udpmux_test.go
+++ b/agent_udpmux_test.go
@@ -83,6 +83,14 @@ func TestMuxAgent(t *testing.T) {
 			require.NoError(t, conn.Close())
 			require.NoError(t, muxedConn.Close())
 			require.NoError(t, udpMux.Close())
+
+			// expect error when reading from closed mux
+			_, err = muxedConn.Read(data)
+			require.Error(t, err)
+
+			// expect error when writing to closed mux
+			_, err = muxedConn.Write(data)
+			require.Error(t, err)
 		})
 	}
 }