fix: respect falsy server snapshots and production context overrides

src/components/Provider.tsx

@@ -65,7 +65,7 @@ function Provider<A extends Action<string> = UnknownAction, S = unknown>(
     const baseContextValue = {
       store,
       subscription,
-      getServerState: serverState ? () => serverState : undefined,
+      getServerState: serverState != null ? () => serverState : undefined,
     }
 
     if (process.env.NODE_ENV === 'production') {

src/components/connect.tsx

@@ -758,8 +758,8 @@ function _connect<
                 'You must pass a valid React context consumer as `props.context`',
               )
             }
-            ResultContext = propsContext
           }
+          ResultContext = propsContext
         }
         return ResultContext
       }, [propsContext, Context])

test/components/connect.spec.tsx

@@ -2236,41 +2236,49 @@ describe('React', () => {
         expect(actualState).toEqual(expectedState)
       })
 
-      it('should use a custom context provider and consumer if passed as a prop to the component', () => {
-        class Container extends Component {
-          render() {
-            return <Passthrough />
+      it.for(['development', 'production'] as const)(
+        'should use a custom context prop in %s',
+        (nodeEnv, { onTestFinished }) => {
+          vi.stubEnv('NODE_ENV', nodeEnv)
+          onTestFinished(() => {
+            vi.unstubAllEnvs()
+          })
+
+          class Container extends Component {
+            render() {
+              return <Passthrough />
+            }
           }
-        }
 
-        const context = React.createContext<ReactReduxContextValue<
-          any,
-          AnyAction
-        > | null>(null)
+          const context = React.createContext<ReactReduxContextValue<
+            any,
+            AnyAction
+          > | null>(null)
 
-        let actualState
+          let actualState
 
-        const expectedState = { foos: {} }
-        const ignoredState = { bars: {} }
+          const expectedState = { foos: {} }
+          const ignoredState = { bars: {} }
 
-        const decorator = connect((state) => {
-          actualState = state
-          return { a: 42 }
-        })
-        const Decorated = decorator(Container)
+          const decorator = connect((state) => {
+            actualState = state
+            return { a: 42 }
+          })
+          const Decorated = decorator(Container)
 
-        const store1 = createStore(() => expectedState)
-        const store2 = createStore(() => ignoredState)
-        rtl.render(
-          <ProviderMock context={context} store={store1}>
-            <ProviderMock store={store2}>
-              <Decorated context={context} />
-            </ProviderMock>
-          </ProviderMock>,
-        )
+          const store1 = createStore(() => expectedState)
+          const store2 = createStore(() => ignoredState)
+          rtl.render(
+            <ProviderMock context={context} store={store1}>
+              <ProviderMock store={store2}>
+                <Decorated context={context} />
+              </ProviderMock>
+            </ProviderMock>,
+          )
 
-        expect(actualState).toEqual(expectedState)
-      })
+          expect(actualState).toEqual(expectedState)
+        },
+      )
 
       it('should ignore non-react-context values that are passed as a prop to the component', () => {
         class Container extends Component {

test/integration/ssr.spec.tsx

@@ -124,6 +124,44 @@ describe('New v8 serverState behavior', () => {
     vi.restoreAllMocks()
   })
 
+  it.for([0, false, ''] as const)(
+    'uses a falsy serverState value of %s',
+    (serverState) => {
+      const store = createStore((state: unknown = 'client') => state)
+
+      function Value() {
+        return <span>{String(useSelector((state: unknown) => state))}</span>
+      }
+
+      const markup = renderToString(
+        <Provider store={store} serverState={serverState}>
+          <Value />
+        </Provider>,
+      )
+
+      expect(markup).toBe(`<span>${String(serverState)}</span>`)
+    },
+  )
+
+  it.for([null, undefined] as const)(
+    'falls back to the store state when serverState is %s',
+    (serverState) => {
+      const store = createStore((state: unknown = 'client') => state)
+
+      function Value() {
+        return <span>{String(useSelector((state: unknown) => state))}</span>
+      }
+
+      const markup = renderToString(
+        <Provider store={store} serverState={serverState}>
+          <Value />
+        </Provider>,
+      )
+
+      expect(markup).toBe('<span>client</span>')
+    },
+  )
+
   it('Handles hydration correctly', async () => {
     const ssrStore = createStore(dataSlice.reducer)