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