From 426eff35b7f7201811d25257c27bc323a2c45a81 Mon Sep 17 00:00:00 2001
From: Colin <colin.rogers@dbtlabs.com>
Date: Fri, 14 Jun 2024 11:30:18 -0700
Subject: [PATCH] add retry to reduce flakiness of TestSecrets

---
 tests/functional/adapter/test_python_model.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tests/functional/adapter/test_python_model.py b/tests/functional/adapter/test_python_model.py
index 6bf0678c7..51346124f 100644
--- a/tests/functional/adapter/test_python_model.py
+++ b/tests/functional/adapter/test_python_model.py
@@ -198,6 +198,11 @@ class TestSecrets:
     def models(self):
         return {"secret_python_model.py": SECRETS_MODE}
 
+    # This test can be flaky because of delays in the integration being set up before trying to use it.
+    @pytest.fixture(scope="class")
+    def profiles_config_update(self):
+        return {"retry_all": True, "connect_retries": 3}
+
     def test_secrets(self, project):
         project.run_sql(
             "create or replace secret test_secret type = generic_string secret_string='secret value';"