1 files changed, 43 insertions, 0 deletions
diff --git a/tests/msw/handlers/topics/topics-count.handler.ts b/tests/msw/handlers/topics/topics-count.handler.ts
new file mode 100644
index 0000000..7e3dab9
--- /dev/null
+++ b/tests/msw/handlers/topics/topics-count.handler.ts
@@ -0,0 +1,43 @@
+import { type ExecutionResult, graphql as executeGraphql } from 'graphql';
+import { HttpResponse, graphql } from 'msw';
+import type { TopicsCountResponse } from '../../../../src/services/graphql';
+import type { GraphQLPostWhere } from '../../../../src/types';
+import { wpTopicsFixture } from '../../../fixtures';
+import { getConnection } from '../../../utils/graphql';
+import { schema } from '../../schema';
+
+export const topicsCountHandler = graphql.query<
+  TopicsCountResponse,
+  GraphQLPostWhere
+>('TopicsCount', async ({ query, variables }) => {
+  const pageParams = new URLSearchParams(window.location.search);
+  const isError = pageParams.get('error') === 'true';
+
+  if (isError) return HttpResponse.json({ data: { topics: null } });
+
+  const { data, errors } = (await executeGraphql({
+    schema,
+    source: query,
+    variableValues: variables,
+    rootValue: {
+      topics({ search, title }: typeof variables) {
+        const filteredTopicsByTitle = title
+          ? wpTopicsFixture.filter((topic) => topic.title.includes(title))
+          : wpTopicsFixture;
+        const filteredTopics = search
+          ? filteredTopicsByTitle.filter((topic) =>
+              topic.title.includes(search)
+            )
+          : filteredTopicsByTitle;
+
+        return getConnection({
+          after: null,
+          data: filteredTopics,
+          first: undefined,
+        });
+      },
+    },
+  })) as ExecutionResult<TopicsCountResponse>;
+
+  return HttpResponse.json({ data, errors });
+});