fix(Searchable): confirm ES data propagate (#8251)

aws-amplify · Sep 22, 2021 · 8e5c791 · 8e5c791
1 parent d17309d
commit 8e5c791
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/...ges/graphql-transformers-e2e-tests/src/__tests__/SearchableModelTransformerV2.e2e.test.ts b/...ges/graphql-transformers-e2e-tests/src/__tests__/SearchableModelTransformerV2.e2e.test.ts
@@ -60,8 +60,32 @@ const createEntries = async () => {
   await runQuery(getCreatePostsMutation('testuser', 'test title', 170, 30, 77.7, true));
   // Waiting for the ES Cluster + Streaming Lambda infra to be setup
   await cf.wait(120, () => Promise.resolve());
+  await waitForESPropagate();
 };
 
+const waitForESPropagate = async (initialWaitSeconds = 5, maxRetryCount = 5 ) => {
+  const expectedCount = 8;
+  let waitInMilliseconds = initialWaitSeconds * 1000;
+  let currentRetryCount = 0;
+  let searchResponse;
+
+  do {
+    await new Promise(r => setTimeout(r, waitInMilliseconds));
+    searchResponse = await GRAPHQL_CLIENT.query(
+      `query {
+        searchPosts {
+          items {
+            id
+          }
+        }
+      }`,
+      {},
+    );
+    currentRetryCount += 1;
+    waitInMilliseconds = waitInMilliseconds * 2;
+  } while (searchResponse.data.searchPosts?.items?.length < expectedCount && currentRetryCount <= maxRetryCount);
+}
+
 beforeAll(async () => {
   const validSchema = `
     type Post @model @searchable {