def mongo_database():
"""Creates a MongoDB. """
mongo = l.MongoDBConnection()
with mongo:
database = mongo.connection.media
yield database
l.clear_data(database)
def test_clear_data(mongo_database):
d.import_data(mongo_database, '', 'product.csv', 'customer.csv',
'rental.csv')
result = sorted(mongo_database.list_collection_names())
assert result == ['customers', 'products', 'rentals']
d.clear_data(mongo_database)
result2 = mongo_database.list_collection_names()
assert result2 == []
def test_g_import_data(self):
#no data, expected to retuen no errors or imports
with patch('builtins.input', side_effect='Y'):
database.clear_data()
expected = ((0, 0, 0), (0, 0, 0))
actual = database.import_data(PATH, "blank_products.csv",
"blank_customers.csv",
"blank_rentals.csv")
self.assertEqual(actual, expected)
def test_f_import_data(self):
#partial/incomplete data, expected to return a mix
PATH = "C:\\_PythonClass\\AdvPython\\SP_Python220B_2019\\students\\bplanica\\lesson05\\assignment\\test input files\\"
with patch('builtins.input', side_effect='Y'):
database.clear_data()
expected = ((2, 3, 3), (2, 1, 1))
actual = database.import_data(PATH, "products.csv", "customers.csv",
"rentals.csv")
self.assertEqual(actual, expected)
def test_clear_data(mongo_database):
d.import_data(mongo_database, "", "products.csv", "customers.csv",
"rentals.csv")
result = mongo_database.list_collection_names()
assert result == ["products", "rentals", "customers"]
d.clear_data(mongo_database)
result2 = mongo_database.list_collection_names()
assert result2 == []
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 test_clear_data(mongo_database):
'''
Testing database clearing.
'''
l.import_data(mongo_database, '', 'product.csv', 'customers.csv',
'rental.csv')
result = mongo_database.list_collection_names()
assert result == ['products', 'rentals', 'customers']
l.clear_data(mongo_database)
result2 = mongo_database.list_collection_names()
assert result2 == []
def mongo_database():
"""
Creates a MongoDB.
:return:
"""
mongo = d.MongoDBConnection()
with mongo:
db = mongo.connection.media
yield db
d.clear_data(db)
def test_h_import_data(self):
with patch('builtins.input', side_effect='Y'):
database.clear_data()
with self.assertLogs(level='ERROR'):
database.import_data(PATH, "prod.csv", "cust.csv", "rent.csv")
def test_a_clear_data(self):
with patch('builtins.input', side_effect='Y'):
database.clear_data()
