def main():
glb = rg_global.RgGlobal('sync_demo')
glb.connect_sync(rg_mom_common.Db.RGMOMD_HOST)
mom = rg_mom_sync.RgMomSync(glb)
master = rg_thread.RgThreadMaster(glb)
demo = mom_test.Demo()
demo.index = mom_test.DemoIndex()
demo.index.index1 = '111'
demo.index.index2 = '111'
demo.v1 = 1.111
mom.lpush(rg_mom_common.Db.RGMOMD_HOST, demo, 1)
demo2 = mom_test.Demo()
demo2.index = mom_test.DemoIndex()
demo2.index.index1 = '111'
demo2.index.index2 = '111'
demo2.v1 = 2.222
# 不应该报入参错误
mom.linsert(rg_mom_common.Db.RGMOMD_HOST, demo, demo2, 1)
def cb(t):
os._exit(0)
master.add_timer(cb, None, 1)
glb.run()
def main():
glb = rg_global.RgGlobal('sync_demo')
glb.connect_sync(rg_mom_common.Db.RGMOMD_HOST)
mom = rg_mom_sync.RgMomSync(glb)
master = rg_thread.RgThreadMaster(glb)
demo = mom_test.Demo()
demo.index = mom_test.DemoIndex()
demo.index.index1 = '111'
demo.index.index2 = '111'
demo.v1 = 1.111
ret = mom.lpop(rg_mom_common.Db.RGMOMD_HOST, demo)
assert (ret == None)
ret = mom.rpop(rg_mom_common.Db.RGMOMD_HOST, demo)
assert (ret == None)
mom.lpush(rg_mom_common.Db.RGMOMD_HOST, demo, 1)
mom.lpush(rg_mom_common.Db.RGMOMD_HOST, demo, 1)
ret = mom.lpop(rg_mom_common.Db.RGMOMD_HOST, demo)
assert (ret != None)
ret = mom.rpop(rg_mom_common.Db.RGMOMD_HOST, demo)
assert (ret != None)
ret, _, _ = mom.hgetall(rg_mom_common.Db.RGMOMD_HOST, demo)
assert (ret == None)
ret, _, _ = mom.hmget(rg_mom_common.Db.RGMOMD_HOST, demo, [6, 7])
assert (ret == None)
def cb(t):
os._exit(0)
master.add_timer(cb, None, 1)
glb.run()
def __init__(self, glb):
super(App, self).__init__(glb) # 模板
glb.regist_app(self) # 模板
self.demo = app_demo.Demo()
self.demo.index = app_demo.DemoIndex()
self.demo.index.index1 = 'async'
self.demo.index.index2 = 'case4'
def on_get_mom_test_Demo(self, msg, privdata):
demo = app_demo.Demo(msg.value)
if demo.v1 == 3.1415 and demo.v2 == 1.1234:
print("ASYNC_CASE1: PASS")
os._exit(0)
else:
print("ASYNC_CASE1: FAULT")
os._exit(1)
def __init__(self, glb):
super(App, self).__init__(glb) # 模板
glb.regist_app(self) # 模板
self.thread_master = RgThreadMaster(glb)
self.demo = app_demo.Demo()
self.demo.index = app_demo.DemoIndex()
self.demo.index.index1 = 'async'
self.demo.index.index2 = 'case6'
def on_hgetall_mom_test_Demo(self, msg, privdata):
res = rg_mom_common.HashResult(msg.value)
demo = app_demo.Demo(res.value)
if demo.v1 == 3.1415 and demo.v2 == 5.1234 and demo.v3 == 1234 and demo.v4 == 8888:
print("ASYNC_CASE3: PASS")
os._exit(0)
else:
print("ASYNC_CASE3: FAULT")
os._exit(1)
def setUp(self):
self.glb = rg_global.RgGlobal('sync_demo')
if self.glb.connect_sync(Db.RGMOMD_HOST) != 0:
sys.exit()
self.db = Db.RGMOMD_HOST
self.mom = rg_mom_sync.RgMomSync(self.glb)
self.demo = app_demo.Demo()
self.demo.index = app_demo.DemoIndex()
self.demo.index.index1 = '111'
self.demo.index.index2 = '222'
def on_hmget_mom_test_Demo(self, msg, privdata):
res = rg_mom_common.HashResult(msg.value)
if (len(res.field_tag) != 2 or 7 not in res.field_tag
or 9 not in res.field_tag):
print("ASYNC_CASE5: FAULT")
os._exit(1)
demo = app_demo.Demo(res.value)
if demo.v1 == 0 and demo.v2 == 1.1234 and demo.v3 == 0 and demo.v4 == 5678:
print("ASYNC_CASE5: PASS")
os._exit(0)
else:
print("ASYNC_CASE5: FAULT")
os._exit(1)
def on_get_mom_test_Demo(self, msg, privdata):
self.flag += 1
if privdata == 'a, exist':
demo = app_demo.Demo(msg.value)
if demo.index.index1 != 'a' or demo.index.index2 != '1':
print("ASYNC_CASE_6: FAULT")
os._exit(1)
elif privdata == 'b':
demo = app_demo.Demo(msg.value)
if demo.index.index1 != 'b' or demo.index.index2 != '1':
print("ASYNC_CASE_6: FAULT")
os._exit(1)
elif privdata == 'a, not exist':
if msg.value != 0:
print("ASYNC_CASE_6: FAULT")
os._exit(1)
elif privdata == 'b, not exist':
if msg.value != 0:
print("ASYNC_CASE_6: FAULT")
os._exit(1)
if self.flag == 4:
print("ASYNC_CASE_6: PASS")
os._exit(0)
def on_hgetall_mom_test_Demo(self, msg, priv):
res = rg_mom_common.HashResult(msg.value)
demo = app_demo.Demo(res.value)
if (6 not in res.field_tag or
7 not in res.field_tag or
9 not in res.field_tag or
len(res.field_tag) != 3):
print("ASYNC_CASE2: FAULT")
os._exit(1)
if demo.v1 == 3.1415 and demo.v2 == 1.1234 and demo.v3 == 0 and demo.v4 == 5678:
print("ASYNC_CASE2: PASS")
os._exit(0)
else:
print("ASYNC_CASE2: FAULT")
os._exit(1)
def main():
glb = rg_global.RgGlobal('sync_demo')
glb.connect_sync(rg_mom_common.Db.RGMOMD_HOST)
mom = rg_mom_sync.RgMomSync(glb)
master = rg_thread.RgThreadMaster(glb)
demo = mom_test.Demo()
demo.index = mom_test.DemoIndex()
demo.index.index1 = '111'
demo.index.index2 = '111'
ret, fields, change_fields = mom.hmget(rg_mom_common.Db.RGMOMD_HOST, demo,
[6, 7])
# 对于不存在的key，返回值应该是None
assert (ret == None)
def cb(t):
os._exit(0)
master.add_timer(cb, None, 1)
glb.run()
def test_sadd(self):
demo = app_demo.Demo()
demo.index = app_demo.DemoIndex()
demo.index.index1 = 's'
demo.index.index2 = 's'
for i in range(5):
demo.v3 = i
self.mom.sadd(self.db, demo, 1)
for i in range(5):
demo.v3 = i
ret = self.mom.scard(self.db, demo)
self.assertEqual(ret, 5)
ret = self.mom.sismember(self.db, demo)
self.assertEqual(ret, 0)
demo.v3 = 2
self.mom.srem(self.db, demo, 1)
ret = self.mom.scard(self.db, demo)
self.assertEqual(ret, 4)
ret = self.mom.sismember(self.db, demo)
self.assertEqual(ret, 1)
def test_base(self):
# 先保证set、get、delete接口是正确的，因为其余的用例基本都要用这两个接口辅助判断
demo = app_demo.Demo()
demo.index = app_demo.DemoIndex()
demo.index.index1 = '111'
demo.index.index2 = '222'
# 当前无数据
demo2 = self.mom.get(self.db, demo)
self.assertEqual(demo2, None)
# set数据，get后正确
demo.v1 = 3.1415
self.mom.set(self.db, demo, 1)
demo2 = self.mom.get(self.db, demo)
self.assertEqual(demo2.index.index1, '111')
self.assertEqual(demo2.index.index2, '222')
self.assertEqual(demo2.v1, 3.1415)
# delete删除数据
self.mom.delete(self.db, demo, 1)
demo2 = self.mom.get(self.db, demo)
self.assertEqual(demo2, None)
def cb(msg, privdata):
self.assertEqual(privdata, 'this is privdata')
self.assertEqual(msg.cmd, Cmd.SET)
demo = app_demo.Demo(msg.value)
self.assertEqual(demo.v1, 3.1415)