def test_explict(self):
m = Mock()
runner = Runner()
runner.add(m.job1, label='one')
class MyPlug(Plug):
explicit = True
def helper(self):
m.plug_one()
@insert()
def one(self):
self.helper()
plug = MyPlug()
plug.add_to(runner)
runner()
compare([
call.job1(),
call.plug_one()
], actual=m.mock_calls)
verify(runner,
(m.job1, set()),
(plug.one, {'one'}),
)
def test_append(self):
m = Mock()
runner = Runner()
runner.add(m.job1, label='one')
class MyPlug(Plug):
@append()
def run_plug(self):
m.do_it()
plug = MyPlug()
plug.add_to(runner)
runner()
compare([
call.job1(),
call.do_it()
], actual=m.mock_calls)
verify(runner,
(m.job1, {'one'}),
(plug.run_plug, set()),
)
def test_remap_name(self):
m = Mock()
runner = Runner()
runner.add(m.job1, label='one')
class MyPlug(Plug):
@insert(label='one')
def run_plug(self):
m.plug_one()
plug = MyPlug()
plug.add_to(runner)
runner()
compare([
call.job1(),
call.plug_one()
], m.mock_calls)
verify(runner,
(m.job1, set()),
(plug.run_plug, {'one'}),
)
def test_ignore(self):
m = Mock()
runner = Runner()
runner.add(m.job1, label='one')
class MyPlug(Plug):
@ignore()
def helper(self): # pragma: no cover
m.plug_bad()
def one(self):
m.plug_good()
plug = MyPlug()
plug.add_to(runner)
runner()
compare([
call.job1(),
call.plug_good()
], actual=m.mock_calls)
verify(runner,
(m.job1, set()),
(plug.one, {'one'}),
)
def test_simple(self):
m = Mock()
runner = Runner()
runner.add(m.job1, label='one')
runner.add(m.job2)
runner.add(m.job3, label='three')
runner.add(m.job4)
class MyPlug(Plug):
def one(self):
m.plug_one()
def three(self):
m.plug_two()
plug = MyPlug()
plug.add_to(runner)
runner()
compare([
call.job1(), call.plug_one(), call.job2(),
call.job3(), call.plug_two(), call.job4()
], m.mock_calls)
verify(runner,
(m.job1, set()),
(plug.one, {'one'}),
(m.job2, set()),
(m.job3, set()),
(plug.three, {'three'}),
(m.job4, set()),
)
def test_requirements_and_returns(self):
m = Mock()
@returns('r1')
def job1():
m.job1()
return 1
@requires('r2')
def job3(r):
m.job3(r)
runner = Runner()
runner.add(job1, label='point')
runner.add(job3)
class MyPlug(Plug):
@requires('r1')
@returns('r2')
def point(self, r):
m.point(r)
return 2
plug = MyPlug()
plug.add_to(runner)
runner()
compare([
call.job1(),
call.point(1),
call.job3(2),
], m.mock_calls)
verify(
runner,
(job1, set()),
(plug.point, {'point'}),
(job3, set()),
)
def test_requirements_and_returns(self):
m = Mock()
@returns('r1')
def job1():
m.job1()
return 1
@requires('r2')
def job3(r):
m.job3(r)
runner = Runner()
runner.add(job1, label='point')
runner.add(job3)
class MyPlug(Plug):
@requires('r1')
@returns('r2')
def point(self, r):
m.point(r)
return 2
plug = MyPlug()
plug.add_to(runner)
runner()
compare([
call.job1(), call.point(1), call.job3(2),
], m.mock_calls)
verify(runner,
(job1, set()),
(plug.point, {'point'}),
(job3, set()),
)
def test_add_plug(self):
m = Mock()
runner = Runner()
runner.add(m.job1, label='one')
class MyPlug(Plug):
def one(self):
m.plug_one()
plug = MyPlug()
runner.add(plug)
runner()
compare([
call.job1(), call.plug_one()
], m.mock_calls)
verify(runner,
(m.job1, set()),
(plug.one, {'one'}),
)