diff --git a/amqtt/mqtt/packet.py b/amqtt/mqtt/packet.py index 38ad4aa..aeab916 100644 --- a/amqtt/mqtt/packet.py +++ b/amqtt/mqtt/packet.py @@ -9,7 +9,8 @@ except ImportError: UTC = timezone.utc from struct import unpack -from typing_extensions import Generic, Self, TypeVar +from typing import Generic +from typing_extensions import Self, TypeVar from amqtt.adapters import ReaderAdapter, WriterAdapter from amqtt.codecs_amqtt import bytes_to_hex_str, decode_packet_id, int_to_bytes, read_or_raise diff --git a/tests/test_cli.py b/tests/test_cli.py index 239c3d8..766fb00 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -56,18 +56,21 @@ async def broker(config_file): def test_cli_help_messages(): """Test that help messages are displayed correctly.""" + env = os.environ.copy() + env['NO_COLOR'] = "1" + amqtt_path = "amqtt" - output = subprocess.check_output([amqtt_path, "--help"]) + output = subprocess.check_output([amqtt_path, "--help"], env=env) assert "Usage: amqtt" in output.decode("utf-8") amqtt_sub_path = "amqtt_sub" - output = subprocess.check_output([amqtt_sub_path, "--help"]) + output = subprocess.check_output([amqtt_sub_path, "--help"], env=env) assert "Usage: amqtt_sub" in output.decode("utf-8") amqtt_pub_path = "amqtt_pub" - output = subprocess.check_output([amqtt_pub_path, "--help"]) + output = subprocess.check_output([amqtt_pub_path, "--help"], env=env) assert "Usage: amqtt_pub" in output.decode("utf-8")