def load_from_dir(self, top):
suite = TestSuite()
for name in self._walk(top):
test = self.load_from_file(name)
if test:
suite.add_test(test)
return suite
def load_args(self, args, env):
if not args:
path = os.path.join(env.ENV_PATH, env.CASES_DIR)
return self.load_from_dir(path)
suite = TestSuite()
for arg in args:
if arg in env.SUITES:
test = self.load(env.SUITES[arg], env)
else:
test = self.load(arg, env)
if test:
suite.add_test(test)
return suite
def load(self, sel, env):
if not sel.startswith('!'):
return
loader = TestLoader()
excluding = sel[1:]
comps = loader.guess_components(env)
if excluding not in comps:
return
suite = TestSuite()
for comp in comps:
if comp != excluding:
test = loader.load_component(env, comp)
suite.add_test(test)
return suite
