def test_retrieve_one(self):
mock_task = MagicMock(name='mocked_task')
attr = {'duedate': None,
'star': None,
'priority': 5,
'status': 1,
'title': 'retrieve one'
}
mock_task.configure_mock(**attr)
del mock_task.completed
self.apic.return_value.getTasks.return_value = [mock_task, ]
hotlist = tasks.HotList()
result = list(hotlist.retrieve())
self.assertEqual(result, [mock_task])
def test_is_not_next_action(self):
self.apic.return_value.getAccountInfo.return_value.hotliststatus = '1'
mock_task = MagicMock(name='mocked_task')
attr = {'duedate': None,
'star': None,
'priority': 5,
'status': 2,
'title': 'retrieve one'
}
mock_task.configure_mock(**attr)
del mock_task.completed
self.apic.return_value.getTasks.return_value = [mock_task, ]
hotlist = tasks.HotList()
result = list(hotlist.retrieve())
self.assertEqual(result, [])
def test_retrieve_multi(self):
mock_task_1st = MagicMock(name='mocked_task_1st')
attr_1st = {'duedate': None,
'star': None,
'priority': 5,
'status': 1,
'title': 'retrieve one'
}
mock_task_1st.configure_mock(**attr_1st)
del mock_task_1st.completed
mock_task_2nd = MagicMock(name='mocked_task_2nd')
attr_2nd = {'duedate': None,
'star': None,
'priority': 5,
'status': 1,
'title': 'retrieve two'
}
mock_task_2nd.configure_mock(**attr_2nd)
del mock_task_2nd.completed
self.apic.return_value.getTasks.return_value = [mock_task_1st, mock_task_2nd]
hotlist = tasks.HotList()
result = list(hotlist.retrieve())
self.assertEqual(result, [mock_task_1st, mock_task_2nd])
def test_retrieve_empty(self):
self.apic.return_value.getTasks.return_value = []
hotlist = tasks.HotList()
result = list(hotlist.retrieve())
self.assertEqual(result, [])
def test_retrieve_error(self):
self.apic.return_value.getTasks.side_effect = Exception('error on server')
hotlist = tasks.HotList()
with self.assertRaisesRegexp(Exception, r'error on server'):
result = list(hotlist.retrieve())