def test_run_error(mocks: Mocks, context: yogi.Context):
"""Checks the run() function in the error case"""
def fn(context, count, duration):
assert -1 == duration
return yogi.ErrorCode.UNKNOWN
mocks.MOCK_ContextRun(fn)
with pytest.raises(yogi.FailureException):
context.run()
def test_run(mocks: Mocks, context: yogi.Context):
"""Checks the run() function in the error-free case"""
def fn(context, count, duration):
assert context == 1234
assert count
assert 1000000 == duration
count.contents.value = 5
return yogi.ErrorCode.OK
mocks.MOCK_ContextRun(fn)
assert context.run(yogi.Duration.from_milliseconds(1)) == 5