Python BuiltIn.should_be_equal_as_strings方法代码示例

# 需要导入模块: from robot.libraries.BuiltIn import BuiltIn [as 别名]
# 或者: from robot.libraries.BuiltIn.BuiltIn import should_be_equal_as_strings [as 别名]

#.........这里部分代码省略.........

        self.built_in.log('Getting verify user email by real name')
        with DBHelper() as db_helper:
            verify_user_email = db_helper.get_verify_user_email_by_real_name(verify_user_real_name.encode('utf-8'))

        self.built_in.log('Using verify user {0} to log in'.format(verify_user_email))
        current_user_request = verify.VerifyRequest(self.base_URL, verify_user_email)
        current_user_request.login()

        for user_id in user_ids:
            self.built_in.log('Using user_id = {0} to commit to {1}'.format(user_id, audit_user_status))
            if audit_user_status == AuditUserStatusEnum.INQUIRE_SUCCESS:
                inv_rets = current_user_request.get_all_valid_investigate_result()
                response = current_user_request.commit_to_first_verify(user_id, 12, '调查备注', **inv_rets)
            elif audit_user_status == AuditUserStatusEnum.VERIFY_FAIL:
                inv_rets = current_user_request.get_all_notmatch_investigate_result()
                response = current_user_request.commit_to_verify_fail(user_id, 12, '补件调查', **inv_rets)
            elif audit_user_status == AuditUserStatusEnum.FIRST_VERIFY_SUCCESS:
                response = current_user_request.commit_to_second_verify(user_id, 1000, 1, 12, '一审备注')
            elif audit_user_status == AuditUserStatusEnum.FIRST_SEND_BACK:
                response = current_user_request.commit_to_first_verify_sendback(user_id, '一审退回')
            elif audit_user_status == AuditUserStatusEnum.SECOND_VERIFY_SUCCESS:
                response = current_user_request.commit_to_pass_second_verify(user_id, second_verify_amount, 1, 12, '二审备注')
            elif audit_user_status == AuditUserStatusEnum.SECOND_SEND_BACK:
                response = current_user_request.commit_to_second_verify_sendback(user_id, '二审退回')
            elif audit_user_status == AuditUserStatusEnum.VERIFY_SUCCESS:
                response = current_user_request.commit_to_pass_signed_approval(user_id, second_verify_amount, 1, 12, '最终审核通过')
            elif audit_user_status == AuditUserStatusEnum.VERIFY_REJECT:
                response = current_user_request.commit_to_refuse(user_id, 'A01', '退件')
            else:
                raise AssertionError('Can not handle this kind of status!!')
            self.built_in.log('Response = {0}'.format(response.json()))

    # 比较审核流水中个人信息部分
    def compare_user_info_for_verify_log(self, nick_name):
        nick_name = nick_name.encode('utf-8')
        with DBHelper() as db_helper:
            user_id = db_helper.get_user_id_by_nick_name(nick_name)
        self.request_utils.login()
        result_from_api = self.request_utils.get_user_verify_logs(user_id)
        user_info_from_api = result_from_api.json()['data']['userInfo']
        with DBHelper() as db_helper:
            user_info_from_db = db_helper.get_user_info_for_verify_log(nick_name)
        self.built_in.should_be_equal_as_strings(user_info_from_api['nickName'], user_info_from_db[0], 'Compare nick name for user info in verify log')
        self.built_in.should_be_equal_as_strings(user_info_from_api['realName'], user_info_from_db[1], 'Compare real name for user info in verify log')
        self.built_in.should_be_equal_as_strings(user_info_from_api['mobile'], user_info_from_db[2], 'Compare mobile for user info in verify log')
        self.built_in.should_be_equal_as_strings(user_info_from_api['idNo'], user_info_from_db[3], 'Compare idNo for user info in verify log')
        self.built_in.should_be_equal_as_strings(user_info_from_api['userType'], unicode(Channel.get_value(user_info_from_db[4]), 'utf-8'), 'Compare channel for user info in verify log')
        self.built_in.should_be_equal_as_strings(user_info_from_api['verifyUserStatus'], unicode(VerifyUserStatus.get_value(user_info_from_db[5]), 'utf-8'), 'Compare verify user status for user info in verify log')

    # 比较审核流水中每个审核阶段的结果
    def compare_verify_log(self, nick_name, index, operation, operation_result, verify_user=None, note=None, amount=None, level=None, cash_draw_ratio=None):
        nick_name = nick_name.encode('utf-8')
        index = int(index)

        with DBHelper() as db_helper:
            user_id = db_helper.get_user_id_by_nick_name(nick_name)
        self.request_utils.login()
        result_from_api = self.request_utils.get_user_verify_logs(user_id)
        verify_logs_from_api = result_from_api.json()['data']['userVerifyLog']

        self.built_in.log("User verify log: " + str(verify_logs_from_api))
        self.built_in.should_be_equal_as_strings(verify_logs_from_api[index]['operation'], operation, 'Compare operation for verify log')
        self.built_in.should_be_equal_as_strings(verify_logs_from_api[index]['operationResult'], operation_result, 'Compare operationResult for verify log')

        if verify_user is not None:
            self.built_in.should_be_equal_as_strings(verify_logs_from_api[index]['verifyUser'], verify_user, 'Compare verify user for verify log')
        if note is not None:
            self.built_in.should_be_equal_as_strings(verify_logs_from_api[index]['note'], note, 'Compare note for verify log')
        if amount is not None:
            self.built_in.should_be_equal_as_strings(verify_logs_from_api[index]['amount'], amount, 'Compare amount for verify log')
        if level is not None:
            self.built_in.should_be_equal_as_strings(verify_logs_from_api[index]['level'], level, 'Compare level for verify log')
        if cash_draw_ratio is not None:
            self.built_in.should_be_equal_as_strings(verify_logs_from_api[index]['cashDrawRatio'], cash_draw_ratio, 'Compare cashDrawRatio for verify log')

    # 获取审核流水中条目数
    def get_verify_log_count(self, nick_name):
        nick_name = nick_name.encode('utf-8')
        with DBHelper() as db_helper:
            user_id = db_helper.get_user_id_by_nick_name(nick_name)
        self.request_utils.login()
        result_from_api = self.request_utils.get_user_verify_logs(user_id)
        return len(result_from_api.json()['data']['userVerifyLog'])

    # 更新用户类型为个人
    def update_user_channel_type_to_personal_register(self, nick_name):
        with DBHelper() as db_helper:
            user_id = db_helper.get_user_id_by_nick_name(nick_name.encode('utf-8'))
            db_helper.update_user_channel_by_user_id(user_id, Channel.PERSONAL_REGISTER)

    # 更新用户类型为BD导入
    def update_user_channel_type_to_db_import(self, nick_name):
        with DBHelper() as db_helper:
            user_id = db_helper.get_user_id_by_nick_name(nick_name.encode('utf-8'))
            db_helper.update_user_channel_by_user_id(user_id, Channel.BD_IMPORT)

    # 获取接口返回的待办任务
    def get_pending_task(self, index):
        return self.flow_task_request_utils.get_pending_tasks()[0][index]