chore: ensure reset_gitlab() succeeds

JohnVillalovos · JohnVillalovos · commit e409638c181d · 2021-12-28T14:05:16.000-08:00
Ensure reset_gitlab() succeeds by waiting to make sure everything has
been deleted as expected. If the timeout is exceeded fail the test.

Not using `wait_for_sidekiq` as it didn't work. During testing I
didn't see any sidekiq processes as being busy even though not
everything was deleted.
diff --git a/tests/functional/conftest.py b/tests/functional/conftest.py
@@ -7,6 +7,10 @@
 import pytest
 
 import gitlab
+import gitlab.base
+
+SLEEP_INTERVAL = 0.1
+TIMEOUT = 60  # seconds before timeout will occur
 
 
 @pytest.fixture(scope="session")
@@ -30,6 +34,37 @@ def reset_gitlab(gl):
         if user.username != "root":
             user.delete(hard_delete=True)
 
+    max_iterations = int(TIMEOUT / SLEEP_INTERVAL)
+
+    # Ensure everything has been reset
+    start_time = time.perf_counter()
+
+    def wait_for_list_empty(
+        rest_manager: gitlab.base.RESTManager, description: str
+    ) -> None:
+        """Wait for the list() to become empty or fail test if timeout is exceeded"""
+        for _ in range(max_iterations):
+            if rest_manager.list() == []:
+                break
+            time.sleep(SLEEP_INTERVAL)
+        assert rest_manager.list() == [], (
+            f"Did not delete all {description}. "
+            f"Elapsed_time: {time.perf_counter() - start_time}"
+        )
+
+    wait_for_list_empty(rest_manager=gl.projects, description="projects")
+    wait_for_list_empty(rest_manager=gl.groups, description="groups")
+    wait_for_list_empty(rest_manager=gl.variables, description="variables")
+
+    for _ in range(max_iterations):
+        if len(gl.users.list()) <= 1:
+            break
+        time.sleep(SLEEP_INTERVAL)
+    assert [user.username for user in gl.users.list()] == ["root"], (
+        f"Did not delete all users. "
+        f"elapsed_time: {time.perf_counter() - start_time}"
+    )
+
 
 def set_token(container, fixture_dir):
     set_token_rb = fixture_dir / "set_token.rb"
