def test_show_rentals(mongo_database):
d.import_data(mongo_database, '', 'product.csv', 'customer.csv',
'rental.csv')
result = d.show_rentals(mongo_database, 'P000004')
expected = {
'C000002': {
'first_name': 'Blanca',
'last_name': 'Bashirian',
'address': '0193 Malvina Lake',
'phone_number': '(240)014-9496 x08349',
'email': '*****@*****.**',
'status': 'Active',
'credit_limit': '689'
},
'C000004': {
'first_name': 'Mittie',
'last_name': 'Turner',
'address': '996 Lorenza Points',
'phone_number': '1-324-023-8861 x025',
'email': '*****@*****.**',
'status': 'Active',
'credit_limit': '565'
}
}
assert len(result) == 2
assert list(result.keys()) == ['C000002', 'C000004']
assert result == expected
def test_show_rentals(mongo_database):
d.import_data(mongo_database, '', 'product.csv', 'customer.csv', 'rental.csv')
result = d.show_rentals(mongo_database, 'P000004')
assert len(result) == 2
assert list(result.keys()) == ['C000002', 'C000004']
def test_show_available_products(mongo_database):
d.import_data(mongo_database, '', 'product.csv', 'customer.csv', 'rental.csv')
result = d.show_available_products(mongo_database)
assert len(result) == 9999
assert 'P000001' in result
assert 'P010999' not in result
def test_clear_data(mongo_database):
d.import_data(mongo_database, '', 'product.csv', 'customer.csv', 'rental.csv')
result = sorted(mongo_database.list_collection_names())
assert result == ['customers', 'products', 'rentals']
d.clear_data(mongo_database)
result2 = mongo_database.list_collection_names()
assert result2 == []
def test_import_data(mongo_database):
result = d.import_data(mongo_database, '', 'product.csv', 'customer.csv',
'rental.csv')
assert result[0][0] == 9999
assert result[0][1] == 0
assert result[0][2] == 9999
assert isinstance(result[0][3], float)
assert result[1][0] == 9999
assert result[1][1] == 0
assert result[1][2] == 9999
assert isinstance(result[1][3], float)