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