server-tests/test_util.py

1 files changed, 32 insertions, 0 deletions

diff --git a/src/server-tests/test_util.py b/src/server-tests/test_util.py
new file mode 100644
index 00000000..2c8bb294
--- /dev/null
+++ b/src/server-tests/test_util.py
@@ -0,0 +1,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 }