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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
"""
Various test utilities
"""
import json
import db_controller as dbc
from model.graph import Topo_Diff
from model.model import Link
from neo4j_test_util import gen_random_name, rand_label
from neo4j_util import generate_random_id__uuid
import neo4j_util
from rz_kernel import RZ_Kernel
from rz_mesh import init_ws_interface
from rz_server import init_webapp
import test_rz_mesh
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
@return: a dict based node object representation and the generated node id
"""
if None == nid:
nid = generate_random_id__uuid()
return {'__label_set': [n_type],
'id': nid,
'name': gen_random_name() }, nid
def generate_random_link_dict(l_type, src_id, dst_id, lid=None):
"""
@param l_type: is converted to a single item type array
@return: a dict based node object representation and the generated node id
"""
if None == lid:
lid = generate_random_id__uuid()
ret_dict = Link.link_ptr(src_id, dst_id)
ret_dict['__type'] = [l_type]
ret_dict['id'] = lid
return ret_dict, lid
def ws_emit__topo_diff():
import logging
r_label = rand_label()
n, n_id = generate_random_node_dict(r_label)
topo_diff = Topo_Diff(node_set_add=[n])
data = json.dumps(topo_diff, cls=Topo_Diff.JSON_Encoder)
with test_rz_mesh.RZ_websocket() as (_, ns_sock):
ns_sock.emit('diff_commit__topo', data)
|
