async def test_turning_a_toggle_on_turns_the_toggleset_on():
toggleset = ToggleSet()
toggle = await toggleset.make_toggle(False)
assert not toggleset.is_on()
assert toggleset.is_off()
await toggle.turn_to(True)
assert toggleset.is_on()
assert not toggleset.is_off()
async def test_turning_a_toggle_off_turns_the_toggleset_off(fn):
toggleset = ToggleSet(fn)
toggle = await toggleset.make_toggle(True)
assert toggleset.is_on() == True
assert toggleset.is_off() == False
await toggle.turn_to(False)
assert toggleset.is_on() == False
assert toggleset.is_off() == True
async def test_all_toggles_must_be_off_for_toggleset_to_be_off():
toggleset = ToggleSet()
toggle1 = await toggleset.make_toggle(True)
toggle2 = await toggleset.make_toggle(True)
assert toggleset.is_on()
assert not toggleset.is_off()
await toggle1.turn_to(False)
assert toggleset.is_on()
assert not toggleset.is_off()
await toggle2.turn_to(False)
assert not toggleset.is_on()
assert toggleset.is_off()
async def test_all_toggles_must_be_off_for_anytoggleset_to_be_off(fn):
toggleset = ToggleSet(fn)
toggle1 = await toggleset.make_toggle(True)
toggle2 = await toggleset.make_toggle(True)
assert toggleset.is_on() == True
assert toggleset.is_off() == False
await toggle1.turn_to(False)
assert toggleset.is_on() == True
assert toggleset.is_off() == False
await toggle2.turn_to(False)
assert toggleset.is_on() == False
assert toggleset.is_off() == True
async def test_created_as_off():
toggleset = ToggleSet()
assert len(toggleset) == 0
assert set(toggleset) == set()
assert Toggle() not in toggleset
assert not toggleset.is_on()
assert toggleset.is_off()
async def test_created_empty(fn, expected):
toggleset = ToggleSet(fn)
assert len(toggleset) == 0
assert set(toggleset) == set()
assert Toggle() not in toggleset
assert toggleset.is_on() == expected
assert toggleset.is_off() == (not expected)
async def test_blocking_when_operator_is_not_ready(resource, settings,
registry, indexers, caplog,
event_type, handlers,
timer):
caplog.set_level(logging.DEBUG)
operator_indexed = ToggleSet(all)
resource_listed = await operator_indexed.make_toggle()
resource_indexed = await operator_indexed.make_toggle()
with pytest.raises(asyncio.TimeoutError):
async with timer, async_timeout.timeout(0.2) as timeout:
await process_resource_event(
lifecycle=all_at_once,
registry=registry,
settings=settings,
resource=resource,
indexers=indexers,
memories=ResourceMemories(),
memobase=Memo(),
raw_event={
'type': event_type,
'object': {}
},
event_queue=asyncio.Queue(),
operator_indexed=operator_indexed,
resource_indexed=resource_indexed,
)
assert timeout.expired
assert 0.2 < timer.seconds < 0.4
assert operator_indexed.is_off()
assert set(operator_indexed) == {resource_listed}
assert not handlers.event_mock.called
async def test_dropping_an_unexistent_toggle():
toggleset = ToggleSet()
toggle = Toggle()
await toggleset.drop_toggle(toggle)
assert len(toggleset) == 0
assert set(toggleset) == set()
assert toggle not in toggleset
assert not toggleset.is_on()
assert toggleset.is_off()
async def test_dropping_a_turned_on_toggle():
toggleset = ToggleSet()
toggle = await toggleset.make_toggle(True)
await toggleset.drop_toggle(toggle)
assert len(toggleset) == 0
assert set(toggleset) == set()
assert toggle not in toggleset
assert not toggleset.is_on()
assert toggleset.is_off()
async def test_making_a_turned_on_toggle():
toggleset = ToggleSet()
toggle = await toggleset.make_toggle(True)
assert len(toggleset) == 1
assert set(toggleset) == {toggle}
assert toggle in toggleset
assert Toggle() not in toggleset
assert toggleset.is_on()
assert not toggleset.is_off()
async def test_dropping_an_unexistent_toggle(fn, expected):
toggleset = ToggleSet(fn)
toggle = Toggle()
await toggleset.drop_toggle(toggle)
assert len(toggleset) == 0
assert set(toggleset) == set()
assert toggle not in toggleset
assert toggleset.is_on() == expected
assert toggleset.is_off() == (not expected)
async def test_dropping_a_turned_on_toggle(fn, expected):
toggleset = ToggleSet(fn)
toggle = await toggleset.make_toggle(True)
await toggleset.drop_toggle(toggle)
assert len(toggleset) == 0
assert set(toggleset) == set()
assert toggle not in toggleset
assert toggleset.is_on() == expected
assert toggleset.is_off() == (not expected)
async def test_making_a_turned_off_toggle(fn):
toggleset = ToggleSet(fn)
toggle = await toggleset.make_toggle(False)
assert len(toggleset) == 1
assert set(toggleset) == {toggle}
assert toggle in toggleset
assert Toggle() not in toggleset
assert toggleset.is_on() == False
assert toggleset.is_off() == True
async def test_dropping_multiple_toggles():
toggleset = ToggleSet()
toggle1 = await toggleset.make_toggle(True)
toggle2 = Toggle()
await toggleset.drop_toggles([toggle1, toggle2])
assert len(toggleset) == 0
assert set(toggleset) == set()
assert toggle1 not in toggleset
assert toggle2 not in toggleset
assert not toggleset.is_on()
assert toggleset.is_off()
async def test_waiting_until_off_wakes_when_turned_off(timer):
toggleset = ToggleSet()
toggle = await toggleset.make_toggle(True)
async def delayed_turning_off(delay: float):
await asyncio.sleep(delay)
await toggle.turn_to(False)
with timer:
asyncio.create_task(delayed_turning_off(0.05))
await asyncio.wait_for(toggleset.wait_for(False), timeout=1.0)
assert toggleset.is_off()
assert timer.seconds < 0.5 # approx. 0.05 plus some code overhead
async def test_waiting_until_on_fails_when_not_turned_on():
toggleset = ToggleSet()
await toggleset.make_toggle(False)
with pytest.raises(asyncio.TimeoutError):
await asyncio.wait_for(toggleset.wait_for(True), timeout=0.1)
assert toggleset.is_off()