fixed eventually_consistent usage

daniel-sanche · daniel-sanche · commit 48f645ec95b6 · 2020-04-03T17:42:52.000-07:00
diff --git a/kms/api-client/snippets_test.py b/kms/api-client/snippets_test.py
@@ -180,8 +180,7 @@ def test_key_policy(self):
                                                  self.member,
                                                  self.role)
 
-        @eventually_consistent.call(exceptions=[Aborted, AssertionError])
-        def _():
+        def check_policy():
             policy = snippets.get_crypto_key_policy(self.project_id,
                                                     self.location,
                                                     self.keyring_id,
@@ -191,6 +190,8 @@ def _():
                 if b.role == self.role and self.member in b.members:
                     found = True
             assert found
+        eventually_consistent.call(check_policy,
+                                   exceptions=[Aborted, AssertionError])
         # remove member
         snippets.remove_member_from_crypto_key_policy(self.project_id,
                                                       self.location,
@@ -199,8 +200,7 @@ def _():
                                                       self.member,
                                                       self.role)
 
-        @eventually_consistent.call(exceptions=[Aborted, AssertionError])
-        def _():
+        def check_policy():
             policy = snippets.get_crypto_key_policy(self.project_id,
                                                     self.location,
                                                     self.keyring_id,
@@ -210,6 +210,8 @@ def _():
                 if b.role == self.role and self.member in b.members:
                     found = True
             assert not found
+        eventually_consistent.call(check_policy,
+                                   exceptions=[Aborted, AssertionError])
 
     def test_symmetric_encrypt_decrypt(self):
         cipher_bytes = snippets.encrypt_symmetric(self.project_id,
