From fa4835f87aaea5143978479910a11fcc6034f400 Mon Sep 17 00:00:00 2001
From: Frost Ming <mianghong@gmail.com>
Date: Fri, 30 Oct 2020 14:52:30 +0800
Subject: [PATCH] Add test case for multiple spinner

---
 tests/test_halo.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/tests/test_halo.py b/tests/test_halo.py
index 29918aa..17d5785 100644
--- a/tests/test_halo.py
+++ b/tests/test_halo.py
@@ -622,8 +622,26 @@ def test_redirect_stdout(self):
 
         self.assertIn('foo', output[0])
 
+    def test_running_multiple_instances(self):
+        spinner = Halo("foo", stream=self._stream)
+        _instance = Halo()
+        # Pretend that another spinner is being displayed under spinner
+        _instance._content = "Some lines\n"
+        Halo._instances.extend([spinner, _instance])
+        spinner.start()
+        time.sleep(1)
+        spinner.stop()
+        spinner.stop_and_persist(text="Done")
+        output = self._get_test_output()["text"]
+        self.assertEqual(output[0], "{} foo".format(frames[0]))
+        self.assertEqual(output[1], "Some lines")
+        self.assertEqual(output[2], "{} foo".format(frames[1]))
+        self.assertEqual(output[3], "Some lines")
+        self.assertEqual(output[-2], "  Done")
+        self.assertEqual(output[-1], "Some lines")
+
     def tearDown(self):
-        pass
+        self._stream.close()
 
 
 if __name__ == '__main__':