def test_find_by_recommend_type(self):
""" Test find by recommend_type """
Recommendation(product_id=PS3,
recommended_product_id=CONTROLLER,
recommendation_type="accessory").save()
Recommendation(product_id=PS4,
recommended_product_id=CONTROLLER,
recommendation_type="accessory").save()
Recommendation(product_id=PS4,
recommended_product_id=MONSTER_HUNTER,
recommendation_type="cross-sell").save()
recommendations = Recommendation.find_by_recommend_type("accessory")
self.assertEqual(len(recommendations), 2)
self.assertEqual(recommendations[0].recommendation_type, "accessory")
self.assertEqual(recommendations[1].recommendation_type, "accessory")
def query_recommendations_by_recommendation_type(recommendation_type):
""" Query a recommendation from the database that have the same recommendation type """
recommendations = Recommendation.find_by_recommend_type(
str(recommendation_type))
if len(recommendations) > 0:
message = [
recommendation.serialize() for recommendation in recommendations
]
return_code = HTTP_200_OK
else:
message = {
'error':
'Recommendation with product_id: \
%s was not found' % str(recommendation_type)
}
return_code = HTTP_404_NOT_FOUND
return message, return_code
