Improve useFocusNode test

jamesplease · Jun 17, 2021 · 7703ff9 · 7703ff9
1 parent 57016e2
commit 7703ff9
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/hooks/use-focus-node.test.js b/src/hooks/use-focus-node.test.js
@@ -1,16 +1,24 @@
 import React from 'react';
 import '@testing-library/jest-dom';
 import { render, act } from '@testing-library/react';
-import { FocusRoot, FocusNode, useFocusNode, useSetFocus } from '../index';
+import {
+  FocusRoot,
+  FocusNode,
+  useFocusNode,
+  useSetFocus,
+  useFocusStoreDangerously,
+} from '../index';
 
 describe('useFocusNode', () => {
   it('returns the expected node', () => {
     let setFocus;
     let focusNode;
+    let focusStore;
 
     function TestComponent() {
       setFocus = useSetFocus();
       focusNode = useFocusNode('nodeA');
+      focusStore = useFocusStoreDangerously();
 
       return (
         <>
@@ -34,6 +42,8 @@ describe('useFocusNode', () => {
       })
     );
 
+    expect(focusNode).toBe(focusStore.getState().nodes.nodeA);
+
     act(() => setFocus('nodeB'));
 
     expect(focusNode).toEqual(
@@ -43,5 +53,7 @@ describe('useFocusNode', () => {
         isFocusedLeaf: false,
       })
     );
+
+    expect(focusNode).toBe(focusStore.getState().nodes.nodeA);
   });
 });