def tearDown(self): mongo = database.MongoDBConnection() with mongo: db = mongo.connection.test_database db['customers'].drop() db['products'].drop() db['rentals'].drop()
def setUp(self): # really wish there was a way to do this easily in memory mongo = database.MongoDBConnection() with mongo: db = mongo.connection.test_database db['customers'].drop() db['products'].drop() db['rentals'].drop()
def mongo_database(): """Creates a MongoDB. """ mongo = l.MongoDBConnection() with mongo: database = mongo.connection.media yield database l.clear_data(database)
def mongo_database(): mongo = d.MongoDBConnection() with mongo: db = mongo.connection.media yield db d.clear_data(db)
def mongo_database(): ''' Creating the MongoDB for testing. ''' mongo = l.MongoDBConnection() with mongo: db = mongo.connection.media yield db l.clear_data(db)
def mongo_database(): """ Creates a MongoDB. :return: """ mongo = d.MongoDBConnection() with mongo: db = mongo.connection.media yield db d.cleanup()
def mongo_database(): """ Creates a MongoDB for testing. """ mongo = d.MongoDBConnection() with mongo: db = mongo.connection.media test_utilities = d.Database() yield (db, test_utilities) test_utilities.clear_data(db)
def mongo_database(): ''' Creates a MongoDB. ''' mongo = d.MongoDBConnection() with mongo: db = mongo.connection.media test_utilities = d.Database() yield (db, test_utilities) test_utilities.clear_data(db)
def setUp(self): """Clean up database before each test.""" mongo = database.MongoDBConnection() with mongo: hp_db = mongo.connection.HPNortonDatabase product_data = hp_db['product_data'] customer_data = hp_db['customer_data'] rental_data = hp_db['rental_data'] # drop all data in collections product_data.drop() customer_data.drop() rental_data.drop()
def setUp(self): ''' Sets up database for each test ''' mongo = database.MongoDBConnection() with mongo: db = mongo.connection.HPNortonDatabase products = db['products'] customers = db['customers'] rentals = db['rentals'] products.drop() customers.drop() rentals.drop()
def test_import_data(self): """ Test import_data() function """ db.LOGGER.info('--- Start Test import_data() ---') actual_res = db.import_data(self.csv_file, self.product_file, self.customer_file, self.rental_file) client = db.MongoDBConnection() with client: hp_norton_db = client.connection.rental products = hp_norton_db['products'] customers = hp_norton_db['customers'] rentals = hp_norton_db['rentals'] self.assertEqual(products.count(), 5) self.assertEqual(customers.count(), 5) self.assertEqual(rentals.count(), 9) expect_res = [(5, 5, 9), (0, 0, 0)] self.assertEqual(actual_res, expect_res) db.LOGGER.info('--- End Test import_data() ---')
def setUp(self): mongo = db.MongoDBConnection() with mongo: db.DATABASE = mongo.connection.test_rentals db.drop_data()