def test_tcp_rpc_device_user_oper(self):
user_info = {"user_name": random_str(),
"password": random_str(),
"des": random_str(),
"jid": random_str(),
"jid_pwd": random_str(),
"type": enum.DEVICE_USER,
"device_type": choice_device_type()}
RgdaTcpRpcClt = ServiceMgrCacher().get_connection(ST_REGISTER_DA,
rdm_type=RT_HASH_RING,
rdm_param=user_info['user_name'])
self.assertEqual(RgdaTcpRpcClt.ping(), "ping response!")
self.assertTrue(user_insert(RgdaTcpRpcClt, **user_info))
self.assertTrue(user_select(RgdaTcpRpcClt, user_name_ls=[user_info['user_name']])[0]['user_name'] == user_info['user_name'])
self.assertTrue(int(user_select(RgdaTcpRpcClt, user_name_ls=[user_info['user_name']])[0]['device_type']) == int(user_info['device_type']))
self.assertTrue(user_delete(RgdaTcpRpcClt, user_name_ls=user_info['user_name']))
def test_device_user_get_users_device_type(self):
user_name = random_str()
password = random_str()
dtype = choice_device_type()
result = register(RgstHttpRpc,
"",
user_name,
password,
enum.DEVICE_USER,
"TEST1",
urllib.quote(Signer().gen_sign("", user_name, password, enum.DEVICE_USER, "TEST1")),
device_type=dtype)
assert result["result"] == error_code.ERROR_SUCCESS
token = LoginGetToken(RgstHttpRpc, user_name, password)
result = get_users(RgstHttpRpc, token, [user_name])
self.assertTrue(int(result['user_info_ls'][0]['device_type']['code']) == int(dtype))
cancel(RgstHttpRpc, "", user_name, password, urllib.quote(Signer().gen_sign("", user_name, password)))
def test_device_user_login(self):
user_name = random_str()
password = random_str()
result = register(RgstHttpRpc,
"",
user_name,
password,
enum.DEVICE_USER,
"TEST1",
urllib.quote(Signer().gen_sign("", user_name, password, enum.DEVICE_USER, "TEST1")),
device_type=choice_device_type(),)
assert result["result"] == error_code.ERROR_SUCCESS
result = login(RgstHttpRpc, user_name, password, type=enum.DEVICE_USER)
assert result["result"] == error_code.ERROR_SUCCESS
LoginGetToken(RgstHttpRpc, user_name, password)
LoginGetTokenUser(RgstHttpRpc, user_name, password)
cancel(RgstHttpRpc, "", user_name, password, urllib.quote(Signer().gen_sign("", user_name, password)))
def make_xmpp_relation(user_ls, add_roster=False, del_roster=False):
"""
xmpp用户关系设置
:param user_ls:用户列表
:param add_roster: 是否添加用户之间为roster关系
:param del_roster: 是否删除用户之间的roster关系
:return: [(jid1, jid_pass1),,,,]
"""
from lib.device_type import choice_device_type
jid_ls = []
# register/login user
http_rpc = ServiceMgrCacher().get_connection(ST_REGISTER, protocol=PT_HTTPS)
for user in user_ls:
login_result = login(http_rpc, user, user, enum.DEVICE_USER)
if login_result.get('result', None) != error_code.ERROR_SUCCESS:
regiger_result = register(http_rpc, "", user, user, enum.DEVICE_USER, user, device_type=choice_device_type())
assert regiger_result.get('result', None) == error_code.ERROR_SUCCESS
login_result = login(http_rpc, user, user, enum.DEVICE_USER)
assert login_result.get('result', None) == error_code.ERROR_SUCCESS
jid_ls.append((login_result['user_info']['jid'], user))
# add/del roster
for idx, user in enumerate(user_ls):
next_idx = idx + 1
while next_idx < len(user_ls):
if add_roster:
try:
UserService().add_roster(user_ls[idx], user_ls[next_idx])
except:
pass
try:
UserService().add_roster(user_ls[next_idx], user_ls[idx])
except:
pass
if del_roster:
try:
UserService().del_roster(user_ls[idx], user_ls[next_idx])
except:
pass
try:
UserService().del_roster(user_ls[next_idx], user_ls[idx])
except:
pass
next_idx += 1
return jid_ls
