学习了Appium python项目施展的课程小伙伴都会有一个疑问,说现在所有的case都是通过一个suite进行一个方法一个方法进行添加的,但是在实际过程中我们不希望这样,我们做出来的功能是这样:

suite = unittest.TestSuite()
suite.addTest(CaseTest("test_02",parame=i))
unittest.TextTestRunner().run(suite)

  这个是我们现在做的现状,但是我们希望的是:

discover = unittest.defaultTestLoader.discover(case_dir,pattern='test_*.py')
unittest.TextTestRunner().run(discover)

  那我们如何才能够我们做出来的换成我们想要的呢?我这里告诉大家答案,你现在还不行!

  为什么这么说呢?首先我们需要知道一个前提,我们现在拿到的discover是什么?我们运行机制是什么?先看下面两个简单代码:

#coding=utf-8
import unittest class StudyTest(unittest.TestCase): def test_study01(self):
print("studytest de 01")
def test_study02(self):
print("studytest de 02")

  

#coding=utf-8
import unittest
class StudyTest01(unittest.TestCase):
def test_study001(self):
print("studytest01 de 001") def test_study002(self):
print("studytest01 de 002") if __name__ == '__main__':
test_dit = './'
discovre = unittest.defaultTestLoader.discover(test_dit,pattern='study_*.py')
print(discovre)
runner = unittest.TextTestRunner()
runner.run(discovre)

  上面两个代码我们简单的实现了discover的实现,这里我们是不是想要的就是这样呢?但是为什么说不行呢?至少你现在不行,首先看一下我们打印的discover是什么。

<unittest.suite.TestSuite tests=[
<unittest.suite.TestSuite tests=[]>,
<unittest.suite.TestSuite tests=[]>,
<unittest.suite.TestSuite tests=[]>,
<unittest.suite.TestSuite tests=[
<unittest.suite.TestSuite tests=[]>,
<unittest.suite.TestSuite tests=[<case.study_test.StudyTest testMethod=test_study01>, <case.study_test.StudyTest testMethod=test_study02>]>]>,
<unittest.suite.TestSuite tests=[
<unittest.suite.TestSuite tests=[<case.study_test01.StudyTest01 testMethod=test_study001>, <case.study_test01.StudyTest01 testMethod=test_study002>]>]>,
<unittest.suite.TestSuite tests=[]>,
<unittest.suite.TestSuite tests=[]>,
<unittest.suite.TestSuite tests=[]>,
<unittest.suite.TestSuite tests=[]>]>

  通过上面的打印我相信大家能够看出来,我把格式整理之后我们所谓的discover其实就是我们所有的case的一个集合(这里这么说不恰当,但是更容易理解),这里面是我们的用例,这个时候我只要去运行我们的case就可以了。那么问题来了,我们需要传递参数如何传递?是不是发现这样我们没办法传递参数?然后小伙伴们通过前面的课程发现了这样一个方法,课程中的方法我们讲到了

一个全局变量,但是这个我们需要一个方法一个方法添加,这不是我们的目的,所以我们可以通过查找在网上也能够找到方法,看一下代码:

#-*- coding: utf-8 -*-

#author: mushishi

#Date:2018年8月26日

import unittest

class ParametrizedTestCase(unittest.TestCase):

"""

继承之后,重写一下,把这个参数传递到unittest里面

"""

def __init__(self, methodName='runTest', param=None):

super(ParametrizedTestCase, self).__init__(methodName)

self.param = param

@staticmethod

def parametrize(testcase_klass, param=None):

testloader = unittest.TestLoader()

testnames = testloader.getTestCaseNames(testcase_klass)

suite = unittest.TestSuite()

for name in testnames:

suite.addTest(testcase_klass(name, param=param))

return suite

class TestOne(ParametrizedTestCase):

def test_first(self):

print ('param =', self.param)

self.assertEqual(1, 1)

def test_two(self):

print ('param =', self.param)

self.assertEqual(2, 2)

if __name__ == '__main__':

suite = unittest.TestSuite()

suite.addTest(ParametrizedTestCase.parametrize(TestOne, param=1))

