From 5d57571e96f47a6bebba0623746d0a17f70eb893 Mon Sep 17 00:00:00 2001
From: Mathias Jakobsen <mathias.jakobsen@overleaf.com>
Date: Wed, 4 Sep 2024 09:55:09 +0100
Subject: [PATCH] Merge pull request #20226 from overleaf/mj-restore-file-badge

[web] Add split test badge for file restore

GitOrigin-RevId: 59bbdd4cc5b76141a8bae06c2e7181e5a5c5ccc0
---
 .../history/components/diff-view/toolbar/toolbar.tsx     | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/services/web/frontend/js/features/history/components/diff-view/toolbar/toolbar.tsx b/services/web/frontend/js/features/history/components/diff-view/toolbar/toolbar.tsx
index ce352582e2..6087f3cc8d 100644
--- a/services/web/frontend/js/features/history/components/diff-view/toolbar/toolbar.tsx
+++ b/services/web/frontend/js/features/history/components/diff-view/toolbar/toolbar.tsx
@@ -7,6 +7,7 @@ import ToolbarRestoreFileButton from './toolbar-restore-file-button'
 import { isFileRemoved } from '../../../utils/file-diff'
 import ToolbarRestoreFileToVersionButton from './toolbar-restore-file-to-version-button'
 import { useFeatureFlag } from '@/shared/context/split-test-context'
+import SplitTestBadge from '@/shared/components/split-test-badge'
 
 type ToolbarProps = {
   diff: Nullable<Diff>
@@ -34,7 +35,13 @@ export default function Toolbar({ diff, selection }: ToolbarProps) {
         <ToolbarRestoreFileButton selection={selection} />
       ) : null}
       {showRestoreFileToVersionButton ? (
-        <ToolbarRestoreFileToVersionButton selection={selection} />
+        <>
+          <ToolbarRestoreFileToVersionButton selection={selection} />
+          <SplitTestBadge
+            splitTestName="revert-file"
+            displayOnVariants={['enabled']}
+          />
+        </>
       ) : null}
     </div>
   )