fix(FocusManager): setFocus validates and fixes provided path

packages/react-sunbeam/src/focus/FocusManager.spec.tsx

@@ -0,0 +1,32 @@
+import * as React from "react"
+import { act, render } from "@testing-library/react"
+import { SunbeamProvider } from "./components/SunbeamProvider"
+import { Focusable } from "./components/Focusable"
+import { FocusManager } from "./FocusManager"
+
+describe("FocusManager", () => {
+    describe("setFocus", () => {
+        it("should validate and fix the provided focusPath", () => {
+            const focusManager = new FocusManager({ initialFocusPath: ["left-parent", "left-child"] })
+
+            render(
+                <SunbeamProvider focusManager={focusManager}>
+                    <Focusable focusKey="left-parent">
+                        <Focusable focusKey="left-child">Left child</Focusable>
+                    </Focusable>
+                    <Focusable focusKey="right-parent">
+                        <Focusable focusKey="right-child">Right child</Focusable>
+                    </Focusable>
+                </SunbeamProvider>
+            )
+
+            expect(focusManager.getFocusPath()).toEqual(["left-parent", "left-child"])
+
+            act(() => {
+                focusManager.setFocus(["right-parent"])
+            })
+
+            expect(focusManager.getFocusPath()).toEqual(["right-parent", "right-child"])
+        })
+    })
+})

packages/react-sunbeam/src/focus/FocusManager.ts

@@ -36,7 +36,11 @@ export class FocusManager {
     }
 
     public setFocus(focusPath: FocusPath): void {
-        this.focusPath = focusPath
+        if (!this.focusableRoot) return
+
+        const fixedFocusPath = validateAndFixFocusPathIfNeeded(focusPath, this.focusableRoot)
+        this.focusPath = fixedFocusPath ?? focusPath
+
         this.notifySubscribers()
     }