Workspaces: Fix directory type

moto/workspaces/models.py

@@ -58,7 +58,8 @@ def __init__(
         self.modification_states: List[
             Dict[str, str]
         ] = []  # modify_workspace_properties
-        self.related_workspaces: List[Dict[str, str]] = []  # create_standy_workspace
+        # create_standy_workspace
+        self.related_workspaces: List[Dict[str, str]] = []
         self.data_replication_settings: Dict[str, Any] = {}
         # The properties of the standby WorkSpace related to related_workspaces
         self.standby_workspaces_properties: List[Dict[str, Any]] = []
@@ -134,14 +135,22 @@ def __init__(
         self.launch_time = directory.launch_time
         self.registration_code = registration_code
         if directory.directory_type == "ADConnector":
-            dir_subnet_ids = directory.connect_settings["SubnetIds"]  # type: ignore[index]
+            # type: ignore[index]
+            dir_subnet_ids = directory.connect_settings["SubnetIds"]
         else:
-            dir_subnet_ids = directory.vpc_settings["SubnetIds"]  # type: ignore[index]
+            # type: ignore[index]
+            dir_subnet_ids = directory.vpc_settings["SubnetIds"]
         self.subnet_ids = subnet_ids or dir_subnet_ids
         self.dns_ip_addresses = directory.dns_ip_addrs
         self.customer_username = "Administrator"
         self.iam_rold_id = f"arn:aws:iam::{account_id}:role/workspaces_DefaultRole"
-        self.directory_type = directory.directory_type
+        dir_type = directory.directory_type
+        if dir_type == "ADConnector":
+            self.directory_type = "AD_CONNECTOR"
+        elif dir_type == "SimpleAD":
+            self.directory_type = "SIMPLE_AD"
+        else:
+            self.directory_type = dir_type
         self.workspace_security_group_id = security_group_id
         self.state = "REGISTERED"
         # Default values for workspace_creation_properties
@@ -456,9 +465,9 @@ def register_workspace_directory(
             tenancy=tenancy,
             tags=tags,
         )
-        self.workspace_directories[workspace_directory.directory_id] = (
-            workspace_directory
-        )
+        self.workspace_directories[
+            workspace_directory.directory_id
+        ] = workspace_directory
 
     def describe_workspace_directories(
         self, directory_ids: Optional[List[str]] = None