fix: stabilise test

canonical · Sep 2, 2024 · 38ccf7d · 38ccf7d
1 parent 0816a02
commit 38ccf7d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/integration/sharding_tests/helpers.py b/tests/integration/sharding_tests/helpers.py
@@ -7,6 +7,7 @@
 from pytest_operator.plugin import OpsTest
 
 from ..helpers import METADATA, get_application_relation_data, get_secret_content
+from tenacity import retry, stop_after_attempt, wait_fixed
 
 SHARD_ONE_APP_NAME = "shard-one"
 SHARD_TWO_APP_NAME = "shard-two"
@@ -27,6 +28,7 @@ def count_users(mongos_client: MongoClient) -> int:
     return users_collection.count_documents({})
 
 
+@retry(stop=stop_after_attempt(10), wait=wait_fixed(15), reraise=True)
 async def get_related_username_password(
     ops_test: OpsTest, app_name: str, relation_name: str
 ) -> Tuple: