Add migration-wait-timeout to the helm chart

airflow/cli/cli_parser.py

@@ -448,7 +448,7 @@ def _check(value):
     ("-t", "--migration-wait-timeout"),
     help="timeout to wait for db to migrate ",
     type=int,
-    default=0,
+    default=60,
 )
 
 # webserver

chart/templates/_helpers.yaml

@@ -546,6 +546,7 @@ Create the name of the cleanup service account to use
   - airflow
   - db
   - check-migrations
+  - --migration-wait-timeout={{ .Values.images.migrationsWaitTimeout }}
   {{- else }}
   - python
   - -c

chart/tests/test_webserver.py

@@ -139,8 +139,8 @@ def test_should_add_extra_containers(self):
 
     @parameterized.expand(
         [
-            ("2.0.0", ["airflow", "db", "check-migrations"]),
-            ("2.1.0", ["airflow", "db", "check-migrations"]),
+            ("2.0.0", ["airflow", "db", "check-migrations", "--migration-wait-timeout=60"]),
+            ("2.1.0", ["airflow", "db", "check-migrations", "--migration-wait-timeout=60"]),
             ("1.10.2", ["python", "-c"]),
         ],
     )

chart/values.schema.json

@@ -372,6 +372,12 @@
                     "x-docsSection": "Images",
                     "default": false
                 },
+                "migrationsWaitTimeout": {
+                    "description": "The time (in seconds) to wait for the DB migrations to complete.",
+                    "type": "number",
+                    "x-docsSection": "Images",
+                    "default": 60
+                },
                 "pod_template": {
                     "description": "Configuration of the pod_template image.",
                     "type": "object",

chart/values.yaml

@@ -56,6 +56,8 @@ images:
   # will use the images from 'defaultAirflowRepository:defaultAirflowTag' values
   # to run and wait for DB migrations .
   useDefaultImageForMigration: false
+  # timeout (in seconds) for airflow-migrations to complete
+  migrationsWaitTimeout: 60
   pod_template:
     repository: ~
     tag: ~

tests/cli/commands/test_db_command.py

@@ -47,7 +47,7 @@ def test_cli_resetdb(self, mock_resetdb):
     def test_cli_check_migrations(self, mock_wait_for_migrations):
         db_command.check_migrations(self.parser.parse_args(['db', 'check-migrations']))
 
-        mock_wait_for_migrations.assert_called_once_with(timeout=0)
+        mock_wait_for_migrations.assert_called_once_with(timeout=60)
 
     @mock.patch("airflow.cli.commands.db_command.db.upgradedb")
     def test_cli_upgradedb(self, mock_upgradedb):