def setUp(self):
sqldb.drop_all()
sqldb.create_all()
# Create an initial Team and Location to facilitate the relationship
team = Team(name='Test Team 1',
group='Accenture CIO',
description='Temporary Test Team')
sqldb.session.add(team)
location = Location(name='Ashburn',
city='Ashburn',
region='Virginia',
country='USA')
sqldb.session.add(location)
sqldb.session.commit()
def set_up(self):
sqldb.drop_all()
sqldb.create_all()
for x in range(1, 6):
name = 'Location {0}'.format(x)
city = 'Location City {0}'.format(x)
country = 'Location Country {0}'.format(x)
location = Location(name=name, city=city, country=country)
sqldb.session.add(location)
# Create Vm Records
hostname = 'vm {0}'.format(x)
vm = VM(hostname=hostname)
vm.location = location
sqldb.session.add(vm)
outlet_type = create_outlet_type(self, x)
pdu = create_pdu_records(self, x, outlet_type)
rack = create_rack_records(self, x, location, pdu=pdu)
create_power_outlet_records(self, x, rack, outlet_type)
sqldb.session.commit()
def tear_down(self):
sqldb.drop_all()
sqldb.session.remove()
def init_db():
sqldb.drop_all()
sqldb.create_all()
def clear_db():
sqldb.drop_all()