Changed numeric value to enum

CiscoDevNet · Jan 20, 2023 · 3188d53 · 3188d53
1 parent f0141bb
commit 3188d53
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vmngclient/api/templates/payloads/cisco_vpn/cisco_vpn_model.py b/vmngclient/api/templates/payloads/cisco_vpn/cisco_vpn_model.py
@@ -9,6 +9,7 @@
 from pydantic import validator  # type: ignore
 
 from vmngclient.api.templates.feature_template import FeatureTemplate
+from vmngclient.api.templates.payloads.aaa.aaa_model import VpnType
 from vmngclient.session import vManageSession
 
 
@@ -66,7 +67,7 @@ class CiscoVPNModel(FeatureTemplate):
 
     @validator("vpn_id")
     def check_id(cls, v, values):
-        if v not in [0, 512]:
+        if v not in [VpnType.VPN_TRANSPORT.value, VpnType.VPN_MANAGMENT.value]:
             if "tenant_org_name" not in values:
                 raise ValueError("Must enter the name of the organization.")
         return v