feat(query): add in and notIn filters to NumberQuery

src/comparators/number.ts

@@ -26,4 +26,10 @@ export const numberComparators: QueryToComparator<NumberQuery> = {
   lte(expected, actual) {
     return actual <= expected
   },
+  in(expected, actual) {
+    return expected.includes(actual)
+  },
+  notIn(expected, actual) {
+    return !numberComparators.in(expected, actual)
+  },
 }

src/model/generateGraphQLHandlers.ts

@@ -57,7 +57,8 @@ function createComparatorGraphQLInputType(
     name,
     fields: Object.keys(comparators).reduce<GraphQLInputFieldConfigMap>(
       (fields, comparatorFn) => {
-        fields[comparatorFn] = { type }
+        const fieldType = ['between', 'notBetween', 'in', 'notIn'].includes(comparatorFn) ? GraphQLList(type) : type
+        fields[comparatorFn] = { type: fieldType }
         return fields
       },
       {},

src/query/queryTypes.ts

@@ -121,6 +121,8 @@ export interface NumberQuery {
   gte: number
   lt: number
   lte: number
+  in: number[]
+  notIn: number[]
 }
 
 export interface BooleanQuery {

test/model/toGraphQLSchema.test.ts

@@ -35,28 +35,30 @@ test('generates a graphql schema', () => {
       notEquals: ID
       contains: ID
       notContains: ID
-      in: ID
-      notIn: ID
+      in: [ID]
+      notIn: [ID]
     }
 
     input StringQueryType {
       equals: String
       notEquals: String
       contains: String
       notContains: String
-      in: String
-      notIn: String
+      in: [String]
+      notIn: [String]
     }
 
     input IntQueryType {
       equals: Int
       notEquals: Int
-      between: Int
-      notBetween: Int
+      between: [Int]
+      notBetween: [Int]
       gt: Int
       gte: Int
       lt: Int
       lte: Int
+      in: [Int]
+      notIn: [Int]
     }
 
     type Mutation {

test/query/number.test.ts

@@ -162,3 +162,31 @@ test('queries entities that are younger or equal a number', () => {
   const names = users.map((user) => user.firstName)
   expect(names).toEqual(['John', 'Alice'])
 })
+
+test('queries entities where property is not contained into the array', () => {
+  const db = setup()
+
+  const users = db.user.findMany({
+    where: {
+      age: {
+        notIn: [16, 24],
+      },
+    },
+  })
+  const names = users.map((user) => user.firstName)
+  expect(names).toEqual(['Kate'])
+})
+
+test('queries entities where property is contained into the array', () => {
+  const db = setup()
+
+  const users = db.user.findMany({
+    where: {
+      age: {
+        in: [16, 24],
+      },
+    },
+  })
+  const names = users.map((user) => user.firstName)
+  expect(names).toEqual(['John', 'Alice'])
+})