Unrecognized

Author: jcobis

packages/compass-collection/src/components/mock-data-generator-modal/script-generation-utils.spec.ts

@@ -346,4 +346,153 @@ describe('Script Generation', () => {
       }
     });
   });
+
+  describe('Unrecognized Field Defaults', () => {
+    it('should use default faker method for unrecognized string fields', () => {
+      const schema = {
+        unknownField: {
+          mongoType: 'string',
+          fakerMethod: 'unrecognized',
+          fakerArgs: [],
+        },
+      };
+
+      const result = generateScript(schema, {
+        databaseName: 'testdb',
+        collectionName: 'test',
+        documentCount: 1,
+      });
+
+      expect(result.success).to.equal(true);
+      if (result.success) {
+        expect(result.script).to.contain('faker.lorem.word()');
+      }
+    });
+
+    it('should use default faker method for unrecognized number fields', () => {
+      const schema = {
+        unknownNumber: {
+          mongoType: 'number',
+          fakerMethod: 'unrecognized',
+          fakerArgs: [],
+        },
+      };
+
+      const result = generateScript(schema, {
+        databaseName: 'testdb',
+        collectionName: 'test',
+        documentCount: 1,
+      });
+
+      expect(result.success).to.equal(true);
+      if (result.success) {
+        expect(result.script).to.contain('faker.number.int()');
+      }
+    });
+
+    it('should use default faker method for unrecognized date fields', () => {
+      const schema = {
+        unknownDate: {
+          mongoType: 'date',
+          fakerMethod: 'unrecognized',
+          fakerArgs: [],
+        },
+      };
+
+      const result = generateScript(schema, {
+        databaseName: 'testdb',
+        collectionName: 'test',
+        documentCount: 1,
+      });
+
+      expect(result.success).to.equal(true);
+      if (result.success) {
+        expect(result.script).to.contain('faker.date.recent()');
+      }
+    });
+
+    it('should use default faker method for unrecognized boolean fields', () => {
+      const schema = {
+        unknownBool: {
+          mongoType: 'boolean',
+          fakerMethod: 'unrecognized',
+          fakerArgs: [],
+        },
+      };
+
+      const result = generateScript(schema, {
+        databaseName: 'testdb',
+        collectionName: 'test',
+        documentCount: 1,
+      });
+
+      expect(result.success).to.equal(true);
+      if (result.success) {
+        expect(result.script).to.contain('faker.datatype.boolean()');
+      }
+    });
+
+    it('should use default faker method for unrecognized ObjectId fields', () => {
+      const schema = {
+        unknownId: {
+          mongoType: 'objectid',
+          fakerMethod: 'unrecognized',
+          fakerArgs: [],
+        },
+      };
+
+      const result = generateScript(schema, {
+        databaseName: 'testdb',
+        collectionName: 'test',
+        documentCount: 1,
+      });
+
+      expect(result.success).to.equal(true);
+      if (result.success) {
+        expect(result.script).to.contain('faker.database.mongodbObjectId()');
+      }
+    });
+
+    it('should use default faker method for unrecognized double fields', () => {
+      const schema = {
+        unknownDouble: {
+          mongoType: 'double',
+          fakerMethod: 'unrecognized',
+          fakerArgs: [],
+        },
+      };
+
+      const result = generateScript(schema, {
+        databaseName: 'testdb',
+        collectionName: 'test',
+        documentCount: 1,
+      });
+
+      expect(result.success).to.equal(true);
+      if (result.success) {
+        expect(result.script).to.contain('faker.number.float()');
+      }
+    });
+
+    it('should fall back to lorem.word for unknown MongoDB types', () => {
+      const schema = {
+        unknownType: {
+          mongoType: 'unknownType',
+          fakerMethod: 'unrecognized',
+          fakerArgs: [],
+        },
+      };
+
+      const result = generateScript(schema, {
+        databaseName: 'testdb',
+        collectionName: 'test',
+        documentCount: 1,
+      });
+
+      expect(result.success).to.equal(true);
+      if (result.success) {
+        expect(result.script).to.contain('faker.lorem.word()');
+      }
+    });
+  });
 });

packages/compass-collection/src/components/mock-data-generator-modal/script-generation-utils.ts

@@ -389,6 +389,38 @@ function generateArrayCode(
  * Generate faker.js call from field mapping
  */
 function generateFakerCall(mapping: FieldMapping): string {
+  const method =
+    mapping.fakerMethod === 'unrecognized'
+      ? getDefaultFakerMethod(mapping.mongoType)
+      : mapping.fakerMethod;
+
   // TODO: Handle arguments properly
-  return `faker.${mapping.fakerMethod}()`;
+  return `faker.${method}()`;
+}
+
+/**
+ * Gets default faker method for unrecognized fields based on MongoDB type
+ */
+export function getDefaultFakerMethod(mongoType: string): string {
+  switch (mongoType.toLowerCase()) {
+    case 'string':
+      return 'lorem.word';
+    case 'number':
+    case 'int32':
+    case 'int64':
+      return 'number.int';
+    case 'double':
+    case 'decimal128':
+      return 'number.float';
+    case 'date':
+      return 'date.recent';
+    case 'objectid':
+      return 'database.mongodbObjectId';
+    case 'boolean':
+      return 'datatype.boolean';
+    case 'binary':
+      return 'string.hexadecimal';
+    default:
+      return 'lorem.word';
+  }
 }