From 6402d5a0212e2c4a6595b6a3b0e1071e459c5762 Mon Sep 17 00:00:00 2001 From: Edin Sarajlic Date: Tue, 15 Oct 2019 12:50:56 +1100 Subject: [PATCH] Add unit test for generic util function: port_knock_units --- .../utilities/test_zaza_utilities_generic.py | 26 +++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/unit_tests/utilities/test_zaza_utilities_generic.py b/unit_tests/utilities/test_zaza_utilities_generic.py index a4eed09..c992e03 100644 --- a/unit_tests/utilities/test_zaza_utilities_generic.py +++ b/unit_tests/utilities/test_zaza_utilities_generic.py @@ -602,3 +602,29 @@ class TestGenericUtils(ut_utils.BaseTestCase): self.assertEqual(actual, expected) + def test_port_knock_units(self): + self.patch( + "zaza.openstack.utilities.generic.is_port_open", + new_callable=mock.MagicMock(), + name="_is_port_open" + ) + + _units = [ + mock.MagicMock(), + mock.MagicMock(), + ] + + self._is_port_open.side_effect = [True, True] + self.assertIsNone(generic_utils.port_knock_units(_units)) + self.assertEqual(self._is_port_open.call_count, len(_units)) + + self._is_port_open.side_effect = [True, False] + self.assertIsNotNone(generic_utils.port_knock_units(_units)) + + # check when func is expecting failure, i.e. should succeed + self._is_port_open.reset_mock() + self._is_port_open.side_effect = [False, False] + self.assertIsNone(generic_utils.port_knock_units(_units, + expect_success=False)) + self.assertEqual(self._is_port_open.call_count, len(_units)) +