Update withData.js - support withRouter HOC (vercel#3286)

Divyendu Singh · timneutkens · commit ccb188ab6b7b · 2017-11-15T10:44:06.000+01:00
Add support for using withRouter as HOC with this example. Passing router context manually fixes this, based on, vercel#2908 (comment)
diff --git a/examples/with-apollo-auth/lib/withData.js b/examples/with-apollo-auth/lib/withData.js
@@ -54,7 +54,13 @@ export default ComposedComponent => {
             <ComposedComponent url={url} {...composedInitialProps} />
           </ApolloProvider>
         )
-        await getDataFromTree(app)
+        await getDataFromTree(app, {
+          router: {
+            query: context.query,
+            pathname: context.pathname,
+            asPath: context.asPath
+          }
+        })
 
         // Extract query data from the Apollo's store
         const state = apollo.getInitialState()
