From 0fa09cf02c4b29cf107de6391bcdc341975ef08a Mon Sep 17 00:00:00 2001 From: Mohammed Naser Date: Tue, 11 Jun 2024 11:30:09 -0400 Subject: [PATCH] Add retries for assertions to wait for reconcile --- magnum_cluster_api/tests/functional/test_driver.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/magnum_cluster_api/tests/functional/test_driver.py b/magnum_cluster_api/tests/functional/test_driver.py index e564e139..ead1a7a2 100644 --- a/magnum_cluster_api/tests/functional/test_driver.py +++ b/magnum_cluster_api/tests/functional/test_driver.py @@ -15,6 +15,7 @@ from unittest import mock import pytest +import tenacity from magnum import objects as magnum_objects # type: ignore from magnum.objects import fields # type: ignore from magnum.tests.unit.objects import utils # type: ignore @@ -32,6 +33,10 @@ def _assert_machine_deployment_config_matches_node_group(self, md, node_group): assert md is not None # TODO: more? + @tenacity.retry( + stop=tenacity.stop_after_attempt(10), + wait=tenacity.wait_fixed(1), + ) def _assert_machine_deployments_for_node_groups( self, *node_groups: magnum_objects.NodeGroup ):