def test_find_by_preferred(self):
""" Find a Supplier by Preferred """
suppliers = SupplierFactory.create_batch(3)
suppliers[0].preferred = "false"
suppliers[0].create()
suppliers[1].name = "Mark Jacob"
suppliers[1].preferred = "true"
suppliers[1].create()
suppliers[2].preferred = "false"
suppliers[2].create()
results = Supplier.find_by_preferred("true")
logging.debug(results)
self.assertNotEqual(results, [])
self.assertEqual(results[0].name, "Mark Jacob")
self.assertEqual(results[0].preferred, "true")
def get(self):
""" Returns all of the Suppliers """
app.logger.info('Request to list Suppliers...')
suppliers = []
args = supplier_args.parse_args()
if args['category']:
app.logger.info('Filtering by category: %s', args['category'])
suppliers = Supplier.find_by_category(args['category'])
elif args['name']:
app.logger.info('Filtering by name: %s', args['name'])
suppliers = Supplier.find_by_name(args['name'])
elif args['preferred'] is not None:
app.logger.info('Filtering by preferred: %s', args['preferred'])
suppliers = Supplier.find_by_preferred(args['preferred'])
else:
suppliers = Supplier.all()
app.logger.info('[%s] Suppliers returned', len(suppliers))
results = [supplier.serialize() for supplier in suppliers]
return results, status.HTTP_200_OK