updates to support python 3.10

amqtt/contexts.py

@@ -207,7 +207,7 @@ class BrokerConfig(Dictable):
         return dict_to_dataclass(data_class=BrokerConfig,
                                  data=d,
                                  config=DaciteConfig(
-                                     cast=[StrEnum],
+                                     cast=[StrEnum, ListenerType],
                                      strict=True)
                                  )
 

tests/test_broker_config.py

@@ -1,18 +1,17 @@
 import logging
 from typing import Any
 
-import pytest
+try:
-from dataclasses import dataclass, field
+    from enum import Enum, StrEnum
-from pathlib import Path
+except ImportError:
-from yaml import CLoader as Loader
+    # support for python 3.10
+    from enum import Enum
+    class StrEnum(str, Enum):  #type: ignore[no-redef]
+        pass
 
+from dacite import from_dict, Config
 
-import yaml
+from amqtt.contexts import BrokerConfig, ListenerType
-from dacite import from_dict, Config, UnexpectedDataError
-from enum import StrEnum
-
-from amqtt.broker import BrokerContext
-from amqtt.contexts import BrokerConfig, ListenerConfig, Dictable
 
 logger = logging.getLogger(__name__)
 
@@ -32,7 +31,7 @@ def test_entrypoint_broker_config(caplog):
     # cfg: dict[str, Any] = yaml.load(config, Loader=Loader)
 
 
-    broker_config = from_dict(data_class=BrokerConfig, data=test_cfg, config=Config(cast=[StrEnum]))
+    broker_config = from_dict(data_class=BrokerConfig, data=test_cfg, config=Config(cast=[StrEnum, ListenerType]))
     assert isinstance(broker_config, BrokerConfig)
 
     assert broker_config.plugins is None