python unittest TestCase间共享数据（全局变量的使用）

最新推荐文章于 2024-12-16 18:00:58 发布

原创最新推荐文章于 2024-12-16 18:00:58 发布 · 6.6k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#unittest #python #单元测试

python 专栏收录该内容

17 篇文章

订阅专栏

本文介绍如何在Python的unittest模块中实现不同测试用例间的资源共享，通过定义全局变量并在需要的地方引用该变量，确保测试过程中创建的数据能在后续测试步骤中得到复用。

使用unittest模块进行单元测试，涉及到以下场景

例如对某个实体，测试方法有创建，更新，实体查询，删除

使用unittest进行单元测试，可以在创建时候记录下返回的ID,在更新、删除等操作的时候就根据这个新创建的ID进行操作，这就涉及到不同的TestCase之间共享数据。

最初我在class TestCase(unittest.TestCase):里增加变量，运行创建时候设置值，但是发现在运行其他方法时候值被清空了，说明这种方法不可行。

最后只好定义全局变量，但是在局部用的时候需要使用globals()['newid'] 来操作全局变量。

例如以下例子，创建时候获取ID，并设置，然后get的时候直接测刚才生成的ID，测delete时候就可以把这条数据删除掉了

newid = None

class MonTemplateCase(unittest.TestCase):
    base = "http://localhost:8080/"

    def test_moncluster_temp_create(self):
        json = {"monitor_template":{"name":"testname1","desc":"desc2","dcid":"1","host_info":{"check_interval":"1","check_period":"5","max_check_attempts":"1","notification_interval":"1","notifications_enabled":"1","retry_interval":"1","contacts":"[1]"},"service_info":[{"id":"14","max_check_attempts":"1","notification_interval":"1","notifications_enabled":"1","retry_interval":"1","contacts":"[1]","check_period":"5","service_description":"CPU","check_command":"14!80 90","check_interval":"5"}]}}
        headers = {'content-type': 'application/json'}
        r = requests.post(self.base + "api/moncluster/template", data=simplejson.dumps(json),headers=headers)
        result = simplejson.loads(r.text)
        print "create result:", result
        if result['action'] is False:
            print result
        else:
            globals()['newid'] = result['data']
        self.assertEqual(True, result['action'])

    def test_comcluster_temp_get(self):
        global newid
        if newid is None:
            raise Exception('id is none cannot get')
        f = urllib2.urlopen(self.base + "api/moncluster/template/%s" % str(newid))
        result = simplejson.loads(f.read())
        print "get result:",result
        self.assertEqual(True, result['action'])