project-inter-server/test/resolvers/StopResolverTests.test.ts

import { beforeEach, describe, expect, it } from "@jest/globals";
import { ApolloServer } from "@apollo/server";
import { ServerContext } from "../../src/ServerContext";
import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository";
import { setUpTestServer } from "../testHelpers/apolloSetupHelpers";
import { generateMockEtas, generateMockOrderedStops, generateMockStops, generateMockSystems } from "../generators";
import { IStop, ISystem } from "../../src/entities/entities";
import assert = require("node:assert");

describe("StopResolvers", () => {
  let testServer: ApolloServer<ServerContext>;
  let repository: UnoptimizedInMemoryRepository;

  beforeEach(() => {
    testServer = setUpTestServer();
    repository = new UnoptimizedInMemoryRepository();
  });

  let mockStop: IStop;
  let mockSystem: ISystem;

  beforeEach(async () => {
    const mockSystems = generateMockSystems();
    mockSystem = mockSystems[0];
    mockSystem.id = "1";
    await repository.addOrUpdateSystem(mockSystem);

    const mockStops = generateMockStops();
    mockStop = mockStops[0];
    mockStop.id = "1";
    await repository.addOrUpdateStop(mockStop);
  })

  async function getResponseForQuery(query: string) {
    return await testServer.executeOperation({
      query,
      variables: {
        systemId: mockSystem.id,
        stopId: mockStop.id,
      },
    }, {
      contextValue: {
        repository,
      }
    });
  }

  describe("orderedStops", () => {
    const query = `
    query GetOrderedStops($systemId: ID!, $stopId: ID!) {
      system(id: $systemId) {
        stop(id: $stopId) {
          orderedStops {
            routeId
            stopId
          }
        }
      }
    }
    `


    it("returns ordered stops if they exist for the stop ID", async () => {
      let mockOrderedStops = generateMockOrderedStops();
      mockOrderedStops = mockOrderedStops.filter((orderedStop) => orderedStop.stopId === mockOrderedStops[0].stopId);
      await Promise.all(mockOrderedStops.map(async orderedStop => {
        orderedStop.stopId = mockStop.id;
        await repository.addOrUpdateOrderedStop(orderedStop);
      }));

      const response = await getResponseForQuery(query);

      assert(response.body.kind === "single");
      expect(response.body.singleResult.errors).toBeUndefined();
      expect((response.body.singleResult.data as any).system.stop.orderedStops).toHaveLength(mockOrderedStops.length);
    });

    it("returns empty array if no ordered stops exist", async () => {
      const response = await getResponseForQuery(query);

      assert(response.body.kind === "single");
      expect(response.body.singleResult.errors).toBeUndefined();
      expect((response.body.singleResult.data as any).system.stop.orderedStops).toHaveLength(0);
    });
  });

  describe("etas", () => {
    const query = `
    query GetEtas($systemId: ID!, $stopId: ID!) {
      system(id: $systemId) {
        stop(id: $stopId) {
          etas {
            secondsRemaining
          }
        }
      }
    }
    `

    it("returns ETAs if they exist for the stop ID", async () => {
      let mockEtas = generateMockEtas();
      mockEtas = mockEtas.filter((eta) => eta.stopId === mockEtas[0].stopId);
      await Promise.all(mockEtas.map(async eta => {
        eta.stopId = mockStop.id;
        await repository.addOrUpdateEta(eta);
      }));

      const response = await getResponseForQuery(query);

      assert(response.body.kind === "single");
      expect(response.body.singleResult.errors).toBeUndefined();
      expect((response.body.singleResult.data as any).system.stop.etas).toHaveLength(mockEtas.length);
    });

    it("returns empty array if no ETAs exist", async () => {
      const response = await getResponseForQuery(query);

      assert(response.body.kind === "single");
      expect(response.body.singleResult.errors).toBeUndefined();
      expect((response.body.singleResult.data as any).system.stop.etas).toHaveLength(0);
    });
  });
});