Fixing unit test for included file changes

1 files changed, 5 insertions, 2 deletions

diff --git a/test/units/plugins/strategies/test_strategy_base.py b/test/units/plugins/strategies/test_strategy_base.py
index 7d8cb42ee6..4c177f7343 100644
--- a/test/units/plugins/strategies/test_strategy_base.py
+++ b/test/units/plugins/strategies/test_strategy_base.py
@@ -299,14 +299,17 @@ class TestStrategyBase(unittest.TestCase):
         mock_task._block = mock_block
         mock_task._role = None
 
+        mock_iterator = MagicMock()
+        mock_iterator.mark_host_failed.return_value = None
+
         mock_inc_file = MagicMock()
         mock_inc_file._task = mock_task
 
         mock_inc_file._filename = "test.yml"
-        res = strategy_base._load_included_file(included_file=mock_inc_file)
+        res = strategy_base._load_included_file(included_file=mock_inc_file, iterator=mock_iterator)
 
         mock_inc_file._filename = "bad.yml"
-        self.assertRaises(AnsibleParserError, strategy_base._load_included_file, included_file=mock_inc_file)
+        self.assertRaises(AnsibleParserError, strategy_base._load_included_file, included_file=mock_inc_file, iterator=mock_iterator)
 
     def test_strategy_base_run_handlers(self):
         workers = []