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))