Update gridComponent to use ComponentType

bvaughn · Oct 28, 2018 · e48ce2c · e48ce2c
1 parent 97eefe0
commit e48ce2c
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/src/createGridComponent.js b/src/createGridComponent.js
@@ -18,7 +18,9 @@ type RenderComponentProps<T> = {|
   rowIndex: number,
   style: Object,
 |};
-export type RenderComponent<T> = (props: RenderComponentProps<T>) => React$Node;
+export type RenderComponent<T> = React$ComponentType<
+  $Shape<RenderComponentProps<T>>
+>;
 
 type ScrollDirection = 'forward' | 'backward';
 

diff --git a/src/test.js.flow b/src/test.js.flow
@@ -261,3 +261,25 @@ const assertCustomType = (value: ItemData) => {};
     {Item}
   </VariableSizeGrid>;
 }
+
+{
+  class Item extends React.PureComponent<{ data: void }> {
+    render() {
+      const { data } = this.props;
+      assertVoid(data);
+      // $FlowFixMe itemData is undefined by default
+      assertEmpty(data);
+      return null;
+    }
+  }
+  <VariableSizeGrid
+    width={0}
+    height={0}
+    rowHeight={0}
+    rowCount={0}
+    columnWidth={0}
+    columnCount={0}
+  >
+    {Item}
+  </VariableSizeGrid>;
+}