def test_insert(empty_warehouse):
"""
Inserts a fact with MAX_ITERATIONS ^ 2 rows.
"""
enable_logging()
Warehouse.use(empty_warehouse)
Store.build()
stores = _get_instances('store')
Store.insert(*stores)
Product.build()
products = _get_instances('product')
Product.insert(*products)
Sales.build()
sales = _get_instances('sales')
start_time = datetime.datetime.now()
print 'Starting bulk insert of fact at ', start_time
try:
Sales.insert(*sales)
except OperationalError:
pytest.fail('The connection broke.')
end_time = datetime.datetime.now()
print 'Ending bulk insert of fact at ', end_time
delta = end_time - start_time
print 'Time taken = ', delta
def _generate_sales():
instances = []
for store_id in xrange(1, MAX_ITERATIONS):
for product_id in xrange(1, MAX_ITERATIONS):
obj = Sales()
obj['product'] = product_id
obj['store'] = store_id
instances.append(obj)
return instances
def test_create_duplicate_trigger(empty_warehouse):
""" Make sure a trigger isn't created if one already exists.
"""
Warehouse.use(empty_warehouse)
Sales.build()
assert not Sales.create_trigger()
def test_create_trigger(empty_warehouse):
""" Make sure a trigger can be created.
"""
Warehouse.use(empty_warehouse)
Sales.build()
assert Sales.trigger_name in Warehouse.trigger_names
