def tearDown(self): """Teardown of test database.""" mongo = MongoDBConnection() with mongo: test_database = mongo.connection[TEST_DATABASE] test_database['rentals'].drop() test_database['customers'].drop() test_database['products'].drop()
def tearDown(cls): ''' Drop mongo collections after each test to ensure a fresh start ''' mongo = MongoDBConnection() with mongo: db = mongo.connection.HPNorton db['product'].drop() db['customer'].drop() db['rentals'].drop()
def database_setup(): """function for setting up clean table each time""" mongo = MongoDBConnection() with mongo: database = mongo.connection.myDB for collection in database.list_collection_names(): database[collection].drop() return database
def clear_db(): mongo = MongoDBConnection() with mongo: db = mongo.connection.media product_db = db['product'] customer_db = db['customer'] rentals_db = db['rentals'] product_db.drop() customer_db.drop() rentals_db.drop()
def clear_db(): '''func to clear database of entries''' mongo = MongoDBConnection() with mongo: db = mongo.connection.media product_db = db['product'] customer_db = db['customer'] rentals_db = db['rentals'] product_db.drop() customer_db.drop() rentals_db.drop()
def test_write_many_to_database(self): """Unit test of the write_many_to_database function.""" entry_count = write_many_to_database(TEST_DATABASE, 'write_test', deepcopy(rental_list)) self.assertEqual(entry_count, 8) result_list = [] mongo = MongoDBConnection() with mongo: query = mongo.connection[TEST_DATABASE]['write_test'].find() for item in query: del item['_id'] result_list.append(item) mongo.connection[TEST_DATABASE]['write_test'].drop() self.assertListEqual(result_list, rental_list)
def setUp(self): """Setup for all unit tests""" self.product_fields = ('product_id', 'description', 'product_type', 'quantity_available') self.customer_fields = ('customer_id', 'name', 'address', 'phone_number', 'email') self.rental_fields = ('rental_id', 'product_id', 'customer_id', 'rental_start', 'rental_end') for item in product_list: item['quantity_available'] = int(item['quantity_available']) mongo = MongoDBConnection() with mongo: test_database = mongo.connection[TEST_DATABASE] test_database['rentals'].insert_many(deepcopy(rental_list)) test_database['customers'].insert_many(deepcopy(customer_list)) test_database['products'].insert_many(deepcopy(product_list))
def setup_db(request): mongo = MongoDBConnection() with mongo: db = mongo.connection.furniture db.product_data.drop() db.customer_data.drop() db.rental_data.drop() files = ["customer_data.csv", "product_data.csv", "rental_data.csv"] test_import_result = [import_data("dat", file) for file in files] def cleanup(): db.product_data.drop() db.customer_data.drop() db.rental_data.drop() request.addfinalizer(cleanup)
def test_connection(): with pytest.raises(AttributeError): mongo_test = MongoDBConnection(host='') with mongo_test: db = mongo_test.connection.furniture