def test_data_get_all_ordering(self):
# Arrange
access_control_data_fixture.insert_data()
ascending_order_by_field = ["+title"]
descending_order_by_field = ["-title"]
# Act
ascending_result = Data.get_all(ascending_order_by_field)
descending_result = Data.get_all(descending_order_by_field)
# Assert
for i in range(len(ascending_result)):
self.assertTrue(ascending_result.all()[i].title == descending_result.all()[len(ascending_result) - i - 1].title)
def test_data_get_all_multi_field_sorting(self):
# Arrange
ascending_order_by_multi_field = ["+title", "+user_id"]
descending_order_by_multi_field = ["+title", "-user_id"]
# Act
ascending_result = Data.get_all(ascending_order_by_multi_field)
descending_result = Data.get_all(descending_order_by_multi_field)
# Assert
self.assertEqual(self.fixture.data_4.user_id, ascending_result.all()[4].user_id)
self.assertEqual(self.fixture.data_5.user_id, ascending_result.all()[3].user_id)
self.assertEqual(self.fixture.data_4.user_id, descending_result.all()[3].user_id)
self.assertEqual(self.fixture.data_5.user_id, descending_result.all()[4].user_id)
def test_data_get_all_return_objects_data_in_collection(self):
# Act
db_content = Data.get_all(DATA_SORTING_FIELDS)
excluded_id_list = [str(db_content[0].pk)]
result = Data.get_all_except([], excluded_id_list)
# Assert
self.assertTrue(result.count() == len(self.fixture.data_collection) - len(excluded_id_list))
def test_data_get_all_except_return_collection_of_data(self):
# Act
db_content = Data.get_all(DATA_SORTING_FIELDS)
excluded_id_list = [str(db_content[0].pk)]
result = Data.get_all_except([], excluded_id_list)
# Assert
self.assertTrue(all(isinstance(item, Data) for item in result))
def test_data_get_all_descending_sorting(self):
# Arrange
descending_order_by_field = ["-title"]
# Act
descending_result = Data.get_all(descending_order_by_field)
# Assert
self.assertTrue(self.fixture.data_2.title == descending_result.all()[len(descending_result)-2].title)
self.assertTrue(self.fixture.data_1.title == descending_result.all()[len(descending_result)-1].title)
def test_data_get_all_ascending_sorting(self):
# Arrange
ascending_order_by_field = ["+title"]
# Act
ascending_result = Data.get_all(ascending_order_by_field)
# Assert
self.assertTrue(self.fixture.data_1.title == ascending_result.all()[0].title)
self.assertTrue(self.fixture.data_2.title == ascending_result.all()[1].title)
def check_data_info():
""" Check OAI-PMH data information.
"""
try:
data = Data.get_all()
for document in data:
oai_data_api.upsert_from_data(document, force_update=False)
except Exception, e:
print('ERROR : Impossible to init the OAI-PMH data : %s' % e.message)
def get_all(user, order_by_field=DATA_SORTING_FIELDS):
"""Get all the data if superuser. Raise exception otherwise.
Parameters:
user:
order_by_field: Order by field.
Returns: data collection
"""
return Data.get_all(order_by_field)
def test_data_get_all_except_inexistant_id_return_objects_data_in_collection(self):
# Act
object_id_list = [str(data.pk) for data in Data.get_all(DATA_SORTING_FIELDS)]
inexistant_object_id = str(ObjectId())
# If the generated object id correspond to one in DB we generate another one
while inexistant_object_id in object_id_list:
inexistant_object_id = str(ObjectId)
excluded_id_list = [inexistant_object_id]
result = Data.get_all_except([], excluded_id_list)
# Assert
self.assertTrue(result.count() == len(self.fixture.data_collection))
def test_data_get_all_except_inexistant_id_return_collection_of_data(self):
# Act
object_id_list = [str(data.pk) for data in Data.get_all(DATA_SORTING_FIELDS)]
inexistant_object_id = str(ObjectId())
# If the generated object id correspond to one in DB we generate another one
while inexistant_object_id in object_id_list:
inexistant_object_id = str(ObjectId)
excluded_id_list = [inexistant_object_id]
result = Data.get_all_except([], excluded_id_list)
# Assert
self.assertTrue(all(isinstance(item, Data) for item in result))
def generate_data_items_from_all_data_in_database():
""" Generate DataItem from all data in db
"""
try:
data = Data.get_all()
for document in data:
try:
logger.info('START processing data : %s' % document.id)
system_data_item_api.upsert_from_data(document,
force_update=False)
except Exception, e:
logger.error(
'ERROR : Impossible to init the DataItem data : %s' %
e.message)
logger.info('All data items are created')
def test_data_get_all_return_objects_data_in_collection(self):
# Act
result = Data.get_all()
# Assert
self.assertTrue(len(self.fixture.data_collection) == result.count())
def test_data_get_all_return_collection_of_data(self):
# Act
result = Data.get_all()
# Assert
self.assertTrue(all(isinstance(item, Data) for item in result))
