feat: 补充code传入方式获取jsonschema单测用例

Author: wangw19

src/__tests__/__snapshots__/code.test.ts.snap

@@ -0,0 +1,121 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`Type类型 1`] = `
+Object {
+  "anyOf": Array [
+    Object {
+      "type": "string",
+    },
+    Object {
+      "enum": Object {
+        "enum": Array [
+          "home",
+          "about",
+          "contact",
+        ],
+      },
+      "type": "string",
+    },
+  ],
+}
+`;
+
+exports[`嵌套接口 1`] = `
+Object {
+  "additionalProperties": false,
+  "definitions": Object {
+    "BBB": Object {
+      "additionalProperties": false,
+      "properties": Object {
+        "some": Object {
+          "type": "string",
+        },
+      },
+      "required": Array [
+        "some",
+      ],
+      "type": "object",
+    },
+  },
+  "properties": Object {
+    "some": Object {
+      "$ref": "#/definitions/BBB",
+    },
+    "title": Object {
+      "type": "string",
+    },
+  },
+  "required": Array [
+    "title",
+    "some",
+  ],
+  "type": "object",
+}
+`;
+
+exports[`嵌套泛型 1`] = `
+Object {
+  "additionalProperties": false,
+  "definitions": Object {},
+  "properties": Object {
+    "value1": Object {
+      "type": "string",
+    },
+    "value2": Object {
+      "type": "string",
+    },
+  },
+  "required": Array [
+    "value1",
+    "value2",
+  ],
+  "type": "object",
+}
+`;
+
+exports[`枚举 1`] = `
+Object {
+  "enum": Array [
+    "Maintainer",
+    "Developer",
+  ],
+  "type": "string",
+}
+`;
+
+exports[`泛型 1`] = `
+Object {
+  "additionalProperties": false,
+  "definitions": Object {
+    "T": Object {},
+  },
+  "properties": Object {
+    "value1": Object {
+      "type": "string",
+    },
+    "value2": Object {
+      "$ref": "#/definitions/T",
+    },
+  },
+  "required": Array [
+    "value1",
+    "value2",
+  ],
+  "type": "object",
+}
+`;
+
+exports[`简单接口 1`] = `
+Object {
+  "additionalProperties": false,
+  "properties": Object {
+    "some": Object {
+      "type": "string",
+    },
+  },
+  "required": Array [
+    "some",
+  ],
+  "type": "object",
+}
+`;

src/__tests__/__snapshots__/generic.test.ts.snap

@@ -256,6 +256,28 @@ Object {
 }
 `;
 
+exports[`Generic类型_3 1`] = `
+Object {
+  "additionalProperties": false,
+  "definitions": Object {
+    "T": Object {},
+  },
+  "properties": Object {
+    "value1": Object {
+      "type": "string",
+    },
+    "value2": Object {
+      "$ref": "#/definitions/T",
+    },
+  },
+  "required": Array [
+    "value1",
+    "value2",
+  ],
+  "type": "object",
+}
+`;
+
 exports[`Generic默认值_类型_1 1`] = `
 Object {
   "additionalProperties": false,

src/__tests__/code.test.ts

@@ -0,0 +1,53 @@
+import genTypeSchema from '../index';
+
+const code = `
+import AAA from './aaa'
+export type Generic_10 = Generic_type_1<string>
+export interface Generic_type_1<T> {
+    value1: string;
+    value2: T;
+}
+interface BBB {
+    some: string;
+}
+export enum Role {
+  Maintainer = 'Maintainer',
+  Developer = 'Developer',
+}
+type Page = 'home' | 'about' | 'contact' | string;
+interface PageInfo {
+  title: string;
+  some: BBB
+}
+`
+genTypeSchema.genJsonDataFromCode(code);
+
+const getSchema = (type: string) => {
+  const schema = genTypeSchema.getJsonSchema(type);
+  return schema;
+};
+
+test('枚举', () => {
+  expect(getSchema('Role')).toMatchSnapshot();
+});
+
+test('Type类型', () => {
+  expect(getSchema('Page')).toMatchSnapshot();
+});
+
+test('简单接口', () => {
+  expect(getSchema('BBB')).toMatchSnapshot();
+});
+
+test('嵌套接口', () => {
+  expect(getSchema('PageInfo')).toMatchSnapshot();
+});
+
+test('泛型', () => {
+  expect(getSchema('Generic_type_1')).toMatchSnapshot();
+});
+
+test('嵌套泛型', () => {
+  expect(getSchema('Generic_10')).toMatchSnapshot();
+});
+

src/__tests__/generic.test.ts

@@ -22,6 +22,10 @@ test('Generic类型_2', () => {
   expect(getSchema('Generic_5')).toMatchSnapshot();
 });
 
+test('Generic类型_3', () => {
+  expect(getSchema('Generic_type_1')).toMatchSnapshot();
+});
+
 test('Generic默认值_类型_1', () => {
   expect(getSchema('Generic_7')).toMatchSnapshot();
 });

src/get-jsonschema-from-data.ts

@@ -32,7 +32,7 @@ export default class genTypeSchema extends typescriptToFileDatas {
    */
   getJsonSchema(
     file: string,
-    type: string,
+    type?: string,
     entry?: { keySet: Set<string>; refKeyTime: Record<string, number> },
     isLast?: boolean
   ): AnyOption {

src/utils.ts

@@ -44,7 +44,6 @@ export function genAst(file: string): AnyOption | null {
  * @return {*}  {(AnyOption | null)}
  */
 export function genAstFromCode(code: string): AnyOption | null {
-  if (!code) return null;
   try {
     const ast: AstType = parse(code, parseConfig);
     return ast;