Patch teardown in some test methods

Specifically, test_setup_called() and test_begin_called()

Signed-off-by: Zack Cerza <zack@redhat.com>

diff --git a/teuthology/test/task/__init__.py b/teuthology/test/task/__init__.py
index a2b804cf621246e140716083b816a19b7a4ad325..f03737e8cf1d9375b4c5cdf1cc3bd7e0f04d5e26 100644 (file)
--- a/teuthology/test/task/__init__.py
+++ b/teuthology/test/task/__init__.py
@@ -146,6 +146,7 @@ class TestTask(object):
             setup=DEFAULT,
             begin=DEFAULT,
             end=DEFAULT,
+            teardown=DEFAULT,
         ):
             with self.klass(self.ctx, self.task_config) as task:
                 task.setup.assert_called_once_with()
@@ -156,6 +157,7 @@ class TestTask(object):
             setup=DEFAULT,
             begin=DEFAULT,
             end=DEFAULT,
+            teardown=DEFAULT,
         ):
             with self.klass(self.ctx, self.task_config) as task:
                 task.begin.assert_called_once_with()