#2618 - Fixed the error that had NIC traffic describe state with enums as keys in the dict

src/primaite/simulator/network/hardware/base.py

@@ -37,6 +37,7 @@ from primaite.simulator.system.core.sys_log import SysLog
 from primaite.simulator.system.processes.process import Process
 from primaite.simulator.system.services.service import Service
 from primaite.simulator.system.software import IOSoftware
+from primaite.utils.converters import convert_dict_enum_keys_to_enum_values
 from primaite.utils.validators import IPV4Address
 
 IOSoftwareClass = TypeVar("IOSoftwareClass", bound=IOSoftware)
@@ -151,7 +152,7 @@ class NetworkInterface(SimComponent, ABC):
         )
         if CAPTURE_NMNE:
             state.update({"nmne": {k: v for k, v in self.nmne.items()}})
-        state.update({"traffic": self.traffic})
+        state.update({"traffic": convert_dict_enum_keys_to_enum_values(self.traffic)})
         return state
 
     @abstractmethod

src/primaite/utils/converters.py

@@ -0,0 +1,25 @@
+from enum import Enum
+from typing import Any, Dict
+
+
+def convert_dict_enum_keys_to_enum_values(d: Dict[Any, Any]) -> Dict[Any, Any]:
+    """
+    Convert dictionary keys from enums to their corresponding values.
+
+    :param d: dict
+        The dictionary with enum keys to be converted.
+    :return: dict
+        The dictionary with enum values as keys.
+    """
+    result = {}
+    for key, value in d.items():
+        if isinstance(key, Enum):
+            new_key = key.value
+        else:
+            new_key = key
+
+        if isinstance(value, dict):
+            result[new_key] = convert_dict_enum_keys_to_enum_values(value)
+        else:
+            result[new_key] = value
+    return result