def test_json_to_mongo(self):
'''
Test method to add JSON data to mongo database
'''
product_counts = add_json_to_mongodb(self.product_json, 'product')
customer_counts = add_json_to_mongodb(self.customer_json, 'customer')
rentals_counts = add_json_to_mongodb(self.rentals_json, 'rentals')
self.assertEqual(product_counts, (10, 0))
self.assertEqual(customer_counts, (7, 0))
self.assertEqual(rentals_counts, (8, 0))
def test_bad_connection(self):
'''
Test method to ensure bad mongo DB connection is properly handled
'''
mongo = create_mongo_connection(host='127.0.0.1', port=27018)
item_counts = add_json_to_mongodb([{'test':'test'}], 'zzz', mongo)
self.assertEqual(item_counts, (0, 1))