Use Response object to convert text to ReadableStream

AdiRishi · Mar 15, 2024 · 873b4a8 · 873b4a8
1 parent 157c46e
commit 873b4a8
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 12 deletions.
diff --git a/src/storage/storage-manager.ts b/src/storage/storage-manager.ts
@@ -37,6 +37,9 @@ export class StorageManager {
   public static async readableStreamToBuffer(stream: ReadableStream): Promise<ArrayBuffer> {
     return await new Response(stream).arrayBuffer();
   }
+  public static textToReadableStream(text: string): ReadableStream {
+    return new Response(text).body!;
+  }
 }
 
 export class InvalidStorageError extends Error {

diff --git a/tests/storage/kv-storage.test.ts b/tests/storage/kv-storage.test.ts
@@ -110,12 +110,7 @@ describe('kv-storage', () => {
     });
 
     test('can write stream value', async () => {
-      const stream = new ReadableStream({
-        start: (controller) => {
-          controller.enqueue('value1');
-          controller.close();
-        },
-      });
+      const stream = StorageManager.textToReadableStream('value1');
       await storage.write('key1', stream);
       const result = await storage.read('key1');
       const dataAsText = await StorageManager.readableStreamToText(result!);

diff --git a/tests/storage/r2-storage.test.ts b/tests/storage/r2-storage.test.ts
@@ -109,12 +109,7 @@ describe('r2-storage', () => {
     });
 
     test('can write stream value', async () => {
-      const stream = new ReadableStream({
-        start: (controller) => {
-          controller.enqueue('value1');
-          controller.close();
-        },
-      });
+      const stream = StorageManager.textToReadableStream('value1');
       await storage.write('key1', stream);
       const result = await storage.read('key1');
       const dataAsText = await StorageManager.readableStreamToText(result!);