def test_pool_management(user_client):
create_from_file(user_client, 'europe.xls')
user_client.post('/objects/process_pool', data=pool1)
user_client.post('/objects/process_pool', data=pool2)
p1, p2 = get_obj(Pool, name='pool1'), get_obj(Pool, name='pool2')
assert len(p1.nodes) == 21
assert len(p1.links) == 20
assert len(p2.nodes) == 12
assert len(p2.links) == 4
assert len(Pool.query.all()) == 5
user_client.post('/objects/delete_pool/{}'.format(p1.id))
user_client.post('/objects/delete_pool/{}'.format(p2.id))
assert len(Pool.query.all()) == 3
def test_link_deletion(user_client):
create_from_file(user_client, 'europe.xls')
for link_name in links:
link = get_obj(Link, name=link_name)
user_client.post('/objects/delete/link/{}'.format(link.id))
assert len(Node.query.all()) == 33
assert len(Link.query.all()) == 38
def test_node_deletion(user_client):
create_from_file(user_client, 'europe.xls')
for node_name in nodes:
node = get_obj(Node, name=node_name)
user_client.post('/objects/delete/node/{}'.format(node.id))
assert len(Node.query.all()) == 18
assert len(Link.query.all()) == 18
def save_log_rule():
tasks = [
get_obj(ScheduledTask, id=id) for id in request.form.getlist('tasks')
]
log_rule = log_rule_factory(**request.form.to_dict())
log_rule.tasks = tasks
db.session.commit()
return jsonify(log_rule.serialized)
def test_user_management(user_client):
for user in ('user1', 'user2', 'user3'):
dict_user = {
'name': user,
'email': '{}@test.com'.format(user),
'access_rights': 'Read-only',
'password': user,
}
user_client.post('/admin/process_user', data=dict_user)
assert len(User.query.all()) == 4
# user deletion
user1 = get_obj(User, name='user1')
user_client.post('/admin/delete_{}'.format(user1.id))
assert len(User.query.all()) == 3
def get(self, task_name):
task = get_obj(Task, name=task_name)
task.run()
return {'result': task.serialized}
def get(self, class_name, object_name):
return get_obj(diagram_classes[class_name], name=object_name).serialized
def get_log_rule(log_rule_id):
log_rule = get_obj(LogRule, id=log_rule_id)
return jsonify(log_rule.serialized)
def delete_log(log_id):
log = get_obj(Log, id=log_id)
db.session.delete(log)
db.session.commit()
return jsonify({'success': True})
def delete(self, class_name, object_name):
obj = get_obj(diagram_classes[class_name], name=object_name)
db.session.delete(obj)
db.session.commit()
return '{} {} successfully deleted'.format(class_name, object_name)
def get(self, task_name):
task = get_obj(Task, name=task_name)
runtime = task.run(run_now=True)
return {'task': task.serialized, 'id': runtime}
