async def test_mix_unmix():
"""
GIVEN
A mix, in non-priority mode, with two input channels.
WHEN
One input channel is unmix'd.
EXPECT
Item from the unmix'd channel is not taken.
"""
ch1 = create_channel()
ch2 = create_channel()
out = create_channel()
m = create_mix(out)
m.add_input(ch1)
m.add_input(ch2)
a = 'a'
b = 'b'
assert ch1.offer(a)
assert ch2.offer(b)
m.remove_input(ch2)
x = await out.take(timeout=0.05)
assert x == a
assert await out.take(timeout=0.05) is None
assert not ch2.empty()
out.close()
await asyncio.sleep(0.05)
async def test_mix_toggle_error_cases():
"""
WHEN
Call toggle with invalid arguments.
EXPECT
Errors to be raised.
"""
out = create_channel()
m = create_mix(out)
with pytest.raises(ValueError, match='no arguments'):
m.toggle()
with pytest.raises(ValueError, match='odd number of arguments'):
m.toggle(1)
ch = create_channel()
with pytest.raises(TypeError,
match='argument 2 must be a channel, not a str'):
m.toggle(ch, {}, '', {})
with pytest.raises(TypeError,
match='argument 1 must be a dict, not a str'):
m.toggle(ch, '')
with pytest.raises(KeyError,
match='argument 1 contains keys bar, foo, only '
'mute, pause, priority are allowed'):
m.toggle(ch, {'mute': True, 'foo': False, 'bar': True})
out.close()
await asyncio.sleep(0.05)
async def test_mix_priority_pause():
"""
GIVEN
A mix, in priority-mode=pause, with two input channels,
one with priority.
WHEN
An item is added to each input channel.
EXPECT
Item from priority channel is put on output channel. Item
from non-priority channel is not taken.
"""
ch1 = create_channel()
ch2 = create_channel()
out = create_channel()
m = create_mix(out)
m.priority_mode = m.PRIORITY_PAUSE
m.add_input(ch1)
m.toggle(ch1, {}, ch2, {'priority': True})
a = 'a'
b = 'b'
assert ch1.offer(a)
assert ch2.offer(b)
x = await out.take(timeout=0.05)
assert x == b
assert out.empty()
assert await out.take(timeout=0.05) is None
assert not ch1.empty()
out.close()
await asyncio.sleep(0.05)
async def test_mix_mute():
"""
GIVEN
A mix, in non-priority mode, with two input channels, one muted.
WHEN
An item is added to each input channel.
EXPECT
Item from non-muted channel is put on output channel. Item from
muted channel is taken but not put on output channel.
"""
ch1 = create_channel()
ch2 = create_channel()
out = create_channel()
m = create_mix(out)
m.add_input(ch1)
m.toggle(ch1, {}, ch2, {'mute': True})
a = 'a'
b = 'b'
assert ch1.offer(a)
assert ch2.offer(b)
x = await out.take(timeout=0.05)
assert x == a
assert out.empty()
assert ch2.empty()
out.close()
await asyncio.sleep(0.05)
async def test_mix_property_invalid():
"""
GIVEN
A mix.
WHEN
Assign invalid value to 'priority_mode' property.
EXPECT
Raises ValueError.
"""
ch = create_channel()
m = create_mix(ch)
with pytest.raises(ValueError, match='invalid priority mode'):
m.priority_mode = 'foo'
ch.close()
await asyncio.sleep(0.05)
async def test_mix_property():
"""
GIVEN
A mix.
WHEN
Set and get the 'priority_mode' property.
EXPECT
Able to set and get the mode.
"""
ch = create_channel()
m = create_mix(ch)
assert m.priority_mode == m.PRIORITY_OFF
m.priority_mode = m.PRIORITY_MUTE
assert m.priority_mode == m.PRIORITY_MUTE
m.priority_mode = m.PRIORITY_PAUSE
assert m.priority_mode == m.PRIORITY_PAUSE
ch.close()
await asyncio.sleep(0.05)
async def test_mix_remove_all_inputs():
"""
GIVEN
A max, in non-priority mode, with two input channels.
WHEN
All inputs are removed.
EXPECT
Items put on input channels are not taken.
"""
ch1 = create_channel()
ch2 = create_channel()
out = create_channel()
m = create_mix(out)
m.add_input(ch1)
m.add_input(ch2)
m.remove_all_inputs()
assert ch1.offer('a')
assert ch2.offer('b')
assert await out.take(timeout=0.05) is None
assert not ch1.empty()
assert not ch2.empty()
out.close()
await asyncio.sleep(0.05)
async def test_mix():
"""
GIVEN
A mix, in non-priority mode, with two input channels.
WHEN
An item is added to each input channel.
EXPECT
All items are transfered to output channel.
"""
ch1 = create_channel()
ch2 = create_channel()
out = create_channel()
m = create_mix(out)
m.add_input(ch1)
m.add_input(ch2)
a = 'a'
b = 'b'
assert ch1.offer(a)
assert ch2.offer(b)
x = await out.take(timeout=0.05)
y = await out.take(timeout=0.05)
assert not frozenset((x, y)).difference((a, b))
out.close()
await asyncio.sleep(0.05)
async def test_mute_restart():
"""
GIVEN
A mix with two muted input channels.
WHEN
An item is added to one input channel, but the channel is
removed from the mix before the item can be taken.
EXPECT
Item to not be taken from input channel.
"""
out = create_channel()
m = create_mix(out)
ch1 = create_channel()
ch2 = create_channel()
m.toggle(ch1, {'mute': True}, ch2, {'mute': True})
await asyncio.sleep(0.05) # Give mix a chance to start.
assert ch1.offer('a')
# ch1 is currently muted, removing it is expected to
# reset the take operation.
m.remove_input(ch1)
assert await out.take(timeout=0.05) is None
assert not ch1.empty()
out.close()
await asyncio.sleep(0.05) # Give mix a chance to clean up.