def test_is_service_available_waiting(self, thread_mgr_get):
"""Test StartRHSMTask - test is_service_available() - waiting."""
# put this into a variable to fit the patch invocation on single line
is_running_import = 'pyanaconda.modules.common.task.task.Task.is_running'
with patch(is_running_import, new_callable=PropertyMock) as is_running:
# fake is_running
is_running.return_value = True
# create the task
task = StartRHSMTask()
# fake get_result()
task.get_result = Mock()
task.get_result.return_value = True
# make sure is_running is True
assert task.is_running
# assure is_running switches to False before
# the method starts waiting on the mock thread
mock_thread = Mock()
def set_running_false(thread):
is_running.return_value = False
return mock_thread
thread_mgr_get.side_effect = set_running_false
# by replacing the thread by Mock instance,
# we can avoid running the method in a thread
# as it will join() Mock instance not a real thread
assert task.is_service_available(timeout=1.0)
# check that the mock thread was joined with the
# expected timeout value
mock_thread.join.assert_called_once_with(1.0)
def test_is_service_available_timeout(self, thread_mgr_get):
"""Test StartRHSMTask - test is_service_available() - timeout."""
# put this into a variable to fit the patch invocation on single line
is_running_import = 'pyanaconda.modules.common.task.task.Task.is_running'
with patch(is_running_import, new_callable=PropertyMock) as is_running:
# fake is_running
is_running.return_value = True
# create the task
task = StartRHSMTask()
# fake get_result()
task.get_result = Mock()
task.get_result.return_value = False
# make sure is_running is True
assert task.is_running
# us a mock thread, so that it's
# join method exists immediately
mock_thread = Mock()
thread_mgr_get.return_value = mock_thread
# test the method times out
assert not task.is_service_available(timeout=1.0)
# check that the mock thread join() was called with expected
# timeout
mock_thread.join.assert_called_once_with(1.0)
def test_is_service_available_failure(self):
"""Test StartRHSMTask - test is_service_available() - failure."""
# create the task
task = StartRHSMTask()
# fake get_result()
task.get_result = Mock()
task.get_result.return_value = False
# test the method
assert not task.is_service_available(1)
def test_is_service_available_success(self):
"""Test StartRHSMTask - test is_service_available() - success."""
# create the task
task = StartRHSMTask()
# fake get_result()
task.get_result = Mock()
task.get_result.return_value = True
# test the method
assert task.is_service_available(1)
