pytest:使用@pytest.mark.usefixtures()给写在类中的方法增加teardown步骤

需求:

  测试用例写在类文件中,有个别用例方法(新增接口)需要做数据恢复,所以需要对此方法增加teardown步骤。

解决过程:

  查看teardown使用方法:

  • setup_method:仅作用于class用例集中的用例,置于class内,每个用例都会调用一次
  • setup_function:作用于独立的def用例,不可作用于class内的用例
  • setup_class:作用于class用例集中的用例,置于class内,只在class用例执行的开始执行setup_class,结束时执行teardown_class
  • setup_module:作用于模块内的所有用例,置于class外,只在所以用例的开始执行setup_module,结束时执行teardown_module

  发现没有针对class中单独的一个方法定制执行teardown,视线转向fixture。

最终解决:

  在类中编写fixture方法,然后测试方法使用@pytest.mark.usefixture("fixture方法名称")指定该用例方法去调用该固件。

class TestBasicConfiguration:
    
    @allure.severity('blocker')
    @allure.story("基础配置模块")
    @pytest.mark.smoke
    @pytest.mark.usefixtures("test_message_signatures_create_teardown")
    def test_message_signatures_create(self, get_case_info, mysql_op, log):
        """
        新增全局短信签名功能验证
        :param get_case_info: 读取用例信息
        :param mysql_op: 数据库连接
        :param log: 日志实例
        :return:
        """
        # 获取当前用例方法名,即json文件中的case_name
        case_name = sys._getframe().f_code.co_name
        # 获取当前模块名,用于在get_case_info中读取对应的json用例文件
        module_name = str(os.path.basename(__file__)).split('.')[0]
        print('模块名称为:{}'.format(module_name))
        url, data, headers = get_case_info(case_name, module_name)
        res = requests.request("POST", url, data=data, headers=headers)
        print(res.json())
        api_res_data = res.json()['code']
        print('接口返回值为:{}'.format(api_res_data))

        # 查询新增的错误码
        data_sql = """SELECT t.sign_name FROM sms_center_uat.sms_global_sign t WHERE t.VALID_STATUS=1 AND t.sign_name='test2'"""
        mysql_op.execute_sql(data_sql)
        data = mysql_op.get_data()
        # mysql_op.close_db()
        db_query_data = data[0][0]
        print('数据库查询结果为:{}'.format(db_query_data))
        try:
            assert api_res_data == "SUCCESS_REQUEST" and db_query_data == "test2"
            log.logger.info('场景:{}断言成功'.format(case_name))
        except AssertionError:
            err_msg = '场景:{0}接口断言失败,接口返回结果:{1} != 数据库查询结果{2}'.format(case_name, api_res_data, db_query_data)
            log.logger.error(err_msg)
            pytest.fail(err_msg)
        finally:
            log.logger.info('场景:{0}验证完成'.format(case_name))

    @pytest.fixture()
    def test_message_signatures_create_teardown(self, mysql_op):
        yield
        print("删除新增的短信签名")
        # 软删除新增的错误码
        data_sql = """UPDATE sms_center_uat.sms_global_sign t SET t.VALID_STATUS=0 WHERE t.VALID_STATUS=1 AND t.sign_name='test2'"""
        mysql_op.execute_sql(data_sql)

 

posted @ 2020-04-23 16:47  秋寻草  阅读(994)  评论(0编辑  收藏  举报