def test_get_all_by_user_id_data_ordering(self):
# Arrange
user_id = 1
ascending_order_by_field = ["+title"]
descending_order_by_field = ["-title"]
# Act
ascending_result = Data.get_all_by_user_id(user_id, ascending_order_by_field)
descending_result = Data.get_all_by_user_id(user_id, 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_get_all_by_user_id_multi_field_sorting(self):
# Arrange
ascending_order_by_multi_field = ["+user_id", "+title"]
descending_order_by_multi_field = ["+user_id", "-title"]
# Act
ascending_result = Data.get_all_by_user_id(1, ascending_order_by_multi_field)
descending_result = Data.get_all_by_user_id(1, descending_order_by_multi_field)
# Assert
self.assertEqual(self.fixture.data_1.title, ascending_result.all()[0].title)
self.assertEqual(self.fixture.data_3.title, ascending_result.all()[1].title)
self.assertEqual(self.fixture.data_3.title, descending_result.all()[1].title)
self.assertEqual(self.fixture.data_1.title, descending_result.all()[2].title)
def test_data_get_all_by_user_id_return_collection_of_data_from_user(self):
# Arrange
user_id = 1
# Act
result = Data.get_all_by_user_id(user_id)
# Assert
self.assertTrue(all(item.user_id == str(user_id) for item in result))
def test_data_get_all_by_user_id_return_empty_collection_of_data_from_user_does_not_exist(self):
# Arrange
user_id = 800
# Act
result = Data.get_all_by_user_id(user_id, DATA_SORTING_FIELDS)
# Assert
self.assertTrue(result.count() == 0)
def test_get_all_by_user_id_data_descending_sorting(self):
# Arrange
descending_order_by_field = ["-title"]
user_id = 1
# Act
descending_result = Data.get_all_by_user_id(user_id, descending_order_by_field)
# Assert
self.assertTrue(self.fixture.data_3.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_get_all_by_user_id_data_ascending_sorting(self):
# Arrange
ascending_order_by_field = ["+title"]
user_id = 1
# Act
ascending_result = Data.get_all_by_user_id(user_id, ascending_order_by_field)
# Assert
self.assertTrue(self.fixture.data_1.title == ascending_result.all()[0].title)
self.assertTrue(self.fixture.data_3.title == ascending_result.all()[1].title)
def get_all_by_user(user, order_by_field=DATA_SORTING_FIELDS):
"""Return all data owned by a user.
Parameters:
user:
order_by_field: Order by field.
Returns: data collection
"""
return Data.get_all_by_user_id(str(user.id), order_by_field)
