def initial_db(request):
fresh_database(request)
with app.app_context():
alice = User(label='alice',
password='******',
is_admin=True)
bob = User(label='bob',
password='******',
is_admin=False)
db.session.add(alice)
db.session.add(bob)
runway = model.Project('runway')
runway.users.append(alice)
db.session.add(runway)
db.session.commit()
def initial_db(request):
fresh_database(request)
with app.app_context():
alice = User(label='alice',
password='******',
is_admin=True)
bob = User(label='bob',
password='******',
is_admin=False)
db.session.add(alice)
db.session.add(bob)
runway = model.Project('runway')
runway.users.append(alice)
db.session.add(runway)
db.session.commit()
def db(request):
session = fresh_database(request)
alice = User(label='alice',
password='******',
is_admin=True)
bob = User(label='bob',
password='******',
is_admin=False)
session.add(alice)
session.add(bob)
runway = model.Project('runway')
runway.users.append(alice)
session.add(runway)
session.commit()
return session
def db(request):
session = fresh_database(request)
# Create a couple projects:
runway = model.Project("runway")
manhattan = model.Project("manhattan")
for proj in [runway, manhattan]:
session.add(proj)
# ...including at least one with nothing in it:
session.add(model.Project('empty-project'))
# ...A variety of networks:
networks = [
{
'creator': None,
'access': None,
'allocated': True,
'label': 'stock_int_pub',
},
{
'creator': None,
'access': None,
'allocated': False,
'network_id': 'ext_pub_chan',
'label': 'stock_ext_pub',
},
{
# For some tests, we want things to initial be attached to a
# network. This one serves that purpose; using the others would
# interfere with some of the network_delete tests.
'creator': None,
'access': None,
'allocated': True,
'label': 'pub_default',
},
{
'creator': runway,
'access': runway,
'allocated': True,
'label': 'runway_pxe'
},
{
'creator': None,
'access': runway,
'allocated': False,
'network_id': 'runway_provider_chan',
'label': 'runway_provider',
},
{
'creator': manhattan,
'access': manhattan,
'allocated': True,
'label': 'manhattan_pxe'
},
{
'creator': None,
'access': manhattan,
'allocated': False,
'network_id': 'manhattan_provider_chan',
'label': 'manhattan_provider',
},
]
for net in networks:
if net['allocated']:
net['network_id'] = \
get_network_allocator().get_new_network_id(session)
session.add(model.Network(**net))
# ... Two switches. One of these is just empty, for testing deletion:
session.add(MockSwitch(label='empty-switch',
hostname='empty',
username='******',
password='******',
type=MockSwitch.api_name))
# ... The other we'll actually attach stuff to for other tests:
switch = MockSwitch(label="stock_switch_0",
hostname='stock',
username='******',
password='******',
type=MockSwitch.api_name)
# ... Some free ports:
session.add(model.Port('free_port_0', switch))
session.add(model.Port('free_port_1', switch))
# ... Some nodes (with projets):
nodes = [
{'label': 'runway_node_0', 'project': runway},
{'label': 'runway_node_1', 'project': runway},
{'label': 'manhattan_node_0', 'project': manhattan},
{'label': 'manhattan_node_1', 'project': manhattan},
{'label': 'free_node_0', 'project': None},
{'label': 'free_node_1', 'project': None},
]
for node_dict in nodes:
obm=MockObm(type=MockObm.api_name,
host=node_dict['label'],
user='******',
password='******')
node = model.Node(label=node_dict['label'], obm=obm)
node.project = node_dict['project']
session.add(model.Nic(node, label='boot-nic', mac_addr='Unknown'))
# give it a nic that's attached to a port:
port_nic = model.Nic(node, label='nic-with-port', mac_addr='Unknown')
port = model.Port(node_dict['label'] + '_port', switch)
port.nic = port_nic
# ... Some headnodes:
headnodes = [
{'label': 'runway_headnode_on', 'project': runway, 'on': True},
{'label': 'runway_headnode_off', 'project': runway, 'on': False},
{'label': 'runway_manhattan_on', 'project': manhattan, 'on': True},
{'label': 'runway_manhattan_off', 'project': manhattan, 'on': False},
]
for hn_dict in headnodes:
headnode = model.Headnode(hn_dict['project'],
hn_dict['label'],
'base-headnode')
headnode.dirty = not hn_dict['on']
hnic = model.Hnic(headnode, 'pxe')
session.add(hnic)
# Connect them to a network, so we can test detaching.
hnic = model.Hnic(headnode, 'public')
hnic.network = session.query(model.Network)\
.filter_by(label='pub_default').one()
# ... and at least one node with no nics (useful for testing delete):
obm=MockObm(type=MockObm.api_name,
host='hostname',
user='******',
password='******')
session.add(model.Node(label='no_nic_node', obm=obm))
session.commit()
return session
def db(request):
return fresh_database(request)
