def test_sender_close(): receiver = mock.Mock() sender = _Sender(receiver, 'name', is_pattern=False) sender.close() assert receiver.mock_calls == [mock.call._close(sender, exc=None)] sender.close() assert receiver.mock_calls == [mock.call._close(sender, exc=None)] receiver.reset_mock() assert receiver.mock_calls == [] sender.close() assert receiver.mock_calls == []
async def test_sender(loop): receiver = mock.Mock() sender = _Sender(receiver, 'name', is_pattern=False) assert isinstance(sender, AbcChannel) assert sender.name == b'name' assert sender.is_pattern is False assert sender.is_active is True with pytest.raises(RuntimeError): await sender.get() assert receiver.mock_calls == [] sender.put_nowait(b'some data') assert receiver.mock_calls == [ mock.call._put_nowait(b'some data', sender=sender), ]