Python Context.poll_one Examples

Example #1

File: test_context.py Project: yohummus/yogi-framework

def test_poll_one(mocks: Mocks, context: yogi.Context):
    """Checks the poll_one() function in the error-free case"""
    def fn(context, count):
        assert context == 1234
        assert count
        count.contents.value = 1
        return yogi.ErrorCode.OK

    mocks.MOCK_ContextPollOne(fn)
    assert context.poll_one() == 1

Example #2

File: test_context.py Project: yohummus/yogi-framework

def test_poll_one_error(mocks: Mocks, context: yogi.Context):
    """Checks the poll_one() function in the error case"""
    mocks.MOCK_ContextPollOne(lambda *_: yogi.ErrorCode.WRONG_OBJECT_TYPE)
    with pytest.raises(yogi.FailureException):
        context.poll_one()