add refetching tests with record for refetchVariables

Author: tsnobip

packages/rescript-relay/__tests__/Test_refetchingRecord-tests.js

@@ -0,0 +1,73 @@
+require("@testing-library/jest-dom/extend-expect");
+const t = require("@testing-library/react");
+const React = require("react");
+const queryMock = require("./queryMock");
+const ReactTestUtils = require("react-dom/test-utils");
+
+const { test_refetching } = require("./Test_refetchingRecord.bs");
+
+describe("Fragment", () => {
+  test("refetching works when using a record for refetchVariables", async () => {
+    queryMock.mockQuery({
+      name: "TestRefetchingRecordQuery",
+      variables: { beforeDate: "2023-01-01T00:00:00.000Z", showOnlineStatus: true, number: 10 },
+      data: {
+        loggedInUser: {
+          id: "user-1",
+          firstName: "First",
+          onlineStatus: "offline",
+          friendsConnection: {
+            totalCount: 20,
+          },
+          friends: [
+            {
+              id: "user-2",
+            },
+          ],
+        },
+      },
+    });
+
+    t.render(test_refetching());
+
+    await t.screen.findByText("First is -");
+    await t.screen.findByText("Friends: 20");
+
+    queryMock.mockQuery({
+      name: "TestRefetchingRecordRefetchQuery",
+      variables: {
+        id: "user-1",
+        showOnlineStatus: true,
+        friendsOnlineStatuses: ["Online", "offline"],
+        beforeDate: null,
+        number:10,
+      },
+      data: {
+        node: {
+          __typename: "User",
+          id: "user-1",
+          firstName: "First",
+          onlineStatus: "Online",
+          friendsConnection: {
+            totalCount: 10,
+          },
+          friends: [
+            {
+              id: "user-2",
+            },
+            {
+              id: "user-3",
+            },
+          ],
+        },
+      },
+    });
+
+    ReactTestUtils.act(() => {
+      t.fireEvent.click(t.screen.getByText("Fetch online status"));
+    });
+
+    await t.screen.findByText("First is online");
+    await t.screen.findByText("Friends: 10");
+  });
+});

packages/rescript-relay/__tests__/Test_refetchingRecord.res

@@ -0,0 +1,98 @@
+module Query = %relay(`
+    query TestRefetchingRecordQuery($beforeDate: Datetime, $number: Number, $showOnlineStatus: Boolean!) {
+      loggedInUser {
+        ...TestRefetchingRecord_user @arguments(
+          beforeDate: $beforeDate
+          number: $number
+          showOnlineStatus: $showOnlineStatus
+        )
+      }
+    }
+`)
+
+module Fragment = %relay(`
+    fragment TestRefetchingRecord_user on User
+      @refetchable(queryName: "TestRefetchingRecordRefetchQuery")
+      @argumentDefinitions(
+        friendsOnlineStatuses: { type: "[OnlineStatus!]" }
+        showOnlineStatus: { type: "Boolean", defaultValue: false }
+        beforeDate: { type: "Datetime" }
+        number: { type: "Number" }
+      ) {
+      firstName
+      onlineStatus @include(if: $showOnlineStatus)
+      friendsConnection(statuses: $friendsOnlineStatuses) {
+        totalCount
+      }
+      friends(beforeDate: $beforeDate, number: $number) {
+        id
+      }
+    }
+`)
+
+module FragmentWithNoArgs = %relay(`
+    fragment TestRefetchingRecordNoArgs_query on Query
+      @refetchable(queryName: "TestRefetchingRecordNoArgsRefetchQuery")
+      {
+      loggedInUser {
+        id
+      }
+    }
+`)
+
+module Test = {
+  @react.component
+  let make = () => {
+    let query = Query.use(
+      ~variables={
+        beforeDate: Date.fromString("2023-01-01T00:00:00.000Z"),
+        showOnlineStatus: true,
+        number: [10],
+      },
+    )
+
+    let (data, refetch) = Fragment.useRefetchable(query.loggedInUser.fragmentRefs)
+
+    let (_, startTransition) = React.useTransition()
+
+    <div>
+      {React.string(
+        data.firstName ++
+        (" is " ++
+        switch data.onlineStatus {
+        | Some(Online) => "online"
+        | _ => "-"
+        }),
+      )}
+      <div> {React.string("Friends: " ++ data.friendsConnection.totalCount->Int.toString)} </div>
+      <button
+        onClick={_ => {
+          startTransition(() => {
+            refetch(
+              ~variables={
+                friendsOnlineStatuses: Some([Online, Offline]),
+                beforeDate: None,
+              },
+            )->RescriptRelay.Disposable.ignore
+          })
+        }}
+      >
+        {React.string("Fetch online status")}
+      </button>
+    </div>
+  }
+}
+
+@live
+let test_refetching = () => {
+  let network = RescriptRelay.Network.makePromiseBased(~fetchFunction=RelayEnv.fetchQuery)
+
+  let environment = RescriptRelay.Environment.make(
+    ~network,
+    ~store=RescriptRelay.Store.make(~source=RescriptRelay.RecordSource.make()),
+  )
+
+  <TestProviders.Wrapper environment>
+    <Test />
+  </TestProviders.Wrapper>
+}