fix: wait for publish operation to complete before reporting success

src/index.test.ts

@@ -7,12 +7,13 @@ test("test upload test.zip artifact", async () => {
   const key = JSON.parse(await readFile("key.json", "utf8")) as Options
 
   const client = new EdgeAddonsAPI(key)
-  const resp = await client.submit({
+  const operationId = await client.submit({
     filePath: "test.zip",
     notes: "Test upload test.zip artifact"
   })
 
-  const publishStatus = await client.getPublishStatus(resp)
+  const publishStatus = await client.getPublishStatus(operationId)
 
-  expect(publishStatus.id).toBe(resp)
+  expect(publishStatus.id).toBe(operationId)
+  expect(publishStatus.status).toBe("Succeeded")
 })

src/index.ts

@@ -88,7 +88,9 @@ export class EdgeAddonsAPI {
       return
     }
 
-    return this.publish(notes)
+    const publishOperationId = await this.publish(notes)
+    await this.waitForPublish(publishOperationId)
+    return publishOperationId
   }
 
   async publish(notes = "") {
@@ -134,6 +136,35 @@ export class EdgeAddonsAPI {
       .json<OperationResponse>()
   }
 
+  async waitForPublish(
+    operationId: string,
+    retryCount = 10,
+    pollTime = 5000
+  ) {
+    let status: OperationResponse["status"]
+    let attempts = 0
+
+    while (status !== "Succeeded" && attempts < retryCount) {
+      const statusResp = await this.getPublishStatus(operationId)
+
+      if (statusResp.status === "Failed") {
+        throw new Error(
+          statusResp.message ||
+            statusResp.errorCode + ":" + (statusResp.errors || []).join(",")
+        )
+      } else if (statusResp.status === "InProgress") {
+        await wait(pollTime)
+      }
+
+      status = statusResp.status
+      attempts++
+    }
+
+    if (status !== "Succeeded") {
+      throw new Error("Publish operation timed out")
+    }
+  }
+
   async waitForUpload(
     operationId: string,
     retryCount = 5,