From f1e822f817a35963a1d397068b89b2667c0b1e78 Mon Sep 17 00:00:00 2001
From: Eugene Shvets <3442792+chaosrealm@users.noreply.github.com>
Date: Thu, 21 Nov 2024 16:33:08 -0800
Subject: [PATCH] Fix `test streaming content` to reflect latest API changes
 (#54)

---
 src/content.test.ts | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/content.test.ts b/src/content.test.ts
index 744658e..a52f74a 100644
--- a/src/content.test.ts
+++ b/src/content.test.ts
@@ -206,7 +206,7 @@ test("test streaming content", { timeout: 180000 }, async () => {
   });
 
   let sawPlan = false;
-  let sawDraft = false;
+  let sawWriteSection = false;
   let sawEditorial = false;
 
   statusStream.on("data", (data) => {
@@ -216,8 +216,8 @@ test("test streaming content", { timeout: 180000 }, async () => {
       case "plan":
         sawPlan = true;
         break;
-      case "first-draft":
-        sawDraft = true;
+      case "write-section":
+        sawWriteSection = true;
         break;
       case "editorial":
         sawEditorial = true;
@@ -233,7 +233,7 @@ test("test streaming content", { timeout: 180000 }, async () => {
   expect(contentResult.version).toBe(0);
   expect(contentResult.commands.length).toBe(1);
   expect(sawPlan).toBe(true);
-  expect(sawDraft).toBe(true);
+  expect(sawWriteSection).toBe(true);
   expect(sawEditorial).toBe(true);
   expect(contentResult.status).toBe(ContentStatus.Draft);
   expect(contentResult.publishedVersion).toBe(undefined);