test: prevent alchemy test suite from being rate-limited.

tests/system/test_alembic.py

@@ -18,6 +18,7 @@
 # CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
 import contextlib
+import time
 
 import pytest
 from sqlalchemy import Column, DateTime, Integer, String, Numeric
@@ -69,6 +70,7 @@ def test_alembic_scenario(alembic_table):
 
     assert alembic_table("account") is None
 
+    _prevent_rate_limiting()
     account = op.create_table(
         "account",
         Column("id", Integer, nullable=False),
@@ -96,6 +98,7 @@ def test_alembic_scenario(alembic_table):
         {"description": None, "id": 3, "name": "savings"},
     ]
 
+    _prevent_rate_limiting()
     op.add_column(
         "account", Column("last_transaction_date", DateTime, comment="when updated")
     )
@@ -107,6 +110,21 @@ def test_alembic_scenario(alembic_table):
         SchemaField("last_transaction_date", "DATETIME", description="when updated"),
     ]
 
+    _prevent_rate_limiting()
+    op.rename_table("account", "accounts")
+    assert alembic_table("account") is None
+    assert alembic_table("accounts", "schema") == [
+        SchemaField("id", "INTEGER", "REQUIRED"),
+        SchemaField("name", "STRING(50)", "REQUIRED", description="The name"),
+        SchemaField("description", "STRING(200)"),
+        SchemaField("last_transaction_date", "DATETIME", description="when updated"),
+    ]
+
+    _prevent_rate_limiting()
+    op.drop_table("accounts")
+    assert alembic_table("accounts") is None
+
+    _prevent_rate_limiting()
     op.create_table(
         "account_w_comment",
         Column("id", Integer, nullable=False),
@@ -115,29 +133,23 @@ def test_alembic_scenario(alembic_table):
         comment="This table has comments",
     )
     assert alembic_table("account_w_comment").description == "This table has comments"
+
+    _prevent_rate_limiting()
     op.drop_table_comment("account_w_comment")
     assert alembic_table("account_w_comment").description is None
 
+    _prevent_rate_limiting()
     op.drop_column("account_w_comment", "description")
     assert alembic_table("account_w_comment", "schema") == [
         SchemaField("id", "INTEGER", "REQUIRED"),
         SchemaField("name", "STRING(50)", "REQUIRED", description="The name"),
     ]
 
+    _prevent_rate_limiting()
     op.drop_table("account_w_comment")
     assert alembic_table("account_w_comment") is None
 
-    op.rename_table("account", "accounts")
-    assert alembic_table("account") is None
-    assert alembic_table("accounts", "schema") == [
-        SchemaField("id", "INTEGER", "REQUIRED"),
-        SchemaField("name", "STRING(50)", "REQUIRED", description="The name"),
-        SchemaField("description", "STRING(200)"),
-        SchemaField("last_transaction_date", "DATETIME", description="when updated"),
-    ]
-    op.drop_table("accounts")
-    assert alembic_table("accounts") is None
-
+    _prevent_rate_limiting()
     op.create_table(
         "transactions",
         Column("account", Integer, nullable=False),
@@ -146,23 +158,39 @@ def test_alembic_scenario(alembic_table):
         bigquery_time_partitioning=TimePartitioning(field="transaction_time"),
     )
 
+    _prevent_rate_limiting()
     op.alter_column("transactions", "amount", nullable=True)
     assert alembic_table("transactions", "schema") == [
         SchemaField("account", "INTEGER", "REQUIRED"),
         SchemaField("transaction_time", "DATETIME", "REQUIRED"),
         SchemaField("amount", "NUMERIC(11, 2)"),
     ]
 
+    _prevent_rate_limiting()
     op.create_table_comment("transactions", "Transaction log")
     assert alembic_table("transactions").description == "Transaction log"
 
+    _prevent_rate_limiting()
     op.drop_table("transactions")
 
     # Another thing we can do is alter the datatype of a nullable column,
     # if allowed by BigQuery's type coercion rules
+    _prevent_rate_limiting()
     op.create_table("identifiers", Column("id", Integer))
 
+    _prevent_rate_limiting()
     op.alter_column("identifiers", "id", type_=Numeric)
     assert alembic_table("identifiers", "schema") == [SchemaField("id", "NUMERIC")]
 
+    _prevent_rate_limiting()
     op.drop_table("identifiers")
+
+
+def _prevent_rate_limiting():
+    """
+    Adds a delay between operations to prevent exceeding the "Maximum rate of table metadata update operations per table".
+
+    The rate limit is set to a maximum of 5 operations per 10 seconds. By introducing a sleep of 2 seconds
+    before each operation, this function helps to ensure that the rate limit is not exceeded.
+    """
+    time.sleep(2)