def test_check_expand_interrupt(range_creator):
build_range = range_creator.create(range(10))
fetch_unless(build_range, lambda i: i == 0)
mp = build_range.mid_point
build_range.mid_point = lambda **kwa: mp() # do not interrupt in there
with pytest.raises(StopIteration):
build_range.check_expand(5, range_before, range_after,
interrupt=lambda: True)
def test_check_expand_interrupt(range_creator):
build_range = range_creator.create(list(range(10)))
fetch_unless(build_range, lambda i: i == 0)
mp = build_range.mid_point
build_range.mid_point = lambda **kwa: mp() # do not interrupt in there
with pytest.raises(StopIteration):
build_range.check_expand(5, range_before, range_after,
interrupt=lambda: True)
def test_check_expand(mocker, range_creator, size_expand, initial, fail_in,
expected, error):
log = mocker.patch('mozregression.build_range.LOG')
build_range = range_creator.create(initial)
fetch_unless(build_range, lambda i: i in fail_in)
build_range.check_expand(size_expand, range_before, range_after)
assert [b for b in build_range] == expected
if error:
assert log.critical.called
for i, call in enumerate(log.critical.call_args_list):
assert error[i] in call[0][0]
def test_check_expand(mocker, range_creator, size_expand, initial, fail_in,
expected, error):
log = mocker.patch('mozregression.build_range.LOG')
build_range = range_creator.create(initial)
fetch_unless(build_range, lambda i: i in fail_in)
build_range.check_expand(size_expand, range_before, range_after)
assert [b for b in build_range] == expected
if error:
assert log.critical.called
for i, call in enumerate(log.critical.call_args_list):
assert error[i] in call[0][0]
