From 3e98da1d7a6c6d31f2918a8c624477dd9ab78328 Mon Sep 17 00:00:00 2001
From: Lena Garber <lgarber@akamai.com>
Date: Fri, 7 Feb 2025 16:21:10 -0500
Subject: [PATCH] cool

---
 plugins/modules/database_mysql_v2.py                   |  4 ++--
 .../targets/database_mysql_v2_complex/tasks/main.yaml  | 10 +++++-----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/plugins/modules/database_mysql_v2.py b/plugins/modules/database_mysql_v2.py
index 889dd1fc..d0644d5f 100644
--- a/plugins/modules/database_mysql_v2.py
+++ b/plugins/modules/database_mysql_v2.py
@@ -158,7 +158,7 @@ def _wait_for_database_status(
         database: MySQLDatabase,
         desired_status: str,
         timeout: int = 30 * 60,
-        step: int = 5,
+        step: int = 2,
     ) -> None:
         def __poll_status() -> bool:
             database._api_get()
@@ -242,7 +242,7 @@ def _update(self, database: MySQLDatabase) -> None:
         updated_fields = handle_updates(
             database,
             params,
-            {"label", "allow_list", "updates", "type", "version"},
+            {"label", "allow_list", "cluster_size", "updates", "type", "version"},
             self.register_action,
         )
 
diff --git a/tests/integration/targets/database_mysql_v2_complex/tasks/main.yaml b/tests/integration/targets/database_mysql_v2_complex/tasks/main.yaml
index 378c5ff8..7e055b83 100644
--- a/tests/integration/targets/database_mysql_v2_complex/tasks/main.yaml
+++ b/tests/integration/targets/database_mysql_v2_complex/tasks/main.yaml
@@ -38,7 +38,7 @@
         allow_list:
           - 0.0.0.0/0
         updates:
-          duration: 2
+          duration: 4
           frequency: weekly
           hour_of_day: 22
           day_of_week: 3
@@ -57,7 +57,7 @@
           - (db_create.database.version | split('.'))[0] == engine_version
           - db_create.database.region == target_region
           - db_create.database.type == 'g6-nanode-1'
-          - db_create.database.updates.duration == 3
+          - db_create.database.updates.duration == 4
           - db_create.database.updates.frequency == 'weekly'
           - db_create.database.updates.hour_of_day == 22
           - db_create.database.updates.day_of_week == 3
@@ -72,7 +72,7 @@
         allow_list:
           - 10.0.0.3/32
         updates:
-          duration: 3
+          duration: 4
           frequency: weekly
           hour_of_day: 16
           day_of_week: 4
@@ -91,7 +91,7 @@
           - (db_update.database.version | split('.'))[0] == engine_updated_version
           - db_update.database.region == target_region
           - db_update.database.type == 'g6-standard-1'
-          - db_update.database.updates.duration == 3
+          - db_update.database.updates.duration == 4
           - db_update.database.updates.frequency == 'weekly'
           - db_update.database.updates.hour_of_day == 16
           - db_update.database.updates.day_of_week == 4
@@ -106,7 +106,7 @@
         allow_list:
           - 10.0.0.3/32
         updates:
-          duration: 3
+          duration: 4
           frequency: weekly
           hour_of_day: 16
           day_of_week: 4