fix tests

apollographql · Jul 31, 2017 · 90b9df6 · 90b9df6
1 parent 3f4c248
commit 90b9df6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 13 deletions.
diff --git a/src/graphql.tsx b/src/graphql.tsx
@@ -593,7 +593,7 @@ export default function graphql<
         const mergedPropsAndData = assign({}, props, clientProps);
 
         if (operationOptions.withRef)
-          mergedPropsAndData.ref = 'wrappedInstance';
+          mergedPropsAndData.ref = this.setWrappedInstance;
         this.renderedElement = createElement(
           WrappedComponent,
           mergedPropsAndData,

diff --git a/test/react-web/client/graphql/shared-operations.test.tsx b/test/react-web/client/graphql/shared-operations.test.tsx
@@ -64,9 +64,7 @@ describe('shared operations', () => {
       expect((decorated as any).getWrappedInstance().someMethod()).toEqual(
         testData,
       );
-      expect((decorated as any).refs.wrappedInstance.someMethod()).toEqual(
-        testData,
-      );
+      expect((decorated as any).wrappedInstance.someMethod()).toEqual(testData);
 
       const DecoratedWithSkip = withApollo(Container, {
         withRef: true,
@@ -88,9 +86,9 @@ describe('shared operations', () => {
       expect(
         (decoratedWithSkip as any).getWrappedInstance().someMethod(),
       ).toEqual(testData);
-      expect(
-        (decoratedWithSkip as any).refs.wrappedInstance.someMethod(),
-      ).toEqual(testData);
+      expect((decoratedWithSkip as any).wrappedInstance.someMethod()).toEqual(
+        testData,
+      );
     });
   });
 
@@ -301,9 +299,7 @@ describe('shared operations', () => {
     expect((decorated as any).getWrappedInstance().someMethod()).toEqual(
       testData,
     );
-    expect((decorated as any).refs.wrappedInstance.someMethod()).toEqual(
-      testData,
-    );
+    expect((decorated as any).wrappedInstance.someMethod()).toEqual(testData);
 
     const DecoratedWithSkip = graphql(query, { withRef: true, skip: true })(
       Container,
@@ -324,9 +320,9 @@ describe('shared operations', () => {
     expect(
       (decoratedWithSkip as any).getWrappedInstance().someMethod(),
     ).toEqual(testData);
-    expect(
-      (decoratedWithSkip as any).refs.wrappedInstance.someMethod(),
-    ).toEqual(testData);
+    expect((decoratedWithSkip as any).wrappedInstance.someMethod()).toEqual(
+      testData,
+    );
   });
 
   it('allows options to take an object', done => {