巧用PyUnit中unittest特性解决接口测试产生脏数据问题

巧用PyUnit中unittest特性解决接口测试产生脏数据问题

一、背景

测试数据创建后需要对其删除，不然可能产生脏数据，对开发和测试、生产环境造成一定影响。其接口框架是基于Python，API规范基于REST。

二、产生原因

（1）改进前：清除资源的操作放在每个正向测试用例里，没有在setUp和tearDown中删除。

（2）改进后：在setUp中初始化资源id=None，每次创建资源时会生产一个资源id，然后在tearDown中，判断资源的id是否为None，if不为None，then delete 资源并将资源id重新赋值为None。这样不管有没有执行成功，最终都会在tearDown中删除被创建成功的资源，初始化工作放在setUp中。

三、代码对比

（1）改进前：清除资源的操作放在每个正向测试用例里，没有在setUp和tearDown中删除。代码如下：

     def test_konwledges_check_by_list_ok(self):
        """
        lc知识点API:成功创建2个知识点后，批量查询知识点详细信息
        """
        title = 'knowledges'
        flag = 0
        response = self.knowledge.add_knowledges(title, flag)
        message = "创建知识点接口返回失败"
        code = 200
        data = self.rest_o.parse_response(response, code, message)
        self.knowledges_id1 = data['identifier']

        response = self.knowledge.add_knowledges(title, flag)
        message = "创建知识点接口返回失败"
        code = 200
        data = self.rest_o.parse_response(response, code, message)
        self.knowledges_id2 = data['identifier']

        knowledge_id_str = [self.knowledges_id1, self.knowledges_id2]

        response = self.knowledge.check_konwledges_details_by_list(knowledge_id_str)
        #print 'response: ' + str(response)
        message = "批量查询知识点详细信息接口返回失败"
        code = 200
        data = self.rest_o.parse_response(response, code, message)
        print data
        assert data[self.knowledges_id1]
        assert data[self.knowledges_id2]

        self.knowledge.delete_konwledges(knowledge_id1)#放在测试用例中删除
        self.knowledge.delete_konwledges(knowledge_id2)#放在测试用例中删除

（2）改进后：

a.setUp和tearDown中改进：

    def setUp(self):
        print 'start run KnowledgeTest（知识点接口）......'
        self.knowledge = Knowledge()
        self.lessons = Lessons()
        self.instructional = Instructional(id)
        self.teaching_material = TeachingMaterial()
        self.rest_o = CoRestful.Restful()
        self.rand_o = CoRand()
        self.knowledges_id = None
        self.knowledges_id1 = None
        self.knowledges_id2 = None
        self.knowledge_relation_id = None

    def tearDown(self):

        #判断资源的id是否为None，if不为None，then delete 资源并将资源id重新赋值为None<span style="font-family: Arial, Helvetica, sans-serif;">，</span>这样不管有没有执行成功，最终都会在tearDown中删除被创建成功的资源
        if self.knowledge_relation_id != None:
            self.knowledge.delete_relation(self.knowledge_relation_id)
        if self.knowledges_id2 != None:
            self.knowledge.delete_konwledges(self.knowledges_id2)
        if self.knowledges_id1 != None:
            self.knowledge.delete_konwledges(self.knowledges_id1)
        if self.knowledges_id != None:
            self.knowledge.delete_konwledges(self.knowledges_id)
        print '------------------------------------------------------'

b.测试用例中将其删除代码注释

def test_konwledges_check_by_list_ok(self):
        """
        lc知识点API:成功创建2个知识点后，批量查询知识点详细信息
        """
        title = 'knowledges'
        flag = 0
        response = self.knowledge.add_knowledges(title, flag)
        message = "创建知识点接口返回失败"
        code = 200
        data = self.rest_o.parse_response(response, code, message)
        self.knowledges_id1 = data['identifier']

        response = self.knowledge.add_knowledges(title, flag)
        message = "创建知识点接口返回失败"
        code = 200
        data = self.rest_o.parse_response(response, code, message)
        self.knowledges_id2 = data['identifier']

        knowledge_id_str = [self.knowledges_id1, self.knowledges_id2]

        response = self.knowledge.check_konwledges_details_by_list(knowledge_id_str)
        #print 'response: ' + str(response)
        message = "批量查询知识点详细信息接口返回失败"
        code = 200
        data = self.rest_o.parse_response(response, code, message)
        print data
        assert data[self.knowledges_id1]
        assert data[self.knowledges_id2]

        #注释原在测试用例中删除的代码
        #self.knowledge.delete_konwledges(knowledge_id1)
        #self.knowledge.delete_konwledges(knowledge_id2)

四、总结

发现问题要多思考，并努力把其解决。