summaryrefslogtreecommitdiff
path: root/src/server-tests/test_util.py
blob: 2c8bb29430f5937f03a61ffcbee3fd9d84e19620 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
"""
Various test utilities
"""
from rhizi_server import init_webapp
from rz_mesh import init_ws_interface
import db_controller as dbc

from neo4j_test_util import rand_id
from rz_kernel import RZ_Kernel

def init_test_db_controller(cfg):
    ret = dbc.DB_Controller(cfg)
    return ret

def init_test_ws_server(cfg, db_ctl):
    """
    Initialize a test websocket server

    @param db_ctl: an initialized DB_Controller
    """
    kernel = RZ_Kernel()
    webapp = init_webapp(cfg, kernel, db_ctl)
    ws_srv = init_ws_interface(cfg, webapp)
    return ws_srv

def generate_random_node_dict(n_type, nid=None):
    """
    @param n_type: is converted to a label set
    """
    if None == nid:
        nid = rand_id()
    return {'__label_set': ['T_' + n_type], 'id': nid }