From 2aba6e3756e85b64ea2712e2574959cd016d449b Mon Sep 17 00:00:00 2001
From: Kok J Sam <samkj.ks@gmail.com>
Date: Mon, 8 May 2017 21:28:48 -0400
Subject: [PATCH 1/2] Allow applyContainerQuery to take Stateless component as
 component

---
 src/index.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/index.tsx b/src/index.tsx
index c1ec421..99534b5 100644
--- a/src/index.tsx
+++ b/src/index.tsx
@@ -52,7 +52,7 @@ export class ContainerQuery extends React.Component<Props, State> {
  */
 
 export function applyContainerQuery<P extends {containerQuery: Params}>(
-  Component: React.ComponentClass<P>,
+  Component: React.ComponentClass<P> | React.StatelessComponent<P>,
   query: Query,
   initialSize?: Size
 ): React.ComponentClass<P> {

From 06c5a0e3f66169997992810371a80b93000e00e7 Mon Sep 17 00:00:00 2001
From: Daiwei Lu <daiweilu123@gmail.com>
Date: Mon, 8 May 2017 22:34:30 -0700
Subject: [PATCH 2/2] Add test

---
 test/client/index.js | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/test/client/index.js b/test/client/index.js
index b2bd064..43b7bef 100644
--- a/test/client/index.js
+++ b/test/client/index.js
@@ -340,4 +340,20 @@ describe('applyContainerQuery', function () {
     }, 100);
   });
 
+  it('accept stateless component', function (done) {
+    let _params;
+
+    const Container = applyContainerQuery(function (props) {
+      _params = props.containerQuery;
+      return <div style={{width: '200px'}}></div>;
+    }, query);
+
+    render(<Container/>, $div);
+
+    setTimeout(() => {
+      expect(_params).toEqual({mobile: true, desktop: false});
+      done();
+    }, 100);
+  });
+
 });