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