Function createTestClient

packages/testtools/src/client.ts:130–279

(
    schema: SchemaDef | string,
    options?: CreateTestClientOptions<SchemaDef>,
)

>(schema: Schema, options?: CreateOptions): Promise<ClientContract<Schema, Options>>;
export async function createTestClient(schema: string, options?: CreateTestClientOptions<SchemaDef>): Promise<any>;
export async function createTestClient(
    schema: SchemaDef | string,
    options?: CreateTestClientOptions<SchemaDef>,
): Promise<any> {
    let workDir = options?.workDir;
    let _schema: SchemaDef;
    const provider = options?.provider ?? getTestDbProvider() ?? 'sqlite';
    const dbName = options?.dbName ?? getTestDbName(provider);
    const dbUrl = match(provider)
        .with('sqlite', () => `file:${dbName}`)
        .with('mysql', () => `${TEST_MYSQL_URL}/${dbName}`)
        .with('postgresql', () => `${TEST_PG_URL}/${dbName}`)
        .exhaustive();
    let model: Model | undefined;

    if (typeof schema === 'string') {
        const generated = await generateTsSchema(
            schema,
            provider,
            dbUrl,
            options?.extraSourceFiles,
            undefined,
            options?.extraZModelFiles,
            options?.extraPluginModelFiles,
        );
        workDir = generated.workDir;
        model = generated.model;
        // replace schema's provider
        _schema = {
            ...generated.schema,
            provider: {
                ...generated.schema.provider,
                type: provider,
            },
        } as SchemaDef;
    } else {
        // replace schema's provider
        _schema = {
            ...schema,
            provider: {
                type: provider,
            },
        };
        workDir ??= createTestProject();
        if (options?.schemaFile) {
            let schemaContent = fs.readFileSync(options.schemaFile, 'utf-8');
            if (dbUrl) {
                // replace `datasource db { }` section
                schemaContent = schemaContent.replace(
                    /datasource\s+db\s*{[^}]*}/m,
                    `datasource db {
    provider = '${provider}'
    url = '${dbUrl}'
    ${options.dataSourceExtensions ? `extensions = [${options.dataSourceExtensions.join(', ')}]` : ''}
}`,
                );
            }
            fs.writeFileSync(path.join(workDir!, 'schema.zmodel'), schemaContent);