diff --git a/test/resolvers/OrderedStopResolverTests.test.ts b/test/resolvers/OrderedStopResolverTests.test.ts index a1cc727..ac603a1 100644 --- a/test/resolvers/OrderedStopResolverTests.test.ts +++ b/test/resolvers/OrderedStopResolverTests.test.ts @@ -3,6 +3,8 @@ import { ApolloServer } from "@apollo/server"; import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository"; import { ServerContext } from "../../src/ServerContext"; import { setUpTestServer } from "../testHelpers/apolloSetupHelpers"; +import { IRoute, IStop, ISystem } from "../../src/entities/entities"; +import { generateMockRoutes, generateMockShuttles, generateMockStops, generateMockSystems } from "../generators"; describe("OrderedStopResolvers", () => { let apolloServer: ApolloServer; @@ -13,6 +15,27 @@ describe("OrderedStopResolvers", () => { repository = new UnoptimizedInMemoryRepository(); }); + let mockSystem: ISystem; + let mockRoute: IRoute; + let mockStop: IStop; + + beforeEach(async () => { + const mockSystems = generateMockSystems(); + mockSystem = mockSystems[0]; + mockSystem.id = "1"; + await repository.addOrUpdateSystem(mockSystem); + + const mockRoutes = generateMockRoutes(); + mockRoute = mockRoutes[0]; + mockRoute.id = "1"; + await repository.addOrUpdateRoute(mockRoute); + + const mockStops = generateMockStops(); + mockStop = mockStops[0]; + mockStop.id = "1"; + await repository.addOrUpdateStop(mockStop); + }); + describe("nextStop", () => { const query = ` query GetNextStop($systemId: ID!, $routeId: ID!, $stopId: ID!) {