巧用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)

四、总结

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