suite.addTest(ParametrizedTestCase.parametrize(TestOne, param=2))

unittest.TextTestRunner(verbosity=2).run(suite)

通过上面这个方法是不是发现可以传递参数。但是可能很多小伙伴不知道什么意思,没事,我们仔细看一下代码的拆分,毕竟这个代码网上很多,但是很多小伙伴不知道为什么,而且很多小伙伴不知道这个怎么和我们的需求串联起来,也不知道和我们这个有什么关系,我们应该怎么处理。我们看下面:

    suite = unittest.TestSuite()
suite.addTest(ParametrizedTestCase.parametrize(TestOne, param=1))

首先我们看我们的程序执行入口,在我们这个入口我们知道我们有了一个容器,这个时候需要往容器里面添加case,这个case的添加我们是有条件的,我们首先需要知道这个case在哪一个class里面,也就是说我们添加我们case的时候是把class和方法一起添加的,那这个和我们上面说的通过拿到py文件,然后想参数话有啥关系?首先需要提出一个疑问:

  1、我们前面的代码只是拿到了我们所有的case,并没有拿到我们的class name,系统只是通过py文件去拿里面case然后添加到一个集合里

  2、我们后面知道的目前改进unittest框架的知识是需要class name的 这个我们好像也没办法操作

有这么两个疑问之后那么我们可以换一个思考方式,既然参数话目前掌握的知识我们只能够通过上面的参数话去操作,那么我们如何去更改自己的代码,然后又不要我们一个一个case的添加?

  其实通过上面的代码大家就已经发现了一个不一样的地方,我们这个不需要每个方法每个方法去添加,只需要有个class name 就行,那么我们这样能够把所有的case都运用到这个里面来吗?答案是肯定的。

我们只需要通过一个简单的转换就可以了。

仔细看上面的代码我们知道,参数话这个代码肯定是公共的,那我们就把这个抽出来作为一个公共的类:

#-*- coding: utf-8 -*-

#author: mushishi

#Date:2018年8月26日

import unittest

class ParametrizedTestCase(unittest.TestCase):

    """

     继承之后,重写一下,把这个参数传递到unittest里面

    """

    def __init__(self, methodName='runTest', param=None):

        super(ParametrizedTestCase, self).__init__(methodName)

        self.param = param

    @staticmethod

    def parametrize(testcase_klass, param=None):

        testloader = unittest.TestLoader()

        testnames = testloader.getTestCaseNames(testcase_klass)

        suite = unittest.TestSuite()

        for name in testnames:

            suite.addTest(testcase_klass(name, param=param))

        return suite
#这个类我们单独抽离出来,我们就是用来把这个作为参数化的集中地

看到这里明白了么?其实我们只是需要把这个抽离出来,然后我们的case就变成了这样:

第一个case类:

#-*- coding: utf-8 -*-
#author: mushishi
#Date:2018年8月26日
from case.ParametrizedTestCase import ParametrizedTestCase
class Test01(ParametrizedTestCase):
def test_01_case(self):
print("这个是test01case里面的参数",self.param)
def test_02_case(self):
print("这个是test01case里面的参数",self.param)

第二个case类:

#-*- coding: utf-8 -*-

#author: mushishi

#Date:2018年8月26日
from case.ParametrizedTestCase import ParametrizedTestCase
import unittest
from case.pounittest_one import Test01
class TestOne(ParametrizedTestCase): def test_first(self): print ('param =', self.param) self.assertEqual(1, 1) def test_two(self): print ('param =', self.param) self.assertEqual(2, 2) if __name__ == '__main__': suite = unittest.TestSuite() suite.addTest(ParametrizedTestCase.parametrize(TestOne, param=1)) suite.addTest(ParametrizedTestCase.parametrize(TestOne, param=2))
suite.addTest(ParametrizedTestCase.parametrize(Test01,param=3)) unittest.TextTestRunner(verbosity=2).run(suite)

这里我引入了两个case类,但是只是添加了这样的方法,是不是解决了呢?其实这样的方法网上很多,但是很多都需要思考怎么做。或者换一个思维。

05-27 22:57