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