feat: adds withMockProvider to testUtils

src/index.ts

@@ -3,6 +3,7 @@ import { ContextType } from 'react';
 import connectContext, { connectContextFactory, useConnectedContextFactory, mergedConnectContextFactory, useMergedConnectedContextFactory } from './connectContext';
 import createContextProvider from './createContextProvider';
 import createMergedStore from './createMergedStore';
+import { withMockProvider } from './testUtils';
 
 export default createContextProvider;
 export {
@@ -13,4 +14,5 @@ export {
   useMergedConnectedContextFactory,
   createMergedStore,
   ContextType,
+  withMockProvider,
 };

src/testUtils.tsx

@@ -0,0 +1,18 @@
+import React from 'react';
+
+import { getProviderContext } from './createContextProvider';
+import { KeyValue } from './typings';
+
+function withMockProvider(Provider: any, Component: React.ReactNode) {
+    const Context: React.Context<any> = getProviderContext(Provider);
+
+    return ({ state = {}, actions = {} }: { state: KeyValue, actions: KeyValue }) => (
+        <Context.Provider value={{ state, actions }}>
+            {Component}
+        </Context.Provider>
+    );
+}
+
+export {
+    withMockProvider,
+};

src/typings.d.ts

@@ -12,7 +12,8 @@ export type ActionCreators = {
     [key: string]: Action;
 };
 
-export type CustomProvider = React.ComponentType<any> | React.FC<any>;
+export type CustomComponent = React.ComponentType<any> | React.FC<any>;
+export type CustomProvider = CustomComponent;
 
 export type CreateContextProviderReturn = [
     CustomProvider,
@@ -44,7 +45,7 @@ export type ConnectContextOptions = {
     afterMerge?: AfterMergeCallback;
 }
 
-export type ConnectContextFactory = (Component: React.ComponentType<any> | React.FC<any>, options: ConnectContextOptions) => React.FunctionComponent;
+export type ConnectContextFactory = (Component: CustomComponent, options: ConnectContextOptions) => React.FunctionComponent;
 
 export type ProviderCollection = CustomProvider[] | {[key: string]: CustomProvider}
 export type ContextCollection = React.Context<any>[] | {[key: string]: React.Context<any>}