@tmdb-graphql-api/resolvers 0.1.0 → 0.1.1-unstable.0

package/src/queries/episodeCredits.test.ts

@@ -7,79 +7,59 @@ import { mockFetch } from 'fetch-mocked';
 import { type FetchResponse, type Getta, type ShortcutProperties } from 'getta';
 import { type GraphQLResolveInfo } from 'graphql';
 import { buildEndpoint } from '../__testUtils__/helpers/buildEndpoint.ts';
+import { createMockLogger } from '../__testUtils__/helpers/createMockLogger.ts';
 import { createRestClient } from '../__testUtils__/helpers/createRestClient.ts';
-import { type Context, type RawEpisode } from '../types.ts';
+import { CAST, CREDITS, CREW, GUEST_STARS } from '../constants.ts';
+import { type Context } from '../types.ts';
 import { resolveEpisodeCredits } from './episodeCredits.ts';
 
 const mockedFetch = mockFetch(jest.fn);
 
 describe('resolveEpisodeCredits >', () => {
-  let restClient: Getta & ShortcutProperties<ShortcutMethodNames>;
-  const setCacheMetadata = jest.fn();
   const tvShowId = 1399;
   const seasonNumber = 1;
   const episodeNumber = 1;
 
-  beforeAll(() => {
-    restClient = createRestClient();
-  });
+  const context = {
+    logger: createMockLogger(jest.fn),
+    restClient: createRestClient(),
+    setCacheMetadata: jest.fn(),
+    tmdbGuestSessionId: 'mock-tmdbGuestSessionId',
+  } as unknown as Context;
 
   beforeEach(() => {
-    mockedFetch.mockGetOnce(
-      buildEndpoint(EPISODE_PATH, { episodeNumber, id: tvShowId, seasonNumber, type: 'credits' }),
-      {
-        body: episodeCredits,
-      }
-    );
+    mockedFetch.mockGetOnce(buildEndpoint(EPISODE_PATH, { episodeNumber, id: tvShowId, seasonNumber, type: CREDITS }), {
+      body: episodeCredits,
+    });
   });
 
   afterEach(async () => {
     mockedFetch.mockReset();
-    await restClient.cache?.clear();
+    await context.restClient.cache?.clear();
+    jest.clearAllMocks();
   });
 
   describe('when fieldName is cast', () => {
     it('should return the correct response', async () => {
-      await expect(
-        resolveEpisodeCredits(
-          episodeDetails as RawEpisode,
-          undefined,
-          { restClient, setCacheMetadata, tmdbGuestSessionId: 'mock-tmdbGuestSessionId' } as unknown as Context,
-          {
-            fieldName: 'cast',
-          } as GraphQLResolveInfo
-        )
-      ).resolves.toMatchSnapshot();
+      const info = { fieldName: CAST } as GraphQLResolveInfo;
+      // @ts-expect-error arg is slightly modified version of episode details
+      await expect(resolveEpisodeCredits(episodeDetails, undefined, context, info)).resolves.toMatchSnapshot();
     });
   });
 
   describe('when fieldName is crew', () => {
     it('should return the correct response', async () => {
-      await expect(
-        resolveEpisodeCredits(
-          episodeDetails as RawEpisode,
-          undefined,
-          { restClient, setCacheMetadata, tmdbGuestSessionId: 'mock-tmdbGuestSessionId' } as unknown as Context,
-          {
-            fieldName: 'crew',
-          } as GraphQLResolveInfo
-        )
-      ).resolves.toMatchSnapshot();
+      const info = { fieldName: CREW } as GraphQLResolveInfo;
+      // @ts-expect-error arg is slightly modified version of episode details
+      await expect(resolveEpisodeCredits(episodeDetails, undefined, context, info)).resolves.toMatchSnapshot();
     });
   });
 
   describe('when fieldName is guestStars', () => {
     it('should return the correct response', async () => {
-      await expect(
-        resolveEpisodeCredits(
-          episodeDetails as RawEpisode,
-          undefined,
-          { restClient, setCacheMetadata, tmdbGuestSessionId: 'mock-tmdbGuestSessionId' } as unknown as Context,
-          {
-            fieldName: 'guestStars',
-          } as GraphQLResolveInfo
-        )
-      ).resolves.toMatchSnapshot();
+      const info = { fieldName: GUEST_STARS } as GraphQLResolveInfo;
+      // @ts-expect-error arg is slightly modified version of episode details
+      await expect(resolveEpisodeCredits(episodeDetails, undefined, context, info)).resolves.toMatchSnapshot();
     });
   });
 
@@ -90,19 +70,18 @@ describe('resolveEpisodeCredits >', () => {
         .mockResolvedValue({ data: {} } as FetchResponse);
 
       const { id, ...rest } = episodeCredits;
+      const info = { fieldName: GUEST_STARS } as GraphQLResolveInfo;
 
       await expect(
         resolveEpisodeCredits(
+          // @ts-expect-error arg is slightly modified version of credits
           { ...episodeDetails, __headers: new Headers(), credits: rest },
           undefined,
           {
+            ...context,
             restClient: { episode: episodeFetchMock } as unknown as Getta & ShortcutProperties<ShortcutMethodNames>,
-            setCacheMetadata,
-            tmdbGuestSessionId: 'mock-tmdbGuestSessionId',
-          } as unknown as Context,
-          {
-            fieldName: 'guestStars',
-          } as GraphQLResolveInfo
+          },
+          info
         )
       ).resolves.toMatchSnapshot();
     });
@@ -113,18 +92,17 @@ describe('resolveEpisodeCredits >', () => {
         .mockResolvedValue({ data: {} } as FetchResponse);
 
       const { id, ...rest } = episodeCredits;
+      const info = { fieldName: GUEST_STARS } as GraphQLResolveInfo;
 
       await resolveEpisodeCredits(
+        // @ts-expect-error arg is slightly modified version of credits
         { ...episodeDetails, __headers: new Headers(), credits: rest },
         undefined,
         {
+          ...context,
           restClient: { episode: episodeFetchMock } as unknown as Getta & ShortcutProperties<ShortcutMethodNames>,
-          setCacheMetadata,
-          tmdbGuestSessionId: 'mock-tmdbGuestSessionId',
-        } as unknown as Context,
-        {
-          fieldName: 'guestStars',
-        } as GraphQLResolveInfo
+        },
+        info
       );
 
       expect(episodeFetchMock).not.toHaveBeenCalled();

package/src/queries/episodeCredits.ts

@@ -1,19 +1,30 @@
-import { GraphQLError, type GraphQLResolveInfo } from 'graphql';
 import { snakeCase } from 'lodash-es';
-import { CREDITS, EPISODE, SEASON, TV } from '../constants.ts';
-import { type Context, type RawEpisodeCredits, type RawEpisodeWithAppendToResponse } from '../types.ts';
+import { CREDITS, EPISODE } from '../constants.ts';
+import { buildObjResolver } from '../helpers/buildObjResolver.ts';
+import {
+  type EnrichedRawEpisode,
+  type EnrichedRawEpisodeWithAppendToResponse,
+  type RawEpisodeCredits,
+} from '../types.ts';
 
-export const resolveEpisodeCredits = async (
-  obj: RawEpisodeWithAppendToResponse & { __headers?: Headers },
-  _args: undefined,
-  { requestID, restClient, setCacheMetadata, tmdbGuestSessionId }: Context,
-  { fieldName, fieldNodes }: GraphQLResolveInfo
-) => {
+const isEnrichedRawEpisodeWithCredits = (
+  obj: EnrichedRawEpisode | (EnrichedRawEpisodeWithAppendToResponse & { __headers?: Headers })
+): obj is EnrichedRawEpisodeWithAppendToResponse & { __headers: Headers } => CREDITS in obj && '__headers' in obj;
+
+export const resolveEpisodeCredits = buildObjResolver<
+  EnrichedRawEpisode | (EnrichedRawEpisodeWithAppendToResponse & { __headers?: Headers }),
+  RawEpisodeCredits['cast'] | RawEpisodeCredits['crew'] | RawEpisodeCredits['guest_stars']
+>(EPISODE, async (obj, ctx, info) => {
+  const { requestID, restClient, tmdbGuestSessionId } = ctx;
+  const { fieldName } = info;
   const typedFieldName = snakeCase(fieldName) as 'cast' | 'crew' | 'guest_stars';
 
-  if (obj.credits && obj.__headers) {
-    setCacheMetadata(`${EPISODE}.${fieldName}`, obj.__headers);
-    return obj.credits[typedFieldName];
+  if (isEnrichedRawEpisodeWithCredits(obj)) {
+    return {
+      data: obj.credits?.[typedFieldName],
+      headers: obj.__headers,
+      source: 'parentNode',
+    };
   }
 
   const { data, errors, headers } = await restClient.episode<RawEpisodeCredits>(
@@ -28,13 +39,9 @@ export const resolveEpisodeCredits = async (
     { requestID, tmdbGuestSessionId }
   );
 
-  if (errors?.length) {
-    throw new GraphQLError(
-      `Failed to resolve ${TV} show ${obj.tv_show_id} ${SEASON} ${obj.season_number} ${EPISODE} ${obj.episode_number} ${fieldName}.`,
-      { nodes: fieldNodes, originalError: errors[0] }
-    );
-  }
-
-  setCacheMetadata(`${EPISODE}.${fieldName}`, headers);
-  return data![typedFieldName];
-};
+  return {
+    data: data?.[typedFieldName],
+    errors,
+    headers,
+  };
+});

package/src/queries/episodeExternalIds.test.ts

@@ -6,27 +6,33 @@ import seasonExternalIds from '@tmdb-graphql-api/test-utils/responses/themoviedb
 import { mockFetch } from 'fetch-mocked';
 import { type FetchResponse, type Getta, type ShortcutProperties } from 'getta';
 import { type GraphQLResolveInfo } from 'graphql';
+import { snakeCase } from 'lodash-es';
 import { buildEndpoint } from '../__testUtils__/helpers/buildEndpoint.ts';
+import { createMockLogger } from '../__testUtils__/helpers/createMockLogger.ts';
 import { createRestClient } from '../__testUtils__/helpers/createRestClient.ts';
-import { type Context, type RawEpisode, type RawEpisodeWithAppendToResponse } from '../types.ts';
+import { EXTERNAL_IDS } from '../constants.ts';
+import { type Context, type EnrichedRawEpisodeWithAppendToResponse } from '../types.ts';
 import { resolveEpisodeExternalIds } from './episodeExternalIds.ts';
 
 const mockedFetch = mockFetch(jest.fn);
 
 describe('resolveEpisodeExternalIds >', () => {
-  let restClient: Getta & ShortcutProperties<ShortcutMethodNames>;
-  const setCacheMetadata = jest.fn();
   const tvShowId = 1399;
   const seasonNumber = 1;
   const episodeNumber = 1;
 
-  beforeAll(() => {
-    restClient = createRestClient();
-  });
+  const context = {
+    logger: createMockLogger(jest.fn),
+    restClient: createRestClient(),
+    setCacheMetadata: jest.fn(),
+    tmdbGuestSessionId: 'mock-tmdbGuestSessionId',
+  } as unknown as Context;
+
+  const info = { fieldName: EXTERNAL_IDS } as GraphQLResolveInfo;
 
   beforeEach(() => {
     mockedFetch.mockGetOnce(
-      buildEndpoint(EPISODE_PATH, { episodeNumber, id: tvShowId, seasonNumber, type: 'external_ids' }),
+      buildEndpoint(EPISODE_PATH, { episodeNumber, id: tvShowId, seasonNumber, type: snakeCase(EXTERNAL_IDS) }),
       {
         body: seasonExternalIds,
       }
@@ -35,17 +41,14 @@ describe('resolveEpisodeExternalIds >', () => {
 
   afterEach(async () => {
     mockedFetch.mockReset();
-    await restClient.cache?.clear();
+    await context.restClient.cache?.clear();
+    jest.clearAllMocks();
   });
 
   it('should return the episode external IDs', async () => {
     await expect(
-      resolveEpisodeExternalIds(
-        episodeDetails as RawEpisode,
-        undefined,
-        { restClient, setCacheMetadata, tmdbGuestSessionId: 'mock-tmdbGuestSessionId' } as unknown as Context,
-        {} as GraphQLResolveInfo
-      )
+      // @ts-expect-error arg is slightly modified version of episode details
+      resolveEpisodeExternalIds(episodeDetails, undefined, context, info)
     ).resolves.toMatchSnapshot();
   });
 
@@ -59,18 +62,14 @@ describe('resolveEpisodeExternalIds >', () => {
 
       await expect(
         resolveEpisodeExternalIds(
-          {
-            ...episodeDetails,
-            __headers: new Headers(),
-            external_ids: rest,
-          } as unknown as RawEpisodeWithAppendToResponse,
+          // @ts-expect-error arg is slightly modified version of episode details
+          { ...episodeDetails, __headers: new Headers(), external_ids: rest },
           undefined,
           {
+            ...context,
             restClient: { episode: episodeFetchMock } as unknown as Getta & ShortcutProperties<ShortcutMethodNames>,
-            setCacheMetadata,
-            tmdbGuestSessionId: 'mock-tmdbGuestSessionId',
-          } as unknown as Context,
-          {} as GraphQLResolveInfo
+          },
+          info
         )
       ).resolves.toMatchSnapshot();
     });
@@ -87,14 +86,13 @@ describe('resolveEpisodeExternalIds >', () => {
           ...episodeDetails,
           __headers: new Headers(),
           external_ids: rest,
-        } as unknown as RawEpisodeWithAppendToResponse,
+        } as unknown as EnrichedRawEpisodeWithAppendToResponse,
         undefined,
         {
+          ...context,
           restClient: { episode: episodeFetchMock } as unknown as Getta & ShortcutProperties<ShortcutMethodNames>,
-          setCacheMetadata,
-          tmdbGuestSessionId: 'mock-tmdbGuestSessionId',
-        } as unknown as Context,
-        {} as GraphQLResolveInfo
+        },
+        info
       );
 
       expect(episodeFetchMock).not.toHaveBeenCalled();

package/src/queries/episodeExternalIds.ts

@@ -1,17 +1,32 @@
-import { GraphQLError, type GraphQLResolveInfo } from 'graphql';
 import { snakeCase } from 'lodash-es';
-import { EPISODE, EXTERNAL_IDS, SEASON, TV } from '../constants.ts';
-import { type Context, type RawEpisodeExternalIds, type RawEpisodeWithAppendToResponse } from '../types.ts';
+import type { Except } from 'type-fest';
+import { EPISODE, EXTERNAL_IDS } from '../constants.ts';
+import { buildObjResolver } from '../helpers/buildObjResolver.ts';
+import {
+  type EnrichedRawEpisode,
+  type EnrichedRawEpisodeWithAppendToResponse,
+  type RawEpisodeExternalIds,
+} from '../types.ts';
 
-export const resolveEpisodeExternalIds = async (
-  obj: RawEpisodeWithAppendToResponse & { __headers?: Headers },
-  _args: undefined,
-  { requestID, restClient, setCacheMetadata, tmdbGuestSessionId }: Context,
-  { fieldNodes }: GraphQLResolveInfo
-) => {
-  if (obj.external_ids && obj.__headers) {
-    setCacheMetadata(`${EPISODE}.${EXTERNAL_IDS}`, obj.__headers);
-    return obj.external_ids;
+const isEnrichedRawEpisodeWithExternalIds = (
+  obj: EnrichedRawEpisode | (EnrichedRawEpisodeWithAppendToResponse & { __headers?: Headers })
+): obj is EnrichedRawEpisodeWithAppendToResponse & { __headers: Headers } =>
+  snakeCase(EXTERNAL_IDS) in obj && '__headers' in obj;
+
+export const resolveEpisodeExternalIds = buildObjResolver<
+  EnrichedRawEpisode | (EnrichedRawEpisodeWithAppendToResponse & { __headers?: Headers }),
+  Except<RawEpisodeExternalIds, 'id'>
+>(EPISODE, async (obj, ctx, info) => {
+  const { requestID, restClient, tmdbGuestSessionId } = ctx;
+  const { fieldName } = info;
+  const typedFieldName = snakeCase(fieldName) as 'external_ids';
+
+  if (isEnrichedRawEpisodeWithExternalIds(obj)) {
+    return {
+      data: obj[typedFieldName],
+      headers: obj.__headers,
+      source: 'parentNode',
+    };
   }
 
   const { data, errors, headers } = await restClient.episode<RawEpisodeExternalIds>(
@@ -20,20 +35,17 @@ export const resolveEpisodeExternalIds = async (
         episodeNumber: obj.episode_number,
         id: obj.tv_show_id,
         seasonNumber: obj.season_number,
-        type: snakeCase(EXTERNAL_IDS),
+        type: typedFieldName,
       },
     },
     { requestID, tmdbGuestSessionId }
   );
 
-  if (errors?.length) {
-    throw new GraphQLError(
-      `Failed to resolve ${TV} show ${obj.tv_show_id} ${SEASON} ${obj.season_number} ${EPISODE} ${obj.episode_number} ${EXTERNAL_IDS}.`,
-      { nodes: fieldNodes, originalError: errors[0] }
-    );
-  }
+  const { id: movieId, ...rest } = data ?? {};
 
-  setCacheMetadata(`${EPISODE}.${EXTERNAL_IDS}`, headers);
-  const { id: movieId, ...rest } = data!;
-  return rest;
-};
+  return {
+    data: rest,
+    errors,
+    headers,
+  };
+});

package/src/queries/episodeStills.test.ts

@@ -7,46 +7,43 @@ import { mockFetch } from 'fetch-mocked';
 import { type FetchResponse, type Getta, type ShortcutProperties } from 'getta';
 import { type GraphQLResolveInfo } from 'graphql';
 import { buildEndpoint } from '../__testUtils__/helpers/buildEndpoint.ts';
+import { createMockLogger } from '../__testUtils__/helpers/createMockLogger.ts';
 import { createRestClient } from '../__testUtils__/helpers/createRestClient.ts';
-import { type Context, type RawEpisode } from '../types.ts';
+import { IMAGES, STILLS } from '../constants.ts';
+import { type Context } from '../types.ts';
 import { resolveEpisodeStills } from './episodeStills.ts';
 
 const mockedFetch = mockFetch(jest.fn);
 
 describe('resolveEpisodeStills >', () => {
-  let restClient: Getta & ShortcutProperties<ShortcutMethodNames>;
-  const setCacheMetadata = jest.fn();
   const tvShowId = 1399;
   const seasonNumber = 1;
   const episodeNumber = 1;
 
-  beforeAll(() => {
-    restClient = createRestClient();
-  });
+  const context = {
+    logger: createMockLogger(jest.fn),
+    restClient: createRestClient(),
+    setCacheMetadata: jest.fn(),
+    tmdbGuestSessionId: 'mock-tmdbGuestSessionId',
+  } as unknown as Context;
+
+  const info = { fieldName: STILLS } as GraphQLResolveInfo;
 
   beforeEach(() => {
-    mockedFetch.mockGetOnce(
-      buildEndpoint(EPISODE_PATH, { episodeNumber, id: tvShowId, seasonNumber, type: 'images' }),
-      {
-        body: episodeStills,
-      }
-    );
+    mockedFetch.mockGetOnce(buildEndpoint(EPISODE_PATH, { episodeNumber, id: tvShowId, seasonNumber, type: IMAGES }), {
+      body: episodeStills,
+    });
   });
 
   afterEach(async () => {
     mockedFetch.mockReset();
-    await restClient.cache?.clear();
+    await context.restClient.cache?.clear();
+    jest.clearAllMocks();
   });
 
   it('should return the episode stills', async () => {
-    await expect(
-      resolveEpisodeStills(
-        episodeDetails as RawEpisode,
-        undefined,
-        { restClient, setCacheMetadata, tmdbGuestSessionId: 'mock-tmdbGuestSessionId' } as unknown as Context,
-        {} as GraphQLResolveInfo
-      )
-    ).resolves.toMatchSnapshot();
+    // @ts-expect-error arg is slightly modified version of episode details
+    await expect(resolveEpisodeStills(episodeDetails, undefined, context, info)).resolves.toMatchSnapshot();
   });
 
   describe('when raw episode data has images', () => {
@@ -59,14 +56,14 @@ describe('resolveEpisodeStills >', () => {
 
       await expect(
         resolveEpisodeStills(
+          // @ts-expect-error arg is slightly modified version of episode details
           { ...episodeDetails, __headers: new Headers(), images: rest },
           undefined,
           {
+            ...context,
             restClient: { episode: episodeFetchMock } as unknown as Getta & ShortcutProperties<ShortcutMethodNames>,
-            setCacheMetadata,
-            tmdbGuestSessionId: 'mock-tmdbGuestSessionId',
-          } as unknown as Context,
-          {} as GraphQLResolveInfo
+          },
+          info
         )
       ).resolves.toMatchSnapshot();
     });
@@ -79,14 +76,14 @@ describe('resolveEpisodeStills >', () => {
       const { id, ...rest } = episodeStills;
 
       await resolveEpisodeStills(
+        // @ts-expect-error arg is slightly modified version of episode details
         { ...episodeDetails, __headers: new Headers(), images: rest },
         undefined,
         {
+          ...context,
           restClient: { episode: episodeFetchMock } as unknown as Getta & ShortcutProperties<ShortcutMethodNames>,
-          setCacheMetadata,
-          tmdbGuestSessionId: 'mock-tmdbGuestSessionId',
-        } as unknown as Context,
-        {} as GraphQLResolveInfo
+        },
+        info
       );
 
       expect(episodeFetchMock).not.toHaveBeenCalled();

package/src/queries/episodeStills.ts

@@ -1,16 +1,30 @@
-import { GraphQLError, type GraphQLResolveInfo } from 'graphql';
-import { EPISODE, IMAGES, SEASON, STILLS, TV } from '../constants.ts';
-import { type Context, type RawEpisodeImages, type RawEpisodeWithAppendToResponse } from '../types.ts';
+import { snakeCase } from 'lodash-es';
+import { EPISODE, IMAGES } from '../constants.ts';
+import { buildObjResolver } from '../helpers/buildObjResolver.ts';
+import {
+  type EnrichedRawEpisode,
+  type EnrichedRawEpisodeWithAppendToResponse,
+  type RawEpisodeImages,
+} from '../types.ts';
 
-export const resolveEpisodeStills = async (
-  obj: RawEpisodeWithAppendToResponse & { __headers?: Headers },
-  _args: undefined,
-  { requestID, restClient, setCacheMetadata, tmdbGuestSessionId }: Context,
-  { fieldNodes }: GraphQLResolveInfo
-) => {
-  if (obj.images && obj.__headers) {
-    setCacheMetadata(`${EPISODE}.${STILLS}`, obj.__headers);
-    return obj.images.stills;
+const isEnrichedRawEpisodeWithImages = (
+  obj: EnrichedRawEpisode | (EnrichedRawEpisodeWithAppendToResponse & { __headers?: Headers })
+): obj is EnrichedRawEpisodeWithAppendToResponse & { __headers: Headers } => IMAGES in obj && '__headers' in obj;
+
+export const resolveEpisodeStills = buildObjResolver<
+  EnrichedRawEpisode | (EnrichedRawEpisodeWithAppendToResponse & { __headers?: Headers }),
+  RawEpisodeImages['stills']
+>(EPISODE, async (obj, ctx, info) => {
+  const { requestID, restClient, tmdbGuestSessionId } = ctx;
+  const { fieldName } = info;
+  const typedFieldName = snakeCase(fieldName) as 'stills';
+
+  if (isEnrichedRawEpisodeWithImages(obj)) {
+    return {
+      data: obj.images?.[typedFieldName],
+      headers: obj.__headers,
+      source: 'parentNode',
+    };
   }
 
   const { data, errors, headers } = await restClient.episode<RawEpisodeImages>(
@@ -25,13 +39,9 @@ export const resolveEpisodeStills = async (
     { requestID, tmdbGuestSessionId }
   );
 
-  if (errors?.length) {
-    throw new GraphQLError(
-      `Failed to resolve ${TV} show ${obj.tv_show_id} ${SEASON} ${obj.season_number} ${EPISODE} ${obj.episode_number} ${STILLS}.`,
-      { nodes: fieldNodes, originalError: errors[0] }
-    );
-  }
-
-  setCacheMetadata(`${EPISODE}.${STILLS}`, headers);
-  return data!.stills;
-};
+  return {
+    data: data?.[typedFieldName],
+    errors,
+    headers,
+  };
+});

package/src/queries/episodeTranslations.test.ts

@@ -7,26 +7,31 @@ import { mockFetch } from 'fetch-mocked';
 import { type FetchResponse, type Getta, type ShortcutProperties } from 'getta';
 import { type GraphQLResolveInfo } from 'graphql';
 import { buildEndpoint } from '../__testUtils__/helpers/buildEndpoint.ts';
+import { createMockLogger } from '../__testUtils__/helpers/createMockLogger.ts';
 import { createRestClient } from '../__testUtils__/helpers/createRestClient.ts';
-import { type Context, type RawEpisode } from '../types.ts';
+import { TRANSLATIONS } from '../constants.ts';
+import { type Context } from '../types.ts';
 import { resolveEpisodeTranslations } from './episodeTranslations.ts';
 
 const mockedFetch = mockFetch(jest.fn);
 
 describe('resolveEpisodeTranslations >', () => {
-  let restClient: Getta & ShortcutProperties<ShortcutMethodNames>;
-  const setCacheMetadata = jest.fn();
   const tvShowId = 1399;
   const seasonNumber = 1;
   const episodeNumber = 1;
 
-  beforeAll(() => {
-    restClient = createRestClient();
-  });
+  const context = {
+    logger: createMockLogger(jest.fn),
+    restClient: createRestClient(),
+    setCacheMetadata: jest.fn(),
+    tmdbGuestSessionId: 'mock-tmdbGuestSessionId',
+  } as unknown as Context;
+
+  const info = { fieldName: TRANSLATIONS } as GraphQLResolveInfo;
 
   beforeEach(() => {
     mockedFetch.mockGetOnce(
-      buildEndpoint(EPISODE_PATH, { episodeNumber, id: tvShowId, seasonNumber, type: 'translations' }),
+      buildEndpoint(EPISODE_PATH, { episodeNumber, id: tvShowId, seasonNumber, type: TRANSLATIONS }),
       {
         body: episodeTranslations,
       }
@@ -35,17 +40,14 @@ describe('resolveEpisodeTranslations >', () => {
 
   afterEach(async () => {
     mockedFetch.mockReset();
-    await restClient.cache?.clear();
+    await context.restClient.cache?.clear();
+    jest.clearAllMocks();
   });
 
   it('should return the episode translations', async () => {
     await expect(
-      resolveEpisodeTranslations(
-        episodeDetails as RawEpisode,
-        undefined,
-        { restClient, setCacheMetadata, tmdbGuestSessionId: 'mock-tmdbGuestSessionId' } as unknown as Context,
-        {} as GraphQLResolveInfo
-      )
+      // @ts-expect-error arg is slightly modified version of episode details
+      resolveEpisodeTranslations(episodeDetails, undefined, context, info)
     ).resolves.toMatchSnapshot();
   });
 
@@ -59,14 +61,14 @@ describe('resolveEpisodeTranslations >', () => {
 
       await expect(
         resolveEpisodeTranslations(
+          // @ts-expect-error arg is slightly modified version of episode details
           { ...episodeDetails, __headers: new Headers(), translations: rest },
           undefined,
           {
+            ...context,
             restClient: { episode: episodeFetchMock } as unknown as Getta & ShortcutProperties<ShortcutMethodNames>,
-            setCacheMetadata,
-            tmdbGuestSessionId: 'mock-tmdbGuestSessionId',
-          } as unknown as Context,
-          {} as GraphQLResolveInfo
+          },
+          info
         )
       ).resolves.toMatchSnapshot();
     });
@@ -79,14 +81,14 @@ describe('resolveEpisodeTranslations >', () => {
       const { id, ...rest } = episodeTranslations;
 
       await resolveEpisodeTranslations(
+        // @ts-expect-error arg is slightly modified version of episode details
         { ...episodeDetails, __headers: new Headers(), translations: rest },
         undefined,
         {
+          ...context,
           restClient: { episode: episodeFetchMock } as unknown as Getta & ShortcutProperties<ShortcutMethodNames>,
-          setCacheMetadata,
-          tmdbGuestSessionId: 'mock-tmdbGuestSessionId',
-        } as unknown as Context,
-        {} as GraphQLResolveInfo
+        },
+        info
       );
 
       expect(episodeFetchMock).not.toHaveBeenCalled();