Fix batch_async obsolete test

openSUSE · Dec 4, 2019 · b86c2b0 · b86c2b0
1 parent 635099e
commit b86c2b0
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/unit/cli/test_batch_async.py b/tests/unit/cli/test_batch_async.py
@@ -140,8 +140,14 @@ def test_batch_fire_done_event(self):
                 "salt/batch/1235/done"
             )
         )
+
+    def test_batch__del__(self):
+        batch = BatchAsync(MagicMock(), MagicMock(), MagicMock())
+        event = MagicMock()
+        batch.event = event
+        batch.__del__()
         self.assertEqual(
-            len(self.batch.event.remove_event_handler.mock_calls), 1)
+            len(event.remove_event_handler.mock_calls), 1)
 
     @tornado.testing.gen_test
     def test_batch_next(self):