@liminalfunctions/framework 1.0.43 → 1.0.46

package/dist/code_generation/templates/main.mustache

@@ -7,18 +7,29 @@ import { Collection_{{built_collection}} } from "{{ built_collection_path }}"
 
 
 export function api(base_url: string, get_auth: () => Promise<any>) {
-    return {
-        collection(collection_id: {{child_collection_id_types}}) {
-            switch(collection_id) {
-                {{#child_collections}}
-                case "{{collection_id}}":
-                    return new Collection_{{built_collection}}([base_url, "{{collection_id}}"], get_auth);
-                {{/child_collections}}
-                default:
-                    throw new Error(`Api does not have the collection ${collection_id}`)
-            }
-        }
-    }
+    return new Api(base_url, get_auth)
 }
 
+class Api {
+    base_url: string;
+    get_auth: () => Promise<any>
+
+    constructor(base_url: string, get_auth: () => Promise<any>) {
+        this.base_url = base_url;
+        this.get_auth = get_auth
+    }
 
+    {{#child_collections}}
+    collection(collection_id: "{{collection_id}}"): Collection_{{built_collection}};
+    {{/child_collections}}
+    collection(collection_id: string) {
+        switch(collection_id) {
+            {{#child_collections}}
+            case "{{collection_id}}":
+                return new Collection_{{built_collection}}([this.base_url, "{{collection_id}}"], this.get_auth);
+            {{/child_collections}}
+            default:
+                throw new Error(`Api does not have the collection ${collection_id}`)
+        }
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@liminalfunctions/framework",
-  "version": "1.0.43",
+  "version": "1.0.46",
   "main": "./dist/index.js",
   "type": "module",
   "scripts": {

package/src/F_Collection.ts

@@ -48,6 +48,13 @@ export class F_Collection<Collection_ID extends string, ZodSchema extends z.ZodO
         this.query_validator_server = query_validator_from_zod(validator, 'server');
         this.query_validator_client = query_validator_from_zod(validator, 'client');
         // TODO: we can make this more closely match the mongoDB PUT operation and allow updates to eg array.3.element fields
+
+        if(!Object.hasOwn(this.validator._zod.def.shape, '_id')){
+            throw new Error(`_id is a required field, because each collection is a mongoDB object.`)
+        }
+        if(this.validator._zod.def.shape._id.meta()?.framework_override_type !== 'mongodb_id'){
+            throw new Error(`_id must be a mongoDB ID. Use the z_mongodb_id special field.`)
+        }
         
         // TODO: find a more elegant way to do this so that the types don't have a cow
         //@ts-ignore

package/src/F_Collection_Registry.ts

@@ -23,7 +23,7 @@ export class F_Collection_Registry<Collections = {}> {
 
     compile(app: Router, api_prefix: string) {
         for(let collection of Object.values(this.collections)){
-            compile(app, collection, api_prefix)
+            compile(app, collection, api_prefix, this)
         }
     }
 }

package/src/F_Compile.ts

@@ -6,20 +6,51 @@ import { F_Collection } from "./F_Collection.js";
 import { F_Security_Model, Authenticated_Request } from "./F_Security_Models/F_Security_Model.js";
 import { convert_null, query_object_to_mongodb_limits, query_object_to_mongodb_query } from "./utils/query_object_to_mongodb_query.js";
 import { z_mongodb_id } from "./utils/mongoose_from_zod.js";
+import { F_Collection_Registry } from "./F_Collection_Registry.js";
 
 /*process.on('unhandledRejection', (reason, promise) => {
     console.log(`CAUGHT UNHANDLED REJECTION`)
     console.log(reason)
 })*/
 
-export function compile<Collection_ID extends string, ZodSchema extends z.ZodObject>(app: Router, collection: F_Collection<Collection_ID, ZodSchema>, api_prefix: string){
+export function compile<Collection_ID extends string, ZodSchema extends z.ZodObject>(
+    app: Router,
+    collection: F_Collection<Collection_ID, ZodSchema>,
+    api_prefix: string,
+    collection_registry: F_Collection_Registry<any>){
     /*app.use((req, res, next) => {
         console.log(`${req.method} ${req.originalUrl}`)
         next();
     })*/
+
+    
+    for(let access_layers of collection.access_layers){
+        for(let layer of access_layers.layers){
+            // verify that the collection is not in its own layers
+            if(layer === collection.collection_id){
+                throw new Error(`Error compiling collection ${collection.collection_id}: a collection cannot be a member of it's own layer. Remove "${collection.collection_id}" from the collection's layers.`)
+            }
+
+            // verify that each layer has a corresponding collection
+            if(!collection_registry.collections[layer]){ 
+                throw new Error(`Error compiling collection ${collection.collection_id}: collection registry does not have a collection with the ID "${layer}". Each layer must be a valid collection ID.`)
+            }
+
+            if(!Object.hasOwn(collection.validator._zod.def.shape, `${layer}_id`)) {
+                throw new Error(`Error compiling collection ${collection.collection_id}: collection does not have a field "${layer}_id. Either remove ${layer} from the collection's layers, or add a field ${layer}_id`)
+            }
+
+            let layer_id_is_mongodb_id = collection.validator._zod.def.shape[`${layer}_id`].meta()?.framework_override_type === 'mongodb_id';
+            if(!layer_id_is_mongodb_id){
+                throw new Error(`Error compiling collection ${collection.collection_id}:  ${layer}_id must be a mongodb ID. use the z_mongodb_id, z_mongodb_id_nullable, or z_mongodb_id_optional special fields.`)
+            }
+        }
+    }
     
+    // set up the Express endpoints
     for(let access_layers of collection.access_layers){
         let base_layers_path_components = access_layers.layers.flatMap(ele => [ele, ':' + ele]);
+        
 
         let get_one_path = [
             api_prefix,

package/src/code_generation/templates/collection.mustache

@@ -54,64 +54,80 @@ export class Collection_{{my_built_collection}} {
         let get_auth = this.get_auth;
         let collection_id = this.collection_id;
         let collection_name_plural = this.collection_name_plural;
-        return {
-            path: [...path, document_id],
-            collection_id: collection_id,
-            document_id: document_id,
-            collection_name_plural: collection_name_plural,
-            get_auth: get_auth,
-            async get(): Promise<{{type_return}}>{
-                try {
-                    let result = await ky.get([...path, document_id].join('/'), {
-                        headers: {
-                            authorization: await get_auth()
-                        },
-                    }).json() as Response<{{type_return}}>;
-                    return result.data;
-                } catch(err){
-                    return Promise.reject(err)
-                }
-            },
+        return new Document(path, collection_id, document_id, collection_name_plural, get_auth);
+    }
+}
+
+
+
+class Document {
+    path: string[];
+    collection_id: string;
+    document_id: string;
+    collection_name_plural: string;
+    get_auth: () => Promise<any>;
+
+    constructor(path: string[], collection_id: string, document_id: string, collection_name_plural: string, get_auth: () => Promise<any>) {
+        this.path = path;
+        this.collection_id = collection_id;
+        this.document_id = document_id;
+        this.collection_name_plural = collection_name_plural;
+        this.get_auth = get_auth;
+    }
+
+    async get(): Promise<{{type_return}}>{
+        try {
+            let result = await ky.get([...this.path, this.document_id].join('/'), {
+                headers: {
+                    authorization: await this.get_auth()
+                },
+            }).json() as Response<{{type_return}}>;
+            return result.data;
+        } catch(err){
+            return Promise.reject(err)
+        }
+    }
 
-            async put(update: {{type_put}}): Promise<{{type_return}}>{
-                try {
-                    let result = await ky.put([...path, document_id].join('/'), {
-                        headers: {
-                            authorization: await get_auth()
-                        },
-                        json: update
-                    }).json() as Response<{{type_return}}>;
-                    return result.data;
-                } catch(err){
-                    return Promise.reject(err)
-                }
-            },
-            
-            async remove(): Promise<{{type_return}}>{
-                try {
-                    let result = await ky.delete([...path, document_id].join('/'), {
-                        headers: {
-                            authorization: await get_auth()
-                        },
-                    }).json() as Response<{{type_return}}>;
-                    return result.data;
-                } catch(err){
-                    return Promise.reject(err)
-                }
-            },
+    async put(update: {{type_put}}): Promise<{{type_return}}>{
+        try {
+            let result = await ky.put([...this.path, this.document_id].join('/'), {
+                headers: {
+                    authorization: await this.get_auth()
+                },
+                json: update
+            }).json() as Response<{{type_return}}>;
+            return result.data;
+        } catch(err){
+            return Promise.reject(err)
+        }
+    }
+    
+    async remove(): Promise<{{type_return}}>{
+        try {
+            let result = await ky.delete([...this.path, this.document_id].join('/'), {
+                headers: {
+                    authorization: await this.get_auth()
+                },
+            }).json() as Response<{{type_return}}>;
+            return result.data;
+        } catch(err){
+            return Promise.reject(err)
+        }
+    }
 
-            {{#has_subcollections}}
-            collection(collection_id: {{child_collection_id_types}}) {
-                switch(collection_id) {
-                    {{#child_collections}}
-                    case "{{collection_id}}":
-                        return new Collection_{{built_collection}}([...path, document_id, "{{collection_id}}"], get_auth);
-                    {{/child_collections}}
-                    default:
-                        throw new Error(`Api does not have the collection ${collection_id}`)
-                }
-            }
-            {{/has_subcollections}}
+    {{#has_subcollections}}
+    {{#child_collections}}
+    collection(collection_id: "{{collection_id}}"): Collection_{{built_collection}};
+    {{/child_collections}}
+    collection(collection_id: string) {
+        switch(collection_id) {
+            {{#child_collections}}
+            case "{{collection_id}}":
+                return new Collection_{{built_collection}}([...this.path, this.document_id, "{{collection_id}}"], this.get_auth);
+            {{/child_collections}}
+            default:
+                throw new Error(`Api does not have the collection ${collection_id}`)
         }
     }
+    {{/has_subcollections}}
 }

package/src/code_generation/templates/main.mustache

@@ -7,18 +7,29 @@ import { Collection_{{built_collection}} } from "{{ built_collection_path }}"
 
 
 export function api(base_url: string, get_auth: () => Promise<any>) {
-    return {
-        collection(collection_id: {{child_collection_id_types}}) {
-            switch(collection_id) {
-                {{#child_collections}}
-                case "{{collection_id}}":
-                    return new Collection_{{built_collection}}([base_url, "{{collection_id}}"], get_auth);
-                {{/child_collections}}
-                default:
-                    throw new Error(`Api does not have the collection ${collection_id}`)
-            }
-        }
-    }
+    return new Api(base_url, get_auth)
 }
 
+class Api {
+    base_url: string;
+    get_auth: () => Promise<any>
+
+    constructor(base_url: string, get_auth: () => Promise<any>) {
+        this.base_url = base_url;
+        this.get_auth = get_auth
+    }
 
+    {{#child_collections}}
+    collection(collection_id: "{{collection_id}}"): Collection_{{built_collection}};
+    {{/child_collections}}
+    collection(collection_id: string) {
+        switch(collection_id) {
+            {{#child_collections}}
+            case "{{collection_id}}":
+                return new Collection_{{built_collection}}([this.base_url, "{{collection_id}}"], this.get_auth);
+            {{/child_collections}}
+            default:
+                throw new Error(`Api does not have the collection ${collection_id}`)
+        }
+    }
+}

package/test/2_0_client_library_basic_type_generation.test.ts

@@ -38,7 +38,9 @@ describe('Client Library Generation: Basic Types', function () {
     })
 
     it(`should be able to generate a plain object`, async function () {
-        const validate_test_collection = z.object({});
+        const validate_test_collection = z.object({
+            _id: z_mongodb_id
+        });
 
         let test_collection = new F_Collection('test_collection', 'test_collection', validate_test_collection);
 
@@ -49,12 +51,13 @@ describe('Client Library Generation: Basic Types', function () {
 
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
-            remove_whitespace(`export type test_collection = {}`)
+            remove_whitespace(`export type test_collection = {"_id": string}`)
         )
     });
 
     it(`should be able to generate a plain object containing a string`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.string(),
         });
 
@@ -68,6 +71,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": string
                 }`)
         )
@@ -75,6 +79,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain object containing a number`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.number(),
         });
 
@@ -88,6 +93,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": number
                 }`)
         )
@@ -95,6 +101,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain object containing a boolean`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.boolean(),
         });
 
@@ -108,6 +115,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": boolean
                 }`)
         )
@@ -115,6 +123,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain object containing a date`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.date(),
         });
 
@@ -128,6 +137,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": Date
                 }`)
         )
@@ -135,6 +145,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain object containing an objectID`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z_mongodb_id,
         });
 
@@ -148,6 +159,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": string
                 }`)
         )
@@ -155,6 +167,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain object containing a nullable string`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.nullable(z.string()),
         });
 
@@ -168,6 +181,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": string | null
                 }`)
         )
@@ -175,6 +189,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain object containing union types`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.string().or(z.number()),
         });
 
@@ -188,6 +203,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": string | number
                 }`)
         )
@@ -195,6 +211,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain object containing union types wrapped in nullable`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.nullable(z.string().or(z.number())),
         });
 
@@ -208,6 +225,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": string | number | null
                 }`)
         )
@@ -215,6 +233,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain object containing union of object types`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.object({
                 sub: z.string()
             }).or(z.object({
@@ -232,6 +251,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": {"sub": string} | {"sub2": number}
                 }`)
         )
@@ -239,6 +259,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate an enum`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.enum(["red", "green", "blue"]),
         });
 
@@ -252,6 +273,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": ("red" | "green" | "blue")
                 }`)
         )
@@ -259,6 +281,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate an array of enum`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.array(z.enum(["red", "green", "blue"])),
         });
 
@@ -272,6 +295,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": ("red" | "green" | "blue")[]
                 }`)
         )
@@ -279,6 +303,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain nested object`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.object({
             }),
         });
@@ -293,6 +318,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": {}
                 }`)
         )
@@ -300,6 +326,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain nested with basic fields`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.object({
                 field_string: z.string(),
                 field_number: z.number(),
@@ -321,6 +348,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": {
                         "field_string": string
                         "field_number": number
@@ -336,6 +364,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate arrays of primitive fields`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             field_string: z.array(z.string()),
             field_number: z.array(z.number()),
             field_boolean: z.array(z.boolean()),
@@ -352,6 +381,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "field_string": string[]
                     "field_number": number[]
                     "field_boolean": boolean[]
@@ -362,6 +392,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate arrays of objects containing primitive fields`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             field_array: z.array(z.object({
                 field_string: z.array(z.string()),
                 field_number: z.array(z.number()),
@@ -381,6 +412,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                "_id": string
                 "field_array": {
                         "field_string": string[]
                         "field_number": number[]
@@ -393,6 +425,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate nested arrays`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             field_array: z.array(
                 z.array(z.object({
                     field_string: z.array(z.string()),
@@ -413,6 +446,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                "_id": string
                 "field_array": {
                         "field_string": string[]
                         "field_number": number[]
@@ -425,6 +459,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain object containing a primitive with a default`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.string().default('ezikiel snograss'),
         });
 
@@ -438,6 +473,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test"?: string
                 }`)
         )
@@ -445,6 +481,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain object containing an optional primitive`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.string().optional(),
         });
 
@@ -458,6 +495,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test"?: string
                 }`)
         )
@@ -465,6 +503,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain object containing a record`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.record(z.string(), z.string())
         });
 
@@ -478,6 +517,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": {[key: string]: string}
                 }`)
         )
@@ -486,6 +526,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain object containing an object record`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.record(z.string(), z.object({
                 test_2: z.string()
             }))
@@ -501,6 +542,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": {[key: string]: { "test_2": string }}
                 }`)
         )
@@ -508,6 +550,7 @@ describe('Client Library Generation: Basic Types', function () {
 
     it(`should be able to generate a plain object containing an array record`, async function () {
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: z.record(z.string(), z.array(z.string()))
         });
 
@@ -521,6 +564,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": {[key: string]: string[]}
                 }`)
         )
@@ -535,6 +579,7 @@ describe('Client Library Generation: Basic Types', function () {
         }).meta({id: 'test_recursive_object'})
 
         const validate_test_collection = z.object({
+            _id: z_mongodb_id,
             test: recursive
         });
 
@@ -548,6 +593,7 @@ describe('Client Library Generation: Basic Types', function () {
         assert.equal(
             remove_whitespace(await readFile('./test/tmp/src/types/test_collection.ts', { encoding: 'utf-8' })),
             remove_whitespace(`export type test_collection = {
+                    "_id": string
                     "test": type_test_recursive_object
                 }
                 type type_test_recursive_object = {