Testing

package.json

@@ -31,9 +31,12 @@
     "cli": "tsx src/codegen/cli.ts",
     "cli:index": "tsx src/codegen/cli.ts --onlyindex true",
     "lint": "eslint \"{src,apps,libs,test}/**/*.ts\" --config .eslintrc.lint.cjs",
-    "lint:fix": "eslint \"{src,apps,libs,test}/**/*.ts\" --fix --config .eslintrc.lint.cjs",
+    "lint:fix": "eslint \"{src,apps,libs,tests}/**/*.ts\" --fix --config .eslintrc.lint.cjs",
     "lint:types": "tsc --pretty --noEmit",
-    "lint:all": "npm run lint:fix && npm run lint:types"
+    "lint:all": "npm run lint:fix && npm run lint:types",
+    "test": "vitest run",
+    "test:watch": "vitest dev",
+    "test:ui": "vitest dev --ui"
   },
   "license": "Apache-2.0",
   "author": {
@@ -82,13 +85,15 @@
     "@types/to-json-schema": "0.2.1",
     "@typescript-eslint/eslint-plugin": "5.57.0",
     "@typescript-eslint/parser": "5.57.0",
+    "@vitest/ui": "^0.29.8",
     "eslint": "8.37.0",
     "eslint-config-prettier": "8.8.0",
     "eslint-plugin-prettier": "4.2.1",
     "eslint-plugin-unused-imports": "2.0.0",
     "prettier": "2.8.7",
     "prompts": "2.4.2",
     "tsx": "3.12.6",
-    "typescript": "4.9.5"
+    "typescript": "4.9.5",
+    "vitest": "^0.29.8"
   }
 }

src/codegen/generator.ts

@@ -88,7 +88,7 @@ export const prepare = async (schema: string, tableName: string) => {
   }
 }
 
-const getSchemaTableColumns = async (
+export const getSchemaTableColumns = async (
   schema: string,
   tableName: string,
 ): Promise<SchemaTableColumn[]> => {

src/helpers/convert.ts

@@ -112,7 +112,7 @@ export const generateModel = ({
   return models
 }
 
-const convertField = ({
+export const convertField = ({
   fieldName,
   fieldDefinition,
   parent,

tests/codegen/__snapshots__/generator.test.ts.snap

@@ -0,0 +1,51 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`Codegen - Generator > getSchemaTableColumns 1`] = `
+[
+  {
+    "description": "",
+    "name": "stringarray",
+    "type": "array(varchar)",
+  },
+  {
+    "description": "",
+    "name": "stringfield",
+    "type": "varchar",
+  },
+  {
+    "description": "",
+    "name": "booleanfield",
+    "type": "boolean",
+  },
+  {
+    "description": "",
+    "name": "datetimefield",
+    "type": "timestamp(3)",
+  },
+  {
+    "description": "",
+    "name": "bigintfield",
+    "type": "bigint",
+  },
+  {
+    "description": "",
+    "name": "simpleobject",
+    "type": "row(sosub1 varchar, sosub2 bigint)",
+  },
+  {
+    "description": "",
+    "name": "complexobject",
+    "type": "row(cosub1 varchar, cosub2 array(varchar), cosub3 array(row(cosub31 varchar, cosub32 varchar)))",
+  },
+  {
+    "description": "",
+    "name": "dash-field",
+    "type": "varchar",
+  },
+  {
+    "description": "",
+    "name": "special-aou",
+    "type": "varchar",
+  },
+]
+`;

tests/codegen/generator.test.ts

@@ -0,0 +1,21 @@
+import { expect, test, describe, vi } from 'vitest'
+import { getSchemaTableColumns } from '../../src/codegen/generator'
+import { sqlTableColumns } from '../mockdata'
+
+describe('Codegen - Generator', () => {
+  test('getSchemaTableColumns', async () => {
+    const fnResult = await getSchemaTableColumns('testSchema', 'testTable')
+    expect(fnResult).toMatchSnapshot()
+  })
+})
+
+//Mock the Presto Client
+vi.mock('src/client/sql-client', () => {
+  const SqlClient = vi.fn()
+  SqlClient.prototype.query = vi.fn(async () => {
+    return sqlTableColumns
+  })
+  SqlClient.prototype.end = vi.fn()
+
+  return { SqlClient }
+})