def loadTestsFromModule(self, module, use_load_tests=True):
     if module.split('.')[0] in ALLOW_APP:
         with enter_app_context(module.split('.')):
             return super(UnittestLoader,
                          self).loadTestsFromModule(module, use_load_tests)
     return super(UnittestLoader,
                  self).loadTestsFromModule(module, use_load_tests)
Exemple #2
0
# coding=utf-8
# 调用应用环境,test中避免调用其他app的环境
from flask.ext.script.framework import enter_app_context

__author__ = 'GaoJie'

enter_app_context('auto').push()
from auto import *
 def loadTestsFromModule(self, module, use_load_tests=True):
     if module.split('.')[0] in ALLOW_APP:
         with enter_app_context(module.split('.')):
             return super(UnittestLoader, self).loadTestsFromModule(module, use_load_tests)
     return super(UnittestLoader, self).loadTestsFromModule(module, use_load_tests)