如何理解HTMLTestRunner中test(result),HTMLTestRunner是如何让UnitTest运行的

我们在用Unittest框架时,生成html格式的报告一般都是用HTMLTestRunner.py这个第三方库,大概使用方法如下:

with open(config.report_file, 'wb') as fp:
    HTMLTestRunner(stream=fp,
                   title='[{}] 接口测试报告'.format(date),
                   verbosity=2,
                   description='每日定时接口测试,监控线上接口情况').run(suite)

我们实例化一个HTMLTestRunner类的对象,并调用该类的run()方法,传入的是unittest.TestSuite类的对象suite,执行测试用例(测试套件)生成测试结果并写入html模板中,生成网页报告。在我们使用的时候,只要调用HTMLTestRunner().run()就行了,那么HTMLTestRunner().run()内部做了哪些事呢?

HTMLTestRunner.py中有一个类HTMLTestRunner,该类有一个方法run,该方法如下:

# HTMLTestRunner.py

class HTMLTestRunner(Template_mixin):
    ......
    ......
    def run(self, test):
        "Run the given test case or test suite."
        result = _TestResult(self.verbosity)
        test(result)
        self.stopTime = datetime.datetime.now()
        self.generateReport(test, result)
        print('\nTime Elapsed: %s' % (self.stopTime-self.startTime), file=sys.stderr)
        return result

run()接收一个参数test,根据注释我们知道该参数代表test case 或者 test suite,也就是我们写的测试用例或由测试用例组成的测试套件,那么test实际上就是TestSuite类或者TestCase类的实例对象,然后其中有一行代码:

test(result)

这时候就纳闷了,test是一个TestSuiteTestCase类的实例对象,把实例对象当成函数调用是什么意思?

我们来写一个类试试:

class Human:

    def __init__(self, name):
        self.name = name

    def eat(self):
        return '%s is eat' % self.name


one = Human('CJ')
one()

# TypeError: 'Human' object is not callable

定义一个Human类,然后实例化一个对象one,把one当作函数直接调用,即one(),结果会报错TypeError: 'Human' object is not callable,看来我们的代码存在问题。

这时候,我们引入一个东西,魔术方法__call__()

python 中一切皆对象,函数也是对象,同时也是可调用对象(callable)。

关于可调用对象,我们平时自定义的函数、内置函数和类都属于可调用对象,但凡是可以把一对括号 () 应用到某个对象身上都可称之为可调用对象,判断对象是否为可调用对象可以用函数 callable

一个类实例要变成一个可调用对象,只需要实现一个特殊方法__call__()

修改上面的例子:

class Human:

    def __init__(self, name):
        self.name = name

    def eat(self):
        return '%s is eat' % self.name

    def __call__(self, *args, **kwargs):
        print('object is callable')


one = Human('CJ')
one()

# object is callable

再次调用one实例对象后,输出了__call()__方法里面打印的内容,这说明,在类里面实现了__call__()魔术方法后,可以把类的实例化对象当成函数调用,而实际调用的就是__call__()方法。

那么再回到上面的框架里面,既然可以写成test(result),那说明test这个对象的类里面,一定实现了__call__()方法,所以我们再去看看__call__()方法里面是怎么处理的。因为testTestSuite类的实例,所以我们看看TestSuite类是怎么实现__call__()

# unittest\suite.py
class TestSuite(BaseTestSuite):
    ......
    def run(self, result, debug=False):
        ......
        return result

unittest\suite.py中,TestSuite里面有一个run()方法,并没有实现__call__()方法,但是TestSuite是继承自BaseTestSuite,我们再看看BaseTestSuite

# unittest\suite.py
class BaseTestSuite(object):
    ......
    def run(self, result):
        for index, test in enumerate(self):
            if result.shouldStop:
                break
            test(result)
            if self._cleanup:
                self._removeTestAtIndex(index)
        return result
    
    def __call__(self, *args, **kwds):
        return self.run(*args, **kwds)

BaseTestSuite中,确实有__call__(),并且也有run()方法,而__call__()实际又调用了self.run()方法。

这里需要注意,由于TestSuite类是继承自BaseTestSuite,并且两者都实现了run()方法,那么实际执行的时候,如果该类是属于TestSuite,那么最终实际执行的是TestSuite().run(),而不是BaseTestSuite().run()

那么,此时,梳理出来的步骤就是:

我们初始化一个HTMLTestRunner类的实例,调用类方法run(),传入的参数是TestSuite类的实例,

HTMLTestRunnerrun()方法中,把TestSuite类的实例当成函数调用,这是由于TestSuite类的父类BaseTestSuite实现了魔术方法__call__(),而__call__()里面是调用self.run(),所有本质上调用的则是TestSuite类的run()方法,继续来看TestSuite里面run()做了些什么东西

# unittest\suite.py
class TestSuite(BaseTestSuite):

    def run(self, result, debug=False):
        topLevel = False
        if getattr(result, '_testRunEntered', False) is False:
            result._testRunEntered = topLevel = True

        for index, test in enumerate(self):
            if result.shouldStop:
                break

            if _isnotsuite(test):
                self._tearDownPreviousClass(test, result)
                self._handleModuleFixture(test, result)
                self._handleClassSetUp(test, result)
                result._previousTestClass = test.__class__

                if (getattr(test.__class__, '_classSetupFailed', False) or
                    getattr(result, '_moduleSetUpFailed', False)):
                    continue

            if not debug:
                test(result)
            else:
                test.debug()

            if self._cleanup:
                self._removeTestAtIndex(index)

        if topLevel:
            self._tearDownPreviousClass(None, result)
            self._handleModuleTearDown(result)
            result._testRunEntered = False
        return result

for index, test in enumerate(self):

self代表的是TestSuite类实例,经过枚举及for循环得到的testTestCase实例

然后下面又是熟悉的test(result),那么跟上面的原理一样,TestCase类或者它的父类里面肯定实现了__call__()方法,并且__call__()里面实际调用的是TestCase().run()方法,所以这里真正执行的就是TestCase().run()TestCase().run()里面则实现的是每个测试用例的执行过程,最终得到执行的结果。

源码也印证了确实如此:

class TestCase(object):
    ....
    ....
    def run(self, result=None):
        ....
        return result
        
    def __call__(self, *args, **kwds):
        return self.run(*args, **kwds)

至此,完整的流程应该是:

我们初始化一个HTMLTestRunner类的实例,调用类方法run(),传入的参数是TestSuite类的实例,

由于TestSuite类实现了__call__()魔术方法,所以在HTMLTestRunnerrun()方法中

TestSuite类实例当成函数调用,达到调用TestSuite类的run()方法的目的,在TestSuiterun()方法里遍历出TestCase对象,而又由于TestCase类也实现了__call__()魔术方法,把TestCase对象当成函数调用,实际执行的是TestCaserun(),最终完成测试得到结果。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,772评论 6 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,458评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,610评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,640评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,657评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,590评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,962评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,631评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,870评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,611评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,704评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,386评论 4 319
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,969评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,944评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,179评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 44,742评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,440评论 2 342

推荐阅读更多精彩内容