Add test case for multiple spinner

manrajgrover · Oct 30, 2020 · a34da90 · a34da90
1 parent f1a95c3
commit a34da90
Showing 1 changed file with 19 additions and 1 deletion.
diff --git 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__':