def main(argv=None):
"""Database main function
"""
args = parse_cmd_arguments(argv)
if args.disable_log:
logger.disable("__main__")
logger.disable("src.database_operations")
if args.all_products:
pprint(db.list_all_products())
elif args.available_products:
pprint(db.show_available_products())
elif args.all_customers:
pprint(db.list_all_customers())
elif args.drop_collections:
db.drop_collections()
elif args.drop_database:
db.drop_database()
elif args.rentals_for_customer:
pprint(db.rentals_for_customer(args.rentals_for_customer))
elif args.customers_renting_product:
pprint(db.show_rentals(args.customers_renting_product))
elif args.insert:
pprint(db.import_data(*args.insert))
def test_rentals_for_customer(_list_rentals_for_customer):
""" rentals for customers """
l.drop_database()
l.insert_to_mongo("./data/customers.csv")
l.insert_to_mongo("./data/product.csv")
l.insert_to_mongo("./data/rental.csv")
my_response = l.rentals_for_customer("C000001")
assert my_response == _list_rentals_for_customer
def test_rentals_for_customer(_list_rentals_for_customer):
""" rentals for customers """
my_response = l.rentals_for_customer("user002")
assert my_response == _list_rentals_for_customer