fix: allow any for array_output in older versions of the schema (#580)

* fix: allow any for array_output * Update validateGridFile.ts * Update validateGridFile.ts * Update validateGridFile.test.ts
quadratichq · Jun 27, 2023 · 8a3c4ad · 8a3c4ad · vercel · Jun 27, 2023
1 parent 0e1ee87
commit 8a3c4ad
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 3 deletions.
diff --git a/src/schemas/GridFileV1.ts b/src/schemas/GridFileV1.ts
@@ -1,6 +1,6 @@
 import z from 'zod';
 
-const ArrayOutputSchema = z.array(z.union([z.string(), z.number(), z.boolean()]));
+const ArrayOutputSchema = z.array(z.any());
 
 export enum BorderType {
   line1 = 0,

diff --git a/src/schemas/GridFileV1_1.ts b/src/schemas/GridFileV1_1.ts
@@ -4,7 +4,7 @@ import { v4 as uuid } from 'uuid';
 import { DEFAULT_FILE_NAME } from '../constants/app';
 
 // Shared schemas
-const ArrayOutputSchema = z.array(z.union([z.string(), z.number(), z.boolean()]));
+const ArrayOutputSchema = z.array(z.any());
 const BorderDirectionSchema = z.object({
   color: z.string().optional(),
   type: z.enum(['line1', 'line2', 'line3', 'dotted', 'dashed', 'double']).optional(),

diff --git a/src/schemas/validateGridFile.test.ts b/src/schemas/validateGridFile.test.ts
@@ -35,7 +35,27 @@ const v1File: GridFileV1 = {
 
 const v1_1File: GridFileV1_1 = {
   version: '1.1',
-  cells: [],
+  cells: [
+    // Test to allow `any` for `array_output`
+    {
+      x: 0,
+      y: 0,
+      type: 'PYTHON',
+      value: '1,2,3',
+      last_modified: '2023-06-27T16:54:40.619Z',
+      evaluation_result: {
+        success: true,
+        std_out: '',
+        output_value: '[[[1, 2, 3]]]',
+        cells_accessed: [],
+        array_output: [[[1, 2, 3]]],
+        formatted_code: '[[[1, 2, 3]]]\n',
+        error_span: null,
+      },
+      python_code: '[[[1,2,3]]]',
+      array_cells: [[0, 0]],
+    },
+  ],
   columns: [],
   rows: [],
   borders: [],