def test_env_exclude_tags(self):
class MyTaskSet(User):
@tag("exclude this")
@task
def excluded(self):
pass
@tag("dont exclude this")
@task
def not_excluded(self):
pass
@task
def dont_exclude_this_either(self):
pass
class MyUser(User):
tasks = [MyTaskSet]
env = Environment(user_classes=[MyUser], exclude_tags=["exclude this"])
env._filter_tasks_by_tags()
self.assertListEqual(MyUser.tasks, [MyTaskSet])
self.assertListEqual(
MyUser.tasks[0].tasks,
[MyTaskSet.not_excluded, MyTaskSet.dont_exclude_this_either],
)
def test_env_include_tags(self):
class MyTaskSet(TaskSet):
@tag('include this')
@task
def included(self):
pass
@tag('dont include this')
@task
def not_included(self):
pass
@task
def dont_include_this_either(self):
pass
class MyUser(User):
tasks = [MyTaskSet]
env = Environment(user_classes=[MyUser], tags=['include this'])
env._filter_tasks_by_tags()
self.assertListEqual(MyUser.tasks, [MyTaskSet])
self.assertListEqual(MyUser.tasks[0].tasks, [MyTaskSet.included])
