From ad80cf17528a5a7a184f11bcbc62969fcadcf9ef Mon Sep 17 00:00:00 2001
From: Mike Friesen <mfriesen@gmail.com>
Date: Thu, 17 Oct 2024 08:23:10 -0500
Subject: [PATCH] update

---
 .../testutils/aws/FkqDocumentService.java     | 27 +++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/fkq-test-utils/src/main/java/com/formkiq/testutils/aws/FkqDocumentService.java b/fkq-test-utils/src/main/java/com/formkiq/testutils/aws/FkqDocumentService.java
index 01b3a1fa..d4069a72 100644
--- a/fkq-test-utils/src/main/java/com/formkiq/testutils/aws/FkqDocumentService.java
+++ b/fkq-test-utils/src/main/java/com/formkiq/testutils/aws/FkqDocumentService.java
@@ -709,4 +709,31 @@ public static GetDocumentVersionsResponse waitForDocumentVersions(final ApiClien
       TimeUnit.SECONDS.sleep(1);
     }
   }
+
+  /**
+   * Wait for Document Attribute.
+   *
+   * @param client {@link ApiClient}
+   * @param siteId {@link String}
+   * @param documentId {@link String}
+   * @param attributeKey {@link String}
+   * @return DocumentAttribute
+   * @throws InterruptedException InterruptedException
+   */
+  public static DocumentAttribute waitForDocumentAtrribute(final ApiClient client, final String siteId,
+                                                       final String documentId, final String attributeKey) throws InterruptedException {
+    DocumentAttributesApi documentAttributesApi = new DocumentAttributesApi(client);
+
+    while (true) {
+
+      try {
+        return documentAttributesApi.getDocumentAttribute(documentId, attributeKey, siteId)
+                .getAttribute();
+      } catch (ApiException e) {
+        // ignore
+      }
+
+      TimeUnit.SECONDS.sleep(1);
+    }
+  }
 }