def test_simple_args(self):
def fn(a, b):
pass
assert visitor.visit_task(fn, ()) == {
'name': 'fn',
'path': (),
'doc': None,
'cron': None,
'argspec': {
'args': ['a', 'b'],
'varargs': None,
'keywords': None,
'defaults': None,
},
}
def test_varargs(self):
def fn(*args, **kwargs):
pass
assert visitor.visit_task(fn, ()) == {
'name': 'fn',
'path': (),
'doc': None,
'cron': None,
'argspec': {
'args': [],
'varargs': 'args',
'keywords': 'kwargs',
'defaults': None,
},
}
def test_arg_defaults(self):
def fn(a, b=1, c=None):
pass
assert visitor.visit_task(fn, ()) == {
'name': 'fn',
'path': (),
'doc': None,
'cron': None,
'argspec': {
'args': ['a', 'b', 'c'],
'varargs': None,
'keywords': None,
'defaults': (1, None),
},
}
def test_simple(self, fn):
t = monthly(task(fn))
data = visitor.visit_task(t, ())
assert data['cron'] == '0 0 0 * *'
def test_task_cron_hash(self, fn):
t = task(cron(time={'minute': '*'})(fn))
data = visitor.visit_task(t, ())
assert data['cron'] == {'time': {'minute': '*'}}
def test_task_cron(self, fn):
t = task(cron('* * * * *')(fn))
data = visitor.visit_task(t, ())
assert data['cron'] == '* * * * *'
