Function loadSchema

packages/testtools/src/schema.ts:143–173

(schema: string, additionalSchemas?: Record<string, string>)

}

export async function loadSchema(schema: string, additionalSchemas?: Record<string, string>) {
    if (!schema.includes('datasource ')) {
        schema = `${makePrelude('sqlite')}\n\n${schema}`;
    }

    // create a temp folder
    const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'zenstack-schema'));

    // create a temp file
    const tempFile = path.join(tempDir, `schema.zmodel`);
    fs.writeFileSync(tempFile, schema);

    if (additionalSchemas) {
        for (const [fileName, content] of Object.entries(additionalSchemas)) {
            let name = fileName;
            if (!name.endsWith('.zmodel')) {
                name += '.zmodel';
            }
            const filePath = path.join(tempDir, name);
            fs.writeFileSync(filePath, content);
        }
    }

    const r = await loadDocumentWithPlugins(tempFile);
    expect(r).toSatisfy(
        (r) => r.success,
        `Failed to load schema: ${(r as any).errors?.map((e: any) => e.toString()).join(', ')}`,
    );
    invariant(r.success);
    return r.model;
}

export async function loadSchemaWithError(schema: string, error: string | RegExp) {
    if (!schema.includes('datasource ')) {