add validation for refine method

vitalics · Jan 21, 2024 · 29e2efd · 29e2efd
1 parent 219c7b7
commit 29e2efd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/builder.ts b/src/builder.ts
@@ -196,6 +196,9 @@ abstract class SchemaBuilder<
    * Schema.parse([{ active: true, name: 'some 1' }, { active: true, name: 'some 2' }]) // throws Error
    */
   refine(fn: (output: Output) => any) {
+    if (typeof fn !== 'function') {
+      throw new TypeError(`Cannot set for not a function for refine. Expect "function", Got: ${typeof fn}`, { cause: { fn, type: typeof fn } })
+    }
     this.refineFns.push(fn)
     return this
   }

diff --git a/tests/base.test.ts b/tests/base.test.ts
@@ -213,3 +213,8 @@ test('refine should throws default error', () => {
   expect(result.error).toBeInstanceOf(Error)
   expect(result.error?.message).toBe('refine error')
 })
+
+test('refine should throws TypeError for not a function', () => {
+  // @ts-expect-error
+  expect(() => s.number().refine(false)).toThrow(TypeError)
+})