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}