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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
# This file is part of rhizi, a collaborative knowledge graph editor.
# Copyright (C) 2014-2015 Rhizi
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
Various test utilities
"""
import json
import string
import db_controller as dbc
from model.graph import Topo_Diff
from model.model import Link, RZDoc
from neo4j_test_util import rand_label
from neo4j_util import generate_random_id__uuid, generate_random_rzdoc_id
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.db_base_url)
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 gen_random_name(size=8, char_set=string.ascii_uppercase + string.digits):
"""
used for random node generation
"""
return ''.join(random.choice(char_set) for _ in range(size))
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 generate_random_diff__topo__minimal(test_label):
"""
@return: a ~minimal Topo_Diff containing three nodes and two links
"""
n_0, n_0_id = generate_random_node_dict(test_label)
n_1, n_1_id = generate_random_node_dict(test_label)
n_2, n_2_id = generate_random_node_dict(test_label)
l_0, l_0_id = generate_random_link_dict(test_label, n_0_id, n_1_id)
l_1, l_1_id = generate_random_link_dict(test_label, n_0_id, n_2_id)
n_set = [n_0, n_1, n_2]
l_set = [l_0, l_1]
topo_diff = Topo_Diff(node_set_add=n_set,
link_set_add=l_set)
return topo_diff
def generate_random_RZDoc(rzdoc_name=None):
if rzdoc_name is None:
rzdoc_name = gen_random_name()
rzdoc = RZDoc(rzdoc_name)
rzdoc.id = neo4j_util.generate_random_rzdoc_id()
return rzdoc
def gen_random_user_signup(self):
seed = gen_random_name()
us_req = User_Signup_Request(rz_username='rz_username_%s' % (seed),
email_address='%s@localhost' % (seed),
first_name='firstname%s' % (seed),
last_name='lastname%s' % (seed),
pw_plaintxt='aaaa12345')
return us_req
|
