Set value on adding setter in createGlobalState

src/factory/createGlobalState.ts

@@ -35,6 +35,7 @@ export function createGlobalState<S>(initialState?: S) {
     useIsomorphicLayoutEffect(() => {
       if (!store.setters.includes(stateSetter)) {
         store.setters.push(stateSetter);
+        storeSetter(store.state);
       }
     });
 

tests/createGlobalState.test.ts → tests/createGlobalState.test.tsx

@@ -1,5 +1,6 @@
-import { act, renderHook } from '@testing-library/react-hooks';
-import createGlobalState from '../src/factory/createGlobalState';
+import { act, renderHook, render } from '@testing-library/react';
+import React from 'react';
+import createGlobalState from 'src/hooks/createGlobalState';
 
 describe('useGlobalState', () => {
   it('should be defined', () => {
@@ -71,6 +72,23 @@ describe('useGlobalState', () => {
     });
   });
 
+  it('should set ref correctly', async () => {
+    const useGlobalValue = createGlobalState<Element>();
+    const CheckComponent = ({ stateValue1 }) => {
+      const [stateValue2] = useGlobalValue();
+      return <>{String(stateValue2 !== undefined && stateValue2 === stateValue1)}</>;
+    };
+
+    const WrapperComponent = () => {
+      const [stateValue, setStateValue] = useGlobalValue();
+      return <div ref={setStateValue}>
+        <CheckComponent stateValue1={stateValue} />
+      </div>;
+    };
+    const { findByText } = render(<WrapperComponent />);
+    expect(await findByText('true')).toBeDefined();
+  });
+
   it('initializes with function', () => {
     const useGlobalValue = createGlobalState(() => 0);
     const { result: result1 } = renderHook(() => useGlobalValue());