From f2ccc802fd9ac9933fbdbbd83fdf6ba205dd26af Mon Sep 17 00:00:00 2001
From: John-Mark Gurney <jmg@funkthat.com>
Date: Tue, 4 May 2021 01:46:37 -0700
Subject: [PATCH] make sure that thing get closed down properly..

---
 multicast.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/multicast.py b/multicast.py
index 218d847..b9298d9 100644
--- a/multicast.py
+++ b/multicast.py
@@ -27,6 +27,12 @@ class StupidProtocol(object):
 	def __init__(self):
 		self.transport = None
 
+	def close(self):
+		return self.transport.close()
+
+	def connection_lost(self, exc):
+		self.transport = None
+
 	def connection_made(self, transport):
 		# Note: the connection_made call seems to be sync.  This
 		# isn't documented, and I don't know how to force a test
@@ -89,3 +95,7 @@ class TestMulticast(unittest.IsolatedAsyncioTestCase):
 
 		self.assertEqual((await l1.recv())[0], msg)
 		self.assertEqual((await l2.recv())[0], msg)
+
+		t1.close()
+		l1.close()
+		l2.close()