def test_update_task_redundancy_connection_failed(self):
"""Test update task redundancy connection fails works."""
pbclient = MagicMock()
pbclient.find_project.side_effect = exceptions.ConnectionError
self.config.pbclient = pbclient
res = _update_tasks_redundancy(self.config, 1, 5)
assert res == "Connection Error! The server http://server is not responding", res
def test_update_task_redundancy_connection_failed(self):
"""Test update task redundancy connection fails works."""
pbclient = MagicMock()
pbclient.find_project.side_effect = exceptions.ConnectionError
self.config.pbclient = pbclient
res = _update_tasks_redundancy(self.config, 1, 5)
assert res == "Connection Error! The server http://server is not responding", res
def test_update_task_redundancy_individually(self, find_mock):
"""Test update task redundancy individually works."""
project = MagicMock()
project.name = 'name'
project.short_name = 'short_name'
project.description = 'description'
project.info = dict()
find_mock.return_value = project
pbclient = MagicMock()
self.config.pbclient = pbclient
res = _update_tasks_redundancy(self.config, 1, 5)
msg = "Task.id = 1 redundancy has been updated to 5"
assert res == msg, res
def test_update_task_redundancy_all_tasks(self, find_mock):
"""Test update task redundancy all tasks works."""
project = MagicMock()
project.name = 'name'
project.short_name = 'short_name'
project.description = 'description'
project.info = dict()
find_mock.return_value = project
pbclient = MagicMock()
pbclient.get_tasks = self.fake_return_tasks
self.config.pbclient = pbclient
res = _update_tasks_redundancy(self.config, None, 5)
msg = "All tasks redundancy have been updated"
assert res == msg, res
def test_update_task_redundancy_individually(self, auto_mock, find_mock):
"""Test update task redundancy individually works."""
auto_mock.return_value = (0, None)
project = MagicMock()
project.name = 'name'
project.short_name = 'short_name'
project.description = 'description'
project.info = dict()
find_mock.return_value = project
pbclient = MagicMock()
self.config.pbclient = pbclient
res = _update_tasks_redundancy(self.config, 1, 5)
msg = "Task.id = 1 redundancy has been updated to 5"
assert res == msg, res
def test_update_task_redundancy_all_tasks(self, auto_mock, find_mock):
"""Test update task redundancy all tasks works."""
auto_mock.return_value = (0, None)
project = MagicMock()
project.name = 'name'
project.short_name = 'short_name'
project.description = 'description'
project.info = dict()
find_mock.return_value = project
pbclient = MagicMock()
pbclient.get_tasks = self.fake_return_tasks
self.config.pbclient = pbclient
res = _update_tasks_redundancy(self.config, None, 5)
msg = "All tasks redundancy have been updated"
assert res == msg, res