From 8146a193eb3b615637d5100ab9c683e6d1becdec Mon Sep 17 00:00:00 2001 From: James Cammarata Date: Fri, 24 Jul 2015 10:53:54 -0400 Subject: [PATCH] Fix strategy base unit tests after 88e1ef8 --- test/units/plugins/strategies/test_strategy_base.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/units/plugins/strategies/test_strategy_base.py b/test/units/plugins/strategies/test_strategy_base.py index 9b84fcba1bc..eb85b9c771d 100644 --- a/test/units/plugins/strategies/test_strategy_base.py +++ b/test/units/plugins/strategies/test_strategy_base.py @@ -41,6 +41,7 @@ class TestStrategyBase(unittest.TestCase): def test_strategy_base_init(self): mock_tqm = MagicMock(TaskQueueManager) mock_tqm._final_q = MagicMock() + mock_tqm._options = MagicMock() strategy_base = StrategyBase(tqm=mock_tqm) def test_strategy_base_run(self): @@ -57,6 +58,7 @@ class TestStrategyBase(unittest.TestCase): mock_tqm._failed_hosts = dict() mock_tqm._unreachable_hosts = dict() + mock_tqm._options = MagicMock() strategy_base = StrategyBase(tqm=mock_tqm) self.assertEqual(strategy_base.run(iterator=mock_iterator, play_context=mock_play_context), 0)