def test_store_result_with_request(self, mock_get_database):
self.backend.taskmeta_collection = MONGODB_COLLECTION
mock_database = MagicMock(spec=['__getitem__', '__setitem__'])
mock_collection = Mock()
mock_request = MagicMock(spec=['parent_id'])
mock_get_database.return_value = mock_database
mock_database.__getitem__.return_value = mock_collection
mock_request.parent_id = sentinel.parent_id
ret_val = self.backend._store_result(sentinel.task_id,
sentinel.result,
sentinel.status,
request=mock_request)
mock_get_database.assert_called_once_with()
mock_database.__getitem__.assert_called_once_with(MONGODB_COLLECTION)
parameters = mock_collection.replace_one.call_args[0][1]
assert parameters['parent_id'] == sentinel.parent_id
assert sentinel.result == ret_val
mock_collection.replace_one.side_effect = InvalidDocument()
with pytest.raises(EncodeError):
self.backend._store_result(sentinel.task_id, sentinel.result,
sentinel.status)
def test_store_result(self, mock_get_database):
self.backend.taskmeta_collection = MONGODB_COLLECTION
mock_database = MagicMock(spec=['__getitem__', '__setitem__'])
mock_collection = Mock()
mock_get_database.return_value = mock_database
mock_database.__getitem__.return_value = mock_collection
ret_val = self.backend._store_result(sentinel.task_id, sentinel.result,
sentinel.status)
mock_get_database.assert_called_once_with()
mock_database.__getitem__.assert_called_once_with(MONGODB_COLLECTION)
mock_collection.save.assert_called_once_with(ANY)
self.assertEqual(sentinel.result, ret_val)
mock_collection.save.side_effect = InvalidDocument()
with self.assertRaises(EncodeError):
self.backend._store_result(sentinel.task_id, sentinel.result,
sentinel.status)