Update integration tests for aggregations to use testHelpers (#4866)

Author: angrykoala

packages/graphql/tests/integration/aggregations/field-level/authorization/field-aggregation-authorization-where.int.test.ts

@@ -17,43 +17,40 @@
  * limitations under the License.
  */
 
-import type { Driver, Session } from "neo4j-driver";
-import { graphql } from "graphql";
-import Neo4jHelper from "../../../neo4j";
-import { Neo4jGraphQL } from "../../../../../src/classes";
-import { UniqueType } from "../../../../utils/graphql-types";
 import { createBearerToken } from "../../../../utils/create-bearer-token";
+import type { UniqueType } from "../../../../utils/graphql-types";
+import { TestHelper } from "../../../utils/tests-helper";
 
 describe(`Field Level Authorization Where Requests`, () => {
-    let neoSchema: Neo4jGraphQL;
-    let driver: Driver;
-    let neo4j: Neo4jHelper;
-    let session: Session;
-    const typeMovie = new UniqueType("Movie");
-    const typeActor = new UniqueType("Actor");
-    const typeDefs = `
-    type ${typeMovie.name} {
-        name: String
-        year: Int
-        createdAt: DateTime
-        ${typeActor.plural}: [${typeActor.name}!]! @relationship(type: "ACTED_IN", direction: IN)
-    }
-
-    type ${typeActor.name} {
-        name: String
-        year: Int
-        createdAt: DateTime
-        testStr: String
-        ${typeMovie.plural}: [${typeMovie.name}!]! @relationship(type: "ACTED_IN", direction: OUT)
-    }`;
+    let testHelper: TestHelper;
+
+    let typeMovie: UniqueType;
+    let typeActor: UniqueType;
+    let typeDefs: string;
     const secret = "secret";
 
-    beforeAll(async () => {
-        neo4j = new Neo4jHelper();
-        driver = await neo4j.getDriver();
-        session = await neo4j.getSession();
+    beforeEach(async () => {
+        testHelper = new TestHelper();
+
+        typeMovie = testHelper.createUniqueType("Movie");
+        typeActor = testHelper.createUniqueType("Actor");
+        typeDefs = `
+        type ${typeMovie.name} {
+            name: String
+            year: Int
+            createdAt: DateTime
+            ${typeActor.plural}: [${typeActor.name}!]! @relationship(type: "ACTED_IN", direction: IN)
+        }
+    
+        type ${typeActor.name} {
+            name: String
+            year: Int
+            createdAt: DateTime
+            testStr: String
+            ${typeMovie.plural}: [${typeMovie.name}!]! @relationship(type: "ACTED_IN", direction: OUT)
+        }`;
 
-        await session.run(`
+        await testHelper.runCypher(`
             CREATE (m:${typeMovie.name}
                 {name: "Terminator",year:1990,createdAt: datetime()})
                 <-[:ACTED_IN]-
@@ -63,7 +60,7 @@ describe(`Field Level Authorization Where Requests`, () => {
         const extendedTypeDefs = `${typeDefs}
         extend type ${typeActor.name} @authorization(filter: [{ operations: [AGGREGATE], where: { node: { testStr: "$jwt.sub" } } }])`;
 
-        neoSchema = new Neo4jGraphQL({
+        await testHelper.initNeo4jGraphQL({
             typeDefs: extendedTypeDefs,
             features: {
                 authorization: {
@@ -73,9 +70,8 @@ describe(`Field Level Authorization Where Requests`, () => {
         });
     });
 
-    afterAll(async () => {
-        await session.close();
-        await driver.close();
+    afterEach(async () => {
+        await testHelper.close();
     });
 
     test("authenticated query", async () => {
@@ -88,11 +84,7 @@ describe(`Field Level Authorization Where Requests`, () => {
             }`;
 
         const token = createBearerToken(secret, { sub: "1234" });
-        const gqlResult = await graphql({
-            schema: await neoSchema.getSchema(),
-            source: query,
-            contextValue: neo4j.getContextValues({ token }),
-        });
+        const gqlResult = await testHelper.runGraphQLWithToken(query, token);
         expect(gqlResult.errors).toBeUndefined();
         expect((gqlResult as any).data[typeMovie.plural][0][`${typeActor.plural}Aggregate`]).toEqual({
             count: 1,
@@ -108,11 +100,7 @@ describe(`Field Level Authorization Where Requests`, () => {
                 }
             }`;
 
-        const gqlResult = await graphql({
-            schema: await neoSchema.getSchema(),
-            source: query,
-            contextValue: neo4j.getContextValues(),
-        });
+        const gqlResult = await testHelper.runGraphQL(query);
 
         expect(gqlResult.errors).toBeUndefined();
         expect(gqlResult.data as any).toEqual({
@@ -130,11 +118,7 @@ describe(`Field Level Authorization Where Requests`, () => {
             }`;
 
         const invalidToken = createBearerToken(secret, { sub: "2222" });
-        const gqlResult = await graphql({
-            schema: await neoSchema.getSchema(),
-            source: query,
-            contextValue: neo4j.getContextValues({ token: invalidToken }),
-        });
+        const gqlResult = await testHelper.runGraphQLWithToken(query, invalidToken);
         expect(gqlResult.errors).toBeUndefined();
         expect((gqlResult as any).data[typeMovie.plural][0][`${typeActor.plural}Aggregate`]).toEqual({
             count: 0,