Unit test covering all v2.0.1 datatypes and enums

ocpp/v201/datatypes.py

@@ -454,7 +454,7 @@ class IdTokenType:
     """
 
     id_token: str
-    type: enums.IdTokenType
+    type: enums.IdTokenEnumType
     additional_info: Optional[List[AdditionalInfoType]] = None
 
 

ocpp/v201/enums.py

@@ -10,6 +10,26 @@ class StrEnum(str, Enum):  # pragma: no cover
         pass  # pragma: no cover
 
 
+class DeprecatedEnumWrapper:
+    """
+    Since enums can't be subclassed in order to add a deprecation warning,
+    this class is included to help warn users of deprecated enums.
+    """
+
+    def __init__(self, enum_class, alias_name):
+        self.enum_class = enum_class
+        self.alias_name = alias_name
+
+    def __getattr__(self, name):
+        warn(
+            (
+                f"Enum '{self.alias_name}' is deprecated, "
+                + "instead use '{self.enum_class.__name__}'"
+            )
+        )
+        return getattr(self.enum_class, name)
+
+
 class Action(StrEnum):
     """An Action is a required part of a Call message."""
 
@@ -721,7 +741,7 @@ class HashAlgorithmType(StrEnum):
     sha512 = "SHA512"
 
 
-class IdTokenType(StrEnum):
+class IdTokenEnumType(StrEnum):
     """
     Allowable values of the IdTokenType field.
     """
@@ -736,6 +756,9 @@ class IdTokenType(StrEnum):
     no_authorization = "NoAuthorization"
 
 
+IdTokenType = DeprecatedEnumWrapper(IdTokenEnumType, "IdTokenType")
+
+
 class InstallCertificateStatusType(StrEnum):
     """
     InstallCertificateStatusEnumType is used by
@@ -1321,7 +1344,7 @@ class VPNType(StrEnum):
 # DataTypes
 
 
-class UnitOfMeasureType(StrEnum):
+class StandardizedUnitsOfMeasureType(StrEnum):
     """
     Allowable values of the optional "unit" field of a Value element, as used
     in MeterValues.req and StopTransaction.req messages. Default value of
@@ -1364,6 +1387,11 @@ class UnitOfMeasureType(StrEnum):
     k = "K"
 
 
+UnitOfMeasureType = DeprecatedEnumWrapper(
+    StandardizedUnitsOfMeasureType, "UnitOfMeasureType"
+)
+
+
 class StatusInfoReasonType(StrEnum):
     """
     Standardized reason codes for StatusInfo defined in Appendix 5. v1.3