feat(react/firestore): add useEnableNetworkMutation (#147)

* feat(react): add useEnableNetworkMutation

* _

Author: HassanBahati

packages/react/src/firestore/index.ts

@@ -1,6 +1,6 @@
 export { useClearIndexedDbPersistenceMutation } from "./useClearIndexedDbPersistenceMutation";
 export { useDisableNetworkMutation } from "./useDisableNetworkMutation";
-// useEnableNetworkMutation
+export { useEnableNetworkMutation } from "./useEnableNetworkMutation";
 export { useWaitForPendingWritesQuery } from "./useWaitForPendingWritesQuery";
 export { useRunTransactionMutation } from "./useRunTransactionMutation";
 export { useWriteBatchCommitMutation } from "./useWriteBatchCommitMutation";

packages/react/src/firestore/useEnableNetworkMutation.test.tsx

@@ -0,0 +1,66 @@
+import { act, renderHook, waitFor } from "@testing-library/react";
+import {
+  doc,
+  disableNetwork,
+  getDocFromServer,
+  setDoc,
+} from "firebase/firestore";
+import { beforeEach, describe, expect, test, vi } from "vitest";
+import {
+  expectFirestoreError,
+  firestore,
+  wipeFirestore,
+} from "~/testing-utils";
+import { useEnableNetworkMutation } from "./useEnableNetworkMutation";
+import { queryClient } from "../../utils";
+import { wrapper } from "../../utils";
+
+describe("useEnableNetworkMutation", () => {
+  beforeEach(async () => {
+    queryClient.clear();
+    await disableNetwork(firestore);
+    await wipeFirestore();
+  });
+
+  test("should successfully enable the Firestore network", async () => {
+    const docRef = doc(firestore, "tests", "useEnableNetworkMutation");
+    const mockData = { library: "tanstack-query-firebase" };
+
+    const { result } = renderHook(() => useEnableNetworkMutation(firestore), {
+      wrapper,
+    });
+
+    // Enable the network
+    await act(() => result.current.mutate());
+    await waitFor(() => expect(result.current.isSuccess).toBe(true));
+
+    await setDoc(docRef, mockData);
+
+    const fetchedDoc = await getDocFromServer(docRef);
+
+    expect(fetchedDoc.exists()).toBe(true);
+    expect(fetchedDoc.data()).toEqual(mockData);
+  });
+
+  test("handles mutation options correctly", async () => {
+    const onSuccessMock = vi.fn();
+    const onErrorMock = vi.fn();
+
+    const { result } = renderHook(
+      () =>
+        useEnableNetworkMutation(firestore, {
+          onSuccess: onSuccessMock,
+          onError: onErrorMock,
+        }),
+      { wrapper }
+    );
+
+    await act(() => result.current.mutate());
+
+    await waitFor(() => {
+      expect(result.current.isSuccess).toBe(true);
+      expect(onSuccessMock).toHaveBeenCalled();
+      expect(onErrorMock).not.toHaveBeenCalled();
+    });
+  });
+});

packages/react/src/firestore/useEnableNetworkMutation.ts

@@ -0,0 +1,21 @@
+import { type UseMutationOptions, useMutation } from "@tanstack/react-query";
+import {
+  type Firestore,
+  type FirestoreError,
+  enableNetwork,
+} from "firebase/firestore";
+
+type FirestoreUseMutationOptions<TData = unknown, TError = Error> = Omit<
+  UseMutationOptions<TData, TError, void>,
+  "mutationFn"
+>;
+
+export function useEnableNetworkMutation(
+  firestore: Firestore,
+  options?: FirestoreUseMutationOptions<void, FirestoreError>
+) {
+  return useMutation<void, FirestoreError>({
+    ...options,
+    mutationFn: () => enableNetwork(firestore),
+  });
+}