class TaskMonitorTest(unittest.TestCase):
URI = "/rest/testuri"
def setUp(self):
super(TaskMonitorTest, self).setUp()
self.host = '127.0.0.1'
self.connection = connection(self.host)
self.task_monitor = TaskMonitor(self.connection)
@mock.patch.object(connection, 'get')
def test_get_associated_resource_with_task(self, mock_get):
task = {
"category": "tasks",
"type": "TaskResourceV2",
"associatedResource": {
"resourceUri": "/rest/associatedresourceuri"
}
}
mock_get.return_value = {"resource": "resource1"}
ret_task, entity = self.task_monitor.get_associated_resource(
task.copy())
self.assertEqual(entity, {"resource": "resource1"})
self.assertEqual(ret_task, task)
mock_get.assert_called_once_with("/rest/associatedresourceuri")
@mock.patch.object(connection, 'get')
def test_get_associated_resource_with_backup(self, mock_get):
backup = {
"category": "backups",
"type": "BACKUP",
"taskUri": "/rest/taskuri",
}
task = {
"category": "TaskResourceV2",
"type": "tasks",
"uri": "/rest/justuri",
}
def inner_get(uri):
if uri == "/rest/taskuri":
return task.copy()
else:
return {"resource": "resource1"}
mock_get.side_effect = inner_get
mock_get.return_value = task.copy()
ret_task, entity = self.task_monitor.get_associated_resource(
backup.copy())
self.assertEqual(entity, {"resource": "resource1"})
self.assertEqual(ret_task, task)
def test_get_associated_resource_support_dump(self):
task = {
"category": "tasks",
"type": "TaskResourceV2",
"associatedResource": {
"resourceUri":
"/rest/appliance/support-dumps/hennig59.eco-MyDump16-E-2016_07_07-17_53_42.867287.sdmp"
}
}
ret_task, entity = self.task_monitor.get_associated_resource(
task.copy())
self.assertEqual(
entity,
"/rest/appliance/support-dumps/hennig59.eco-MyDump16-E-2016_07_07-17_53_42.867287.sdmp"
)
self.assertEqual(ret_task, task)
def test_get_associated_resource_with_task_empty(self):
try:
self.task_monitor.get_associated_resource({})
except HPOneViewUnknownType as e:
self.assertEqual(MSG_INVALID_TASK, e.msg)
else:
self.fail()
def test_get_associated_resource_with_invalid_task(self):
try:
self.task_monitor.get_associated_resource(
{"category": "networking"})
except HPOneViewUnknownType as e:
self.assertEqual(MSG_UNKNOWN_OBJECT_TYPE, e.msg)
else:
self.fail()
def test_get_associated_resource_with_invalid_type(self):
try:
self.task_monitor.get_associated_resource({
"category": "tasks",
"type": "TaskResource"
})
except HPOneViewInvalidResource as e:
self.assertEqual(MSG_TASK_TYPE_UNRECONIZED % "TaskResource", e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'get')
def test_is_task_running(self, mock_get):
mock_get.return_value = {"uri": "uri", "taskState": "Pending"}
self.assertTrue(self.task_monitor.is_task_running({"uri": "uri"}))
@mock.patch.object(TaskMonitor, 'get')
def test_is_task_running_false(self, mock_get):
mock_get.return_value = {"uri": "uri", "taskState": "Warning"}
self.assertFalse(self.task_monitor.is_task_running({"uri": "uri"}))
@mock.patch.object(TaskMonitor, 'is_task_running')
def test_wait_for_task_timeout(self, mock_is_running):
mock_is_running.return_value = True
timeout = 2
try:
self.task_monitor.wait_for_task({"uri": "uri"}, timeout)
except HPOneViewTimeout as e:
self.assertEqual(MSG_TIMEOUT % timeout, e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch('time.sleep')
def test_wait_for_task_increasing_sleep(self, mock_sleep, mock_is_running):
mock_is_running.return_value = True
timeout = 0.1
# should call sleep increasing 1 until 10
calls = [
call(1),
call(2),
call(3),
call(4),
call(5),
call(6),
call(7),
call(8),
call(9),
call(10),
call(10),
call(10)
]
try:
self.task_monitor.wait_for_task({"uri": "uri"}, timeout)
except HPOneViewTimeout as e:
mock_sleep.assert_has_calls(calls)
self.assertEqual(MSG_TIMEOUT % timeout, e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_with_error_message(self, mock_get, mock_is_running):
task = {
"uri": "uri",
"taskState": "Error",
"taskErrors": [{
"message": "Error Message"
}]
}
mock_is_running.return_value = False
mock_get.return_value = task
try:
self.task_monitor.wait_for_task(task.copy())
except HPOneViewTaskError as e:
self.assertEqual("Error Message", e.msg)
self.assertEqual(None, e.error_code)
else:
self.fail("Expected exception not raised")
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_with_error_message_and_error_code(
self, mock_get, mock_is_running):
task = {
"uri":
"uri",
"taskState":
"Error",
"taskErrors": [{
"message": "Error Message",
"errorCode": "ProfileAlreadyExistsInServer"
}]
}
mock_is_running.return_value = False
mock_get.return_value = task
try:
self.task_monitor.wait_for_task(task.copy())
except HPOneViewTaskError as e:
self.assertEqual("Error Message", e.msg)
self.assertEqual("ProfileAlreadyExistsInServer", e.error_code)
else:
self.fail("Expected exception not raised")
def test_wait_for_task_empty(self):
try:
self.task_monitor.wait_for_task({})
except HPOneViewUnknownType as e:
self.assertEqual(MSG_INVALID_TASK, e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_with_error_empty(self, mock_get, mock_is_running):
task = {
"uri": "uri",
"taskState": "Error",
"taskStatus": "Failed",
}
mock_is_running.return_value = False
mock_get.return_value = task
try:
self.task_monitor.wait_for_task(task.copy())
except HPOneViewTaskError as e:
self.assertEqual("Failed", e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_with_error_unknown(self, mock_get, mock_is_running):
task = {
"uri": "uri",
"taskState": "Error",
}
mock_is_running.return_value = False
mock_get.return_value = task
try:
self.task_monitor.wait_for_task(task.copy())
except HPOneViewTaskError as e:
self.assertEqual(MSG_UNKNOWN_EXCEPTION, e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'get_associated_resource')
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task(self, mock_get, mock_is_running, mock_assoc_res):
task = {
"uri": "uri",
"type": "TaskResourceV2",
"name": "update",
"taskState": "Completed",
}
mock_is_running.return_value = False
mock_get.return_value = task
mock_assoc_res.return_value = task.copy(), {"resource": "resource1"}
ret_entity = self.task_monitor.wait_for_task(task.copy())
self.assertEqual(ret_entity, {"resource": "resource1"})
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_unexpected_result(self, mock_get, mock_is_running):
task = {
"uri": "uri",
"type": "Undefined",
"name": "Undefined",
"taskState": "Completed",
}
mock_is_running.return_value = False
mock_get.return_value = task
ret_entity = self.task_monitor.wait_for_task(task.copy())
self.assertEqual(ret_entity, task.copy())
@mock.patch.object(TaskMonitor, 'get_associated_resource')
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_delete(self, mock_get, mock_is_running,
mock_assoc_res):
task = {
"uri": "uri",
"type": "TaskResourceV2",
"name": "Delete",
"taskState": "Completed",
}
mock_is_running.return_value = False
mock_get.return_value = task
mock_assoc_res.return_value = task.copy(), {"resource": "resource1"}
ret = self.task_monitor.wait_for_task(task.copy())
# may return a different type
self.assertEqual(True, ret)
@mock.patch.object(connection, 'get')
def test_get(self, mock_get):
self.task_monitor.get({"uri": "an uri"})
mock_get.assert_called_once_with("an uri")
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_get_completed_task(self, mock_get, mock_is_running):
task = {
"uri": "uri",
"type": "TaskResourceV2",
"name": "Create unexpected zoning report'",
"taskState": "Completed",
"taskOutput": []
}
mock_is_running.return_value = False
mock_get.return_value = task
response = self.task_monitor.get_completed_task(task.copy())
self.assertEqual(task, response)
class TaskMonitorTest(unittest.TestCase):
URI = "/rest/testuri"
def setUp(self):
super(TaskMonitorTest, self).setUp()
self.host = '127.0.0.1'
self.connection = connection(self.host)
self.task_monitor = TaskMonitor(self.connection)
@mock.patch.object(connection, 'get')
def test_get_associated_resource_with_task(self, mock_get):
task = {
"category": "tasks",
"type": "TaskResourceV2",
"associatedResource": {
"resourceUri": "/rest/associatedresourceuri"
}}
mock_get.return_value = {"resource": "resource1"}
ret_task, entity = self.task_monitor.get_associated_resource(task.copy())
self.assertEqual(entity, {"resource": "resource1"})
self.assertEqual(ret_task, task)
mock_get.assert_called_once_with("/rest/associatedresourceuri")
@mock.patch.object(connection, 'get')
def test_get_associated_resource_with_backup(self, mock_get):
backup = {
"category": "backups",
"type": "BACKUP",
"taskUri": "/rest/taskuri",
}
task = {
"category": "TaskResourceV2",
"type": "tasks",
"uri": "/rest/justuri",
}
def inner_get(uri):
if uri == "/rest/taskuri":
return task.copy()
else:
return {"resource": "resource1"}
mock_get.side_effect = inner_get
mock_get.return_value = task.copy()
ret_task, entity = self.task_monitor.get_associated_resource(backup.copy())
self.assertEqual(entity, {"resource": "resource1"})
self.assertEqual(ret_task, task)
def test_get_associated_resource_support_dump(self):
task = {
"category": "tasks",
"type": "TaskResourceV2",
"associatedResource": {
"resourceUri": "/rest/appliance/support-dumps/hennig59.eco-MyDump16-E-2016_07_07-17_53_42.867287.sdmp"
}}
ret_task, entity = self.task_monitor.get_associated_resource(task.copy())
self.assertEqual(
entity,
"/rest/appliance/support-dumps/hennig59.eco-MyDump16-E-2016_07_07-17_53_42.867287.sdmp")
self.assertEqual(ret_task, task)
def test_get_associated_resource_with_task_empty(self):
try:
self.task_monitor.get_associated_resource({})
except HPOneViewUnknownType as e:
self.assertEqual(MSG_INVALID_TASK, e.msg)
else:
self.fail()
def test_get_associated_resource_with_invalid_task(self):
try:
self.task_monitor.get_associated_resource({"category": "networking"})
except HPOneViewUnknownType as e:
self.assertEqual(MSG_UNKNOWN_OBJECT_TYPE, e.msg)
else:
self.fail()
def test_get_associated_resource_with_invalid_type(self):
try:
self.task_monitor.get_associated_resource({"category": "tasks",
"type": "TaskResource"})
except HPOneViewInvalidResource as e:
self.assertEqual(MSG_TASK_TYPE_UNRECONIZED % "TaskResource", e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'get')
def test_is_task_running(self, mock_get):
mock_get.return_value = {"uri": "uri",
"taskState": "Pending"}
self.assertTrue(self.task_monitor.is_task_running({"uri": "uri"}))
@mock.patch.object(TaskMonitor, 'get')
def test_is_task_running_false(self, mock_get):
mock_get.return_value = {"uri": "uri",
"taskState": "Warning"}
self.assertFalse(self.task_monitor.is_task_running({"uri": "uri"}))
@mock.patch.object(TaskMonitor, 'is_task_running')
def test_wait_for_task_timeout(self, mock_is_running):
mock_is_running.return_value = True
timeout = 2
try:
self.task_monitor.wait_for_task({"uri": "uri"}, timeout)
except HPOneViewTimeout as e:
self.assertEqual(MSG_TIMEOUT % timeout, e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch('time.sleep')
def test_wait_for_task_increasing_sleep(self, mock_sleep, mock_is_running):
mock_is_running.return_value = True
timeout = 0.1
# should call sleep increasing 1 until 10
calls = [call(1), call(2), call(3), call(4), call(5), call(6), call(7),
call(8), call(9), call(10), call(10), call(10)]
try:
self.task_monitor.wait_for_task({"uri": "uri"}, timeout)
except HPOneViewTimeout as e:
mock_sleep.assert_has_calls(calls)
self.assertEqual(MSG_TIMEOUT % timeout, e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_with_error_message(self, mock_get, mock_is_running):
task = {"uri": "uri",
"taskState": "Error",
"taskErrors": [{"message": "Error Message"}]}
mock_is_running.return_value = False
mock_get.return_value = task
try:
self.task_monitor.wait_for_task(task.copy())
except HPOneViewTaskError as e:
self.assertEqual("Error Message", e.msg)
else:
self.fail()
def test_wait_for_task_empty(self):
try:
self.task_monitor.wait_for_task({})
except HPOneViewUnknownType as e:
self.assertEqual(MSG_INVALID_TASK, e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_with_error_empty(self, mock_get, mock_is_running):
task = {"uri": "uri",
"taskState": "Error",
"taskStatus": "Failed",
}
mock_is_running.return_value = False
mock_get.return_value = task
try:
self.task_monitor.wait_for_task(task.copy())
except HPOneViewTaskError as e:
self.assertEqual("Failed", e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_with_error_unknown(self, mock_get, mock_is_running):
task = {"uri": "uri",
"taskState": "Error",
}
mock_is_running.return_value = False
mock_get.return_value = task
try:
self.task_monitor.wait_for_task(task.copy())
except HPOneViewTaskError as e:
self.assertEqual(MSG_UNKNOWN_EXCEPTION, e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'get_associated_resource')
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task(self, mock_get, mock_is_running, mock_assoc_res):
task = {"uri": "uri",
"type": "TaskResourceV2",
"name": "update",
"taskState": "Completed",
}
mock_is_running.return_value = False
mock_get.return_value = task
mock_assoc_res.return_value = task.copy(), {"resource": "resource1"}
ret_entity = self.task_monitor.wait_for_task(task.copy())
self.assertEqual(ret_entity, {"resource": "resource1"})
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_unexpected_result(self, mock_get, mock_is_running):
task = {"uri": "uri",
"type": "Undefined",
"name": "Undefined",
"taskState": "Completed",
}
mock_is_running.return_value = False
mock_get.return_value = task
ret_entity = self.task_monitor.wait_for_task(task.copy())
self.assertEqual(ret_entity, task.copy())
@mock.patch.object(TaskMonitor, 'get_associated_resource')
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_delete(self, mock_get, mock_is_running, mock_assoc_res):
task = {"uri": "uri",
"type": "TaskResourceV2",
"name": "Delete",
"taskState": "Completed",
}
mock_is_running.return_value = False
mock_get.return_value = task
mock_assoc_res.return_value = task.copy(), {"resource": "resource1"}
ret = self.task_monitor.wait_for_task(task.copy())
# may return a different type
self.assertEqual(True, ret)
@mock.patch.object(connection, 'get')
def test_get(self, mock_get):
self.task_monitor.get({"uri": "an uri"})
mock_get.assert_called_once_with("an uri")
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_get_completed_task(self, mock_get, mock_is_running):
task = {"uri": "uri",
"type": "TaskResourceV2",
"name": "Create unexpected zoning report'",
"taskState": "Completed",
"taskOutput": []
}
mock_is_running.return_value = False
mock_get.return_value = task
response = self.task_monitor.get_completed_task(task.copy())
self.assertEqual(task, response)
def monitor_profile_update(oneview_client, task):
task_monitor = TaskMonitor(oneview_client.connection)
# Flag that sets if atleast one task is in running state
running_flag = True
# Create a toolbar to monitor creation of profiles
toolbar_width = 100
# setup toolbar
sys.stdout.write("\n\t\t\t\t[%s]" % (" " * toolbar_width))
sys.stdout.flush()
sys.stdout.write("\b" *
(toolbar_width + 1)) # return to start of line, after '['
prevPercentComplete = 0
while running_flag:
running_flag = False
percentComplete = 0
runningCount = 0
if task_monitor.is_task_running(task):
# Set the flag if atleast one task is running
running_flag = True
# Get the percent completed
percentComplete += task_monitor.get(task)['percentComplete']
runningCount += 1
if runningCount > 0:
percentComplete = percentComplete / runningCount
else:
percentComplete = 100
percentDiff = percentComplete - prevPercentComplete
prevPercentComplete = percentComplete
# Update toolbar
print("#" * int(percentDiff), end='')
sys.stdout.flush()
#print("percentComplete : {} %".format(percentComplete))
sleep(10)
sys.stdout.write("]\n")
sys.stdout.flush()
TASK_COMPLETED_STATES = ['Warning', 'Completed']
TASK_ERROR_STATES = ['Error', 'Terminated', 'Killed']
task = task_monitor.get(task)
profile_name = task['associatedResource']['resourceName']
if task['taskState'] in TASK_ERROR_STATES:
msg = None
error_code = None
if 'taskErrors' in task and len(task['taskErrors']) > 0:
err = task['taskErrors'][0]
if 'message' in err:
msg = err['message']
error_code = err.get('errorCode')
logging.info(
"\nFailed to update server profile - \"{}\".\n\tError msg: {}".
format(profile_name, msg))
raise Exception(
"\nFailed to update server profile - \"{}\".\n\tError msg: {}".
format(profile_name, msg))
class TaskMonitorTest(unittest.TestCase):
def setUp(self):
super(TaskMonitorTest, self).setUp()
self.host = '127.0.0.1'
self.connection = connection(self.host)
self.task_monitor = TaskMonitor(self.connection)
@mock.patch.object(connection, 'get')
def test_get_associated_resource_with_task(self, mock_get):
task = {
"category": "tasks",
"type": "TaskResourceV2",
"associatedResource": {
"resourceUri": "/rest/associatedresourceuri"
}
}
mock_get.return_value = {"resource": "resource1"}
ret_task, entity = self.task_monitor.get_associated_resource(
task.copy())
self.assertEqual(entity, {"resource": "resource1"})
self.assertEqual(ret_task, task)
mock_get.assert_called_once_with("/rest/associatedresourceuri")
@mock.patch.object(connection, 'get')
def test_get_associated_resource_with_backup(self, mock_get):
backup = {
"category": "backups",
"type": "BACKUP",
"taskUri": "/rest/taskuri",
}
task = {
"category": "TaskResourceV2",
"type": "tasks",
"uri": "/rest/justuri",
}
def inner_get(uri):
if uri == "/rest/taskuri":
return task.copy()
else:
return {"resource": "resource1"}
mock_get.side_effect = inner_get
mock_get.return_value = task.copy()
ret_task, entity = self.task_monitor.get_associated_resource(
backup.copy())
self.assertEqual(entity, {"resource": "resource1"})
self.assertEqual(ret_task, task)
def test_get_associated_resource_support_dump(self):
task = {
"category": "tasks",
"type": "TaskResourceV2",
"associatedResource": {
"resourceUri":
"/rest/appliance/support-dumps/hennig59.eco-MyDump16-E-2016_07_07-17_53_42.867287.sdmp"
}
}
ret_task, entity = self.task_monitor.get_associated_resource(
task.copy())
self.assertEqual(
entity,
"/rest/appliance/support-dumps/hennig59.eco-MyDump16-E-2016_07_07-17_53_42.867287.sdmp"
)
self.assertEqual(ret_task, task)
def test_get_associated_resource_with_task_empty(self):
try:
self.task_monitor.get_associated_resource({})
except HPOneViewUnknownType as e:
self.assertEqual(MSG_INVALID_TASK, e.msg)
else:
self.fail()
def test_get_associated_resource_with_invalid_task(self):
try:
self.task_monitor.get_associated_resource(
{"category": "networking"})
except HPOneViewUnknownType as e:
self.assertEqual(MSG_UNKNOWN_OBJECT_TYPE, e.msg)
else:
self.fail()
def test_get_associated_resource_with_invalid_type(self):
try:
self.task_monitor.get_associated_resource({
"category": "tasks",
"type": "TaskResource"
})
except HPOneViewInvalidResource as e:
self.assertEqual(MSG_TASK_TYPE_UNRECONIZED % "TaskResource", e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'get')
def test_is_task_running(self, mock_get):
mock_get.return_value = {"uri": "uri", "taskState": "Pending"}
self.assertTrue(self.task_monitor.is_task_running({"uri": "uri"}))
@mock.patch.object(TaskMonitor, 'get')
def test_is_task_running_false(self, mock_get):
mock_get.return_value = {"uri": "uri", "taskState": "Warning"}
self.assertFalse(self.task_monitor.is_task_running({"uri": "uri"}))
@mock.patch.object(TaskMonitor, 'get')
def test_is_task_running_ignore_network_failure(self, mock_get):
mock_get.side_effect = [{
"uri": "uri",
"taskState": "Pending"
},
EnvironmentError(ETIMEDOUT, ERR_MSG)]
connection_failure_control = dict(
last_success=self.task_monitor.get_current_seconds())
# 1. Success get
self.assertTrue(
self.task_monitor.is_task_running({"uri": "uri"},
connection_failure_control))
# 2. Timeout error, expected True anyway
self.assertTrue(
self.task_monitor.is_task_running({"uri": "uri"},
connection_failure_control))
@mock.patch.object(TaskMonitor, 'get')
def test_is_task_running_network_failure_without_timeout_control(
self, mock_get):
mock_get.side_effect = EnvironmentError(ETIMEDOUT, ERR_MSG)
self.assertRaises(EnvironmentError, self.task_monitor.is_task_running,
{"uri": "uri"})
@mock.patch.object(TaskMonitor, 'get')
def test_is_task_running_generic_failure_with_timeout_control(
self, mock_get):
mock_get.side_effect = Exception(ERR_MSG)
connection_failure_control = dict(
last_success=self.task_monitor.get_current_seconds())
self.assertRaises(Exception, self.task_monitor.is_task_running,
{"uri": "uri"}, connection_failure_control)
@mock.patch.object(TaskMonitor, 'get')
def test_is_task_running_ignore_network_failure_reset_count(
self, mock_get):
mock_get.side_effect = [{
"uri": "uri",
"taskState": "Pending"
},
EnvironmentError(ECONNABORTED, ERR_MSG), {
"uri": "uri",
"taskState": "Pending"
},
EnvironmentError(ETIMEDOUT, ERR_MSG)]
connection_failure_control = dict(
last_success=self.task_monitor.get_current_seconds())
# 1. Success get
self.assertTrue(
self.task_monitor.is_task_running({"uri": "uri"},
connection_failure_control))
# 2. Inside the timeout, must continue
self.assertTrue(
self.task_monitor.is_task_running({"uri": "uri"},
connection_failure_control))
# Force exceed the timeout
seconds_to_decrement = TaskMonitor.CONNECTION_FAILURE_TIMEOUT + 10
connection_failure_control['last_success'] -= seconds_to_decrement
# 3. Success get (reset timeout)
self.assertTrue(
self.task_monitor.is_task_running({"uri": "uri"},
connection_failure_control))
# 4. Inside the timeout again, must continue
self.assertTrue(
self.task_monitor.is_task_running({"uri": "uri"},
connection_failure_control))
@mock.patch.object(TaskMonitor, 'get')
def test_is_task_running_ignore_network_failure_exceed_timeout(
self, mock_get):
mock_get.side_effect = [{
"uri": "uri",
"taskState": "Pending"
},
EnvironmentError(ECONNABORTED, ERR_MSG)]
conn_failure_control = dict(
last_success=self.task_monitor.get_current_seconds())
# 1. Success get
self.assertTrue(
self.task_monitor.is_task_running({"uri": "uri"},
conn_failure_control))
# Decrement last success to force exceed the timeout
seconds_to_decrement = TaskMonitor.CONNECTION_FAILURE_TIMEOUT + 10
conn_failure_control['last_success'] -= seconds_to_decrement
# 2. Should fail, timeout exceeded
self.assertRaises(EnvironmentError, self.task_monitor.is_task_running,
{"uri": "uri"}, conn_failure_control)
@mock.patch.object(TaskMonitor, 'is_task_running')
def test_wait_for_task_timeout(self, mock_is_running):
mock_is_running.return_value = True
timeout = 2
try:
self.task_monitor.wait_for_task({"uri": "uri"}, timeout)
except HPOneViewTimeout as e:
self.assertEqual(MSG_TIMEOUT % timeout, e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch('time.sleep')
def test_wait_for_task_increasing_sleep(self, mock_sleep, mock_is_running):
mock_is_running.return_value = True
timeout = 0.1
# should call sleep increasing 1 until 10
calls = [
call(1),
call(2),
call(3),
call(4),
call(5),
call(6),
call(7),
call(8),
call(9),
call(10),
call(10),
call(10)
]
try:
self.task_monitor.wait_for_task({"uri": "uri"}, timeout)
except HPOneViewTimeout as e:
mock_sleep.assert_has_calls(calls)
self.assertEqual(MSG_TIMEOUT % timeout, e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_with_error_message(self, mock_get, mock_is_running):
task = {
"uri": "uri",
"taskState": "Error",
"taskErrors": [{
"message": "Error Message"
}]
}
mock_is_running.return_value = False
mock_get.return_value = task
try:
self.task_monitor.wait_for_task(task.copy())
except HPOneViewTaskError as e:
self.assertEqual("Error Message", e.msg)
self.assertEqual(None, e.error_code)
else:
self.fail("Expected exception not raised")
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_with_error_message_and_error_code(
self, mock_get, mock_is_running):
task = {
"uri":
"uri",
"taskState":
"Error",
"taskErrors": [{
"message": "Error Message",
"errorCode": "ProfileAlreadyExistsInServer"
}]
}
mock_is_running.return_value = False
mock_get.return_value = task
try:
self.task_monitor.wait_for_task(task.copy())
except HPOneViewTaskError as e:
self.assertEqual("Error Message", e.msg)
self.assertEqual("ProfileAlreadyExistsInServer", e.error_code)
else:
self.fail("Expected exception not raised")
def test_wait_for_task_empty(self):
try:
self.task_monitor.wait_for_task({})
except HPOneViewUnknownType as e:
self.assertEqual(MSG_INVALID_TASK, e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_with_error_empty(self, mock_get, mock_is_running):
task = {
"uri": "uri",
"taskState": "Error",
"taskStatus": "Failed",
}
mock_is_running.return_value = False
mock_get.return_value = task
try:
self.task_monitor.wait_for_task(task.copy())
except HPOneViewTaskError as e:
self.assertEqual("Failed", e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_with_error_unknown(self, mock_get, mock_is_running):
task = {
"uri": "uri",
"taskState": "Error",
}
mock_is_running.return_value = False
mock_get.return_value = task
try:
self.task_monitor.wait_for_task(task.copy())
except HPOneViewTaskError as e:
self.assertEqual(MSG_UNKNOWN_EXCEPTION, e.msg)
else:
self.fail()
@mock.patch.object(TaskMonitor, 'get_associated_resource')
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task(self, mock_get, mock_is_running, mock_assoc_res):
task = {
"uri": "uri",
"type": "TaskResourceV2",
"name": "update",
"taskState": "Completed",
}
mock_is_running.return_value = False
mock_get.return_value = task
mock_assoc_res.return_value = task.copy(), {"resource": "resource1"}
ret_entity = self.task_monitor.wait_for_task(task.copy())
self.assertEqual(ret_entity, {"resource": "resource1"})
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_unexpected_result(self, mock_get, mock_is_running):
task = {
"uri": "uri",
"type": "Undefined",
"name": "Undefined",
"taskState": "Completed",
}
mock_is_running.return_value = False
mock_get.return_value = task
ret_entity = self.task_monitor.wait_for_task(task.copy())
self.assertEqual(ret_entity, task.copy())
@mock.patch.object(TaskMonitor, 'get_associated_resource')
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_delete(self, mock_get, mock_is_running,
mock_assoc_res):
task = {
"uri": "uri",
"type": "TaskResourceV2",
"name": "Delete",
"taskState": "Completed",
}
mock_is_running.return_value = False
mock_get.return_value = task
mock_assoc_res.return_value = task.copy(), {"resource": "resource1"}
ret = self.task_monitor.wait_for_task(task.copy())
# may return a different type
self.assertEqual(True, ret)
@mock.patch.object(TaskMonitor, 'get_associated_resource')
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_remove(self, mock_get, mock_is_running,
mock_assoc_res):
task = {
"uri": "uri",
"type": "TaskResourceV2",
"name": "Remove",
"taskState": "Removed",
}
mock_is_running.return_value = False
mock_get.return_value = task
mock_assoc_res.return_value = task.copy(), {"resource": "resource1"}
ret = self.task_monitor.wait_for_task(task.copy())
self.assertEqual(True, ret)
@mock.patch.object(TaskMonitor, 'get_associated_resource')
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_remove_san_manager(self, mock_get, mock_is_running,
mock_assoc_res):
task = {
"uri": "uri",
"type": "TaskResourceV2",
"name": "Remove SAN manager",
"taskState": "Completed",
}
mock_is_running.return_value = False
mock_get.return_value = task
mock_assoc_res.return_value = task.copy(), {"resource": "resource1"}
ret = self.task_monitor.wait_for_task(task.copy())
self.assertEqual(True, ret)
@mock.patch.object(TaskMonitor, 'get_associated_resource')
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_wait_for_task_delete_server_hardware_type(self, mock_get,
mock_is_running,
mock_assoc_res):
task = {
"uri": "uri",
"type": "TaskResourceV2",
"name": "Delete server hardware type",
"taskState": "Completed",
}
mock_is_running.return_value = False
mock_get.return_value = task
mock_assoc_res.return_value = task.copy(), {"resource": "resource1"}
ret = self.task_monitor.wait_for_task(task.copy())
self.assertEqual(True, ret)
@mock.patch.object(connection, 'get')
def test_get(self, mock_get):
self.task_monitor.get({"uri": "an uri"})
mock_get.assert_called_once_with("an uri")
@mock.patch.object(TaskMonitor, 'is_task_running')
@mock.patch.object(TaskMonitor, 'get')
def test_get_completed_task(self, mock_get, mock_is_running):
task = {
"uri": "uri",
"type": "TaskResourceV2",
"name": "Create unexpected zoning report'",
"taskState": "Completed",
"taskOutput": []
}
mock_is_running.return_value = False
mock_get.return_value = task
response = self.task_monitor.get_completed_task(task.copy())
self.assertEqual(task, response)