From 11117811aa5897ea26309a5e867541dfbe1b8832 Mon Sep 17 00:00:00 2001
From: Camila Belo <camilaibs@gmail.com>
Date: Tue, 3 Aug 2021 14:53:53 +0200
Subject: [PATCH] test(techdocs-common): fix by page mock for azure

Co-authored-by: Eric Peterson <ericpeterson@spotify.com>
Signed-off-by: Camila Belo <camilaibs@gmail.com>
---
 .../__mocks__/@azure/storage-blob.ts          | 44 +++++++++++++++----
 1 file changed, 36 insertions(+), 8 deletions(-)

diff --git a/packages/techdocs-common/__mocks__/@azure/storage-blob.ts b/packages/techdocs-common/__mocks__/@azure/storage-blob.ts
index f7303ce1cb..7279fc31eb 100644
--- a/packages/techdocs-common/__mocks__/@azure/storage-blob.ts
+++ b/packages/techdocs-common/__mocks__/@azure/storage-blob.ts
@@ -102,6 +102,36 @@ class BlockBlobClientFailUpload extends BlockBlobClient {
   }
 }
 
+class ContainerClientIterator {
+  private containerName: string;
+
+  constructor(containerName) {
+    this.containerName = containerName;
+  }
+
+  async next() {
+    if (
+      this.containerName === 'delete_stale_files_success' ||
+      this.containerName === 'delete_stale_files_error'
+    ) {
+      return {
+        value: {
+          segment: {
+            blobItems: [{ name: `stale_file.png` }],
+          },
+        },
+      };
+    }
+    return {
+      value: {
+        segment: {
+          blobItems: [],
+        },
+      },
+    };
+  }
+}
+
 export class ContainerClient {
   private readonly containerName;
 
@@ -126,14 +156,12 @@ export class ContainerClient {
     return new BlockBlobClient(blobName);
   }
 
-  listBlobsFlat({ prefix }: { prefix: string }) {
-    if (
-      this.containerName === 'delete_stale_files_success' ||
-      this.containerName === 'delete_stale_files_error'
-    ) {
-      return [{ name: `${prefix}stale_file.png` }];
-    }
-    return [];
+  listBlobsFlat() {
+    return {
+      byPage: () => {
+        return new ContainerClientIterator(this.containerName);
+      },
+    };
   }
 
   deleteBlob() {