feat(mock): infer type from enum values for string and number (orval-…

…labs#1491)

* feat(mock): infer type from enum values for string and number

* cleanup

packages/mock/src/faker/getters/scalar.ts

@@ -108,7 +108,9 @@ export const getMockScalar = ({
  };
  }
 
- switch (item.type) {
+ const type = getItemType(item);
+
+ switch (type) {
  case 'number':
  case 'integer': {
  let value = getNullable(
@@ -303,3 +305,15 @@ export const getMockScalar = ({
  }
  }
 };
+
+function getItemType(item: MockSchemaObject) {
+ if (item.type) return item.type;
+ if (!item.enum) return;
+
+ const uniqTypes = new Set(item.enum.map((value) => typeof value));
+ if (uniqTypes.size > 1) return;
+
+ const type = Array.from(uniqTypes.values()).at(0);
+ if (!type) return;
+ return ['string', 'number'].includes(type) ? type : undefined;
+}

tests/configs/mock.config.ts

@@ -126,4 +126,12 @@ export default defineConfig({
  target: '../specifications/enum-refs.yaml',
  },
  },
+ typelessEnum: {
+ input: '../specifications/typelessEnum.yaml',
+ output: {
+ schemas: '../generated/mock/typelessEnum/schemas',
+ target: '../generated/mock/typelessEnum',
+ mock: true,
+ },
+ },
 });

tests/specifications/typelessEnum.yaml

@@ -0,0 +1,29 @@
+openapi: 3.0.3
+info:
+ title: Typeless enums
+ version: 1.0.0
+paths:
+ /api/colors:
+ get:
+ summary: sample colors
+ responses:
+ '200':
+ description: OK
+ content:
+ application/json:
+ schema:
+ $ref: '#/components/schemas/ColorObject'
+components:
+ schemas:
+ Colors1:
+ type: string
+ enum: [red, blue, yellow]
+ Colors2:
+ enum: [green, purple, orange]
+ ColorObject:
+ type: object
+ properties:
+ color:
+ oneOf:
+ - $ref: '#/components/schemas/Colors1'
+ - $ref: '#/components/schemas/Colors2'