From 961a4778f254b50a27d63dea94b4e35bfb08f410 Mon Sep 17 00:00:00 2001 From: Tres Seaver Date: Fri, 4 Oct 2019 14:11:35 -0400 Subject: [PATCH] tests(spanner): harden 'test_reload_instance' against EC Closes #8139 --- spanner/tests/system/test_system.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/spanner/tests/system/test_system.py b/spanner/tests/system/test_system.py index 730190444edf..59441af047c7 100644 --- a/spanner/tests/system/test_system.py +++ b/spanner/tests/system/test_system.py @@ -146,7 +146,13 @@ def test_reload_instance(self): # Make sure metadata unset before reloading. instance.display_name = None - instance.reload() + def _expected_display_name(instance): + return instance.display_name == Config.INSTANCE.display_name + + retry = RetryInstanceState(_expected_display_name) + + retry(instance.reload)() + self.assertEqual(instance.display_name, Config.INSTANCE.display_name) @unittest.skipUnless(CREATE_INSTANCE, "Skipping instance creation")