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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
# 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/>.
import logging
import time
import unittest
import db_controller
from db_op import DBO_factory__default
import neo4j_test_util
from rz_config import RZ_Config
from rz_kernel import RZ_Kernel
import test_util
from test_util__pydev import debug__pydev_pd_arg
class TestRZDoc(unittest.TestCase):
@classmethod
def setUpClass(self):
cfg = RZ_Config.init_from_file('res/etc/rhizi-server.conf')
self.db_ctl = db_controller.DB_Controller(cfg.db_base_url)
self.log = logging.getLogger('rhizi')
self.log.addHandler(logging.StreamHandler())
# bootstrap kernel
kernel = RZ_Kernel()
kernel.db_ctl = self.db_ctl
kernel.db_op_factory = DBO_factory__default()
kernel.start()
self.kernel = kernel
while not self.kernel.is_DB_status__ok(): # wait for kernel to initialize...
time.sleep(0.3)
def setUp(self): pass
def test_rzdoc_commit_log(self):
rzdoc_a = 'test_commit_log_doc_a'
rzdoc_b = 'test_commit_log_doc_b'
for rzdoc_name in [rzdoc_a, rzdoc_b]:
lookup_ret = self.kernel.rzdoc__lookup_by_name(rzdoc_name)
if lookup_ret != None:
self.kernel.rzdoc__delete(lookup_ret)
ret_create = self.kernel.rzdoc__create(rzdoc_a)
ret_create = self.kernel.rzdoc__create(rzdoc_b)
node_a, _ = test_util.generate_random_node_dict('type_a')
node_b, _ = test_util.generate_random_node_dict('type_b')
topo_diff_a = Topo_Diff(node_set_add=[node_a], meta={'sentence': 'a'})
topo_diff_b = Topo_Diff(node_set_add=[node_b], meta={'sentence': 'b'})
class FakeRZDoc(object):
def __init__(self, id):
self.id = id
class FakeContext(object):
def __init__(self, id):
self.rzdoc = FakeRZDoc(id)
self.user_name = None
ctx_a = FakeContext(rzdoc_a)
ctx_b = FakeContext(rzdoc_b)
self.kernel.diff_commit__topo(topo_diff=topo_diff_a, ctx=ctx_a)
self.kernel.diff_commit__topo(topo_diff=topo_diff_b, ctx=ctx_b)
commit_log = self.kernel.rzdoc__commit_log(rzdoc=ctx_a, limit=10)
def test_rzdoc_lifecycle(self):
test_label = neo4j_test_util.rand_label()
rzdoc_name = test_label
lookup_ret__by_name = self.kernel.rzdoc__lookup_by_name(rzdoc_name)
self.assertIsNone(lookup_ret__by_name)
# create
rzdoc, m_rzdoc = self.kernel.rzdoc__create(rzdoc_name)
# lookup
for rzd in [rzdoc, m_rzdoc]:
lookup_ret__by_id = self.kernel.rzdoc__lookup_by_id(rzd.id)
lookup_ret__by_name = self.kernel.rzdoc__lookup_by_name(rzd.name)
self.assertTrue(None != lookup_ret__by_id)
self.assertTrue(None != lookup_ret__by_name)
# delete
self.kernel.rzdoc__delete(rzdoc)
lookup_ret__by_name = self.kernel.rzdoc__lookup_by_name(rzdoc_name)
self.assertIsNone(lookup_ret__by_name)
def test_rzdoc_search(self):
rzdoc_common_name = test_util.gen_random_name()
rzdoc_post_a = test_util.generate_random_RZDoc(rzdoc_common_name + '_a')
rzdoc_post_b = test_util.generate_random_RZDoc(rzdoc_common_name + '_b')
rzdoc_pre_c = test_util.generate_random_RZDoc('c_' + rzdoc_common_name)
rzdoc_plain = test_util.generate_random_RZDoc(rzdoc_common_name)
for rzdoc in [rzdoc_post_a, rzdoc_post_b, rzdoc_pre_c, rzdoc_plain]:
self.kernel.rzdoc__create(rzdoc_name=rzdoc.name)
ret = self.kernel.rzdoc__search(rzdoc_common_name)
self.assertEquals(4, len(ret))
ret = self.kernel.rzdoc__search(rzdoc_common_name + '_')
self.assertEquals(2, len(ret))
ret = self.kernel.rzdoc__search('c_' + rzdoc_common_name)
self.assertEquals(1, len(ret))
@classmethod
def tearDownClass(self):
self.kernel.shutdown()
@debug__pydev_pd_arg
def main():
unittest.main(defaultTest='TestRZDoc.test_rzdoc_lifcycle', verbosity=2)
if __name__ == "__main__":
main()
|
