From 1b71d69fbdfcf6dac8573d7fd0a048315aa6e3f6 Mon Sep 17 00:00:00 2001
From: Jessica Chowdhury <jessica@trbl.design>
Date: Wed, 13 Aug 2025 12:41:34 +0100
Subject: [PATCH] chore: remove log and fix tests

---
 .../src/versions/drafts/replaceWithDraftIfAvailable.ts       | 2 +-
 packages/ui/src/elements/Status/index.tsx                    | 2 +-
 test/localization/e2e.spec.ts                                | 5 +++++
 test/versions/e2e.spec.ts                                    | 4 ++--
 4 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/packages/payload/src/versions/drafts/replaceWithDraftIfAvailable.ts b/packages/payload/src/versions/drafts/replaceWithDraftIfAvailable.ts
index f74225fa14..f4d0ac25e3 100644
--- a/packages/payload/src/versions/drafts/replaceWithDraftIfAvailable.ts
+++ b/packages/payload/src/versions/drafts/replaceWithDraftIfAvailable.ts
@@ -93,7 +93,7 @@ export const replaceWithDraftIfAvailable = async <T extends TypeWithID>({
   }
 
   let draft = versionDocs[0]
-  console.log('replaceWithDraftIfAvailable', versionDocs[0])
+
   if (!draft) {
     return doc
   }
diff --git a/packages/ui/src/elements/Status/index.tsx b/packages/ui/src/elements/Status/index.tsx
index ebd3f05ef6..6828b772ce 100644
--- a/packages/ui/src/elements/Status/index.tsx
+++ b/packages/ui/src/elements/Status/index.tsx
@@ -198,7 +198,7 @@ export const Status: React.FC = () => {
               />
             </React.Fragment>
           )}
-          {!isTrashed && canUpdate && statusToRender === 'changed' && (
+          {!isTrashed && canUpdate && statusToRender === 'changed' || statusToRender === 'draft' && (
             <React.Fragment>
               &nbsp;&mdash;&nbsp;
               <Button
diff --git a/test/localization/e2e.spec.ts b/test/localization/e2e.spec.ts
index f4f0666fff..407b45035b 100644
--- a/test/localization/e2e.spec.ts
+++ b/test/localization/e2e.spec.ts
@@ -685,6 +685,11 @@ describe('Localization', () => {
       await saveDocAndAssert(page, '#action-save-draft')
 
       await page.goto(urlPostsWithDrafts.list)
+
+      const columns = page.getByRole('button', { name: 'Columns' })
+      await columns.click()
+      await page.locator('#_status').click()
+
       await expect(page.locator('.row-1 .cell-title')).toContainText(spanTitle)
       await expect(page.locator('.row-1 .cell-_status')).toContainText('Draft')
 
diff --git a/test/versions/e2e.spec.ts b/test/versions/e2e.spec.ts
index 13b2428d25..0bcf5134fa 100644
--- a/test/versions/e2e.spec.ts
+++ b/test/versions/e2e.spec.ts
@@ -1019,7 +1019,7 @@ describe('Versions', () => {
 
       await textField.fill('spanish draft')
       await saveDocAndAssert(page, '#action-save-draft')
-      await expect(status).toContainText('Changed')
+      await expect(status).toContainText('Draft')
 
       await changeLocale(page, 'en')
       await textField.fill('english published')
@@ -1052,7 +1052,7 @@ describe('Versions', () => {
 
       const publishedDoc = data.docs[0]
 
-      expect(publishedDoc.text).toStrictEqual({
+      expect(publishedDoc?.text).toStrictEqual({
         en: 'english published',
         es: 'spanish published',
       })