From dd8582a9c9025dac3d2c56802eb6b4ca5d6a54ad Mon Sep 17 00:00:00 2001
From: Blake Jones <bjones325@gatech.edu>
Date: Fri, 19 Apr 2024 14:24:40 -0400
Subject: [PATCH] Added support for mw1.xlarge and mw1.2xlarge environment
 classes

---
 README.md    | 2 +-
 variables.tf | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 97afcb8..422078c 100644
--- a/README.md
+++ b/README.md
@@ -135,7 +135,7 @@ No modules.
 | <a name="input_create_s3_bucket"></a> [create\_s3\_bucket](#input\_create\_s3\_bucket) | Create new S3 bucket for MWAA. | `string` | `true` | no |
 | <a name="input_create_security_group"></a> [create\_security\_group](#input\_create\_security\_group) | Create security group for MWAA | `bool` | `true` | no |
 | <a name="input_dag_s3_path"></a> [dag\_s3\_path](#input\_dag\_s3\_path) | (Required) The relative path to the DAG folder on your Amazon S3 storage bucket. For example, dags. | `string` | `"dags"` | no |
-| <a name="input_environment_class"></a> [environment\_class](#input\_environment\_class) | (Optional) Environment class for the cluster. Possible options are mw1.small, mw1.medium, mw1.large.<br>Will be set by default to mw1.small. Please check the AWS Pricing for more information about the environment classes. | `string` | `"mw1.small"` | no |
+| <a name="input_environment_class"></a> [environment\_class](#input\_environment\_class) | (Optional) Environment class for the cluster. Possible options are mw1.small, mw1.medium, mw1.large, mw1.xlarge, mw1.2xlarge.<br>Will be set by default to mw1.small. Please check the AWS Pricing for more information about the environment classes. | `string` | `"mw1.small"` | no |
 | <a name="input_execution_role_arn"></a> [execution\_role\_arn](#input\_execution\_role\_arn) | (Required) The Amazon Resource Name (ARN) of the task execution role that the Amazon MWAA and its environment can assume<br>Mandatory if `create_iam_role=false` | `string` | `null` | no |
 | <a name="input_force_detach_policies"></a> [force\_detach\_policies](#input\_force\_detach\_policies) | IAM role Force detach policies | `bool` | `false` | no |
 | <a name="input_iam_role_additional_policies"></a> [iam\_role\_additional\_policies](#input\_iam\_role\_additional\_policies) | Additional policies to be added to the IAM role | `map(string)` | `{}` | no |
diff --git a/variables.tf b/variables.tf
index 2d3ac3f..bab2232 100644
--- a/variables.tf
+++ b/variables.tf
@@ -31,15 +31,15 @@ variable "dag_s3_path" {
 
 variable "environment_class" {
   description = <<-EOD
-  (Optional) Environment class for the cluster. Possible options are mw1.small, mw1.medium, mw1.large.
+  (Optional) Environment class for the cluster. Possible options are mw1.small, mw1.medium, mw1.large, mw1.xlarge, mw1.2xlarge.
   Will be set by default to mw1.small. Please check the AWS Pricing for more information about the environment classes.
   EOD
   type        = string
   default     = "mw1.small"
 
   validation {
-    condition     = contains(["mw1.small", "mw1.medium", "mw1.large"], var.environment_class)
-    error_message = "Invalid input, options: \"mw1.small\", \"mw1.medium\", \"mw1.large\"."
+    condition     = contains(["mw1.small", "mw1.medium", "mw1.large", "mw1.xlarge", "mw1.2xlarge"], var.environment_class)
+    error_message = "Invalid input, options: \"mw1.small\", \"mw1.medium\", \"mw1.large\", \"mw1.xlarge\", \"mw1.2xlarge\"."
   }
 }