102 lines
2.4 KiB
JavaScript
102 lines
2.4 KiB
JavaScript
import 'dotenv/config';
|
|
import { Client } from '@planetscale/database';
|
|
import { drizzle } from 'drizzle-orm/planetscale-serverless';
|
|
import { describe, expect } from 'vitest';
|
|
import { mysql as schema } from './schema.mjs';
|
|
|
|
if (!process.env['PLANETSCALE_CONNECTION_STRING']) {
|
|
throw new Error('PLANETSCALE_CONNECTION_STRING is not defined');
|
|
}
|
|
|
|
describe('planetscale', async (it) => {
|
|
it('drizzle(string)', async () => {
|
|
const db = drizzle(
|
|
process.env['PLANETSCALE_CONNECTION_STRING'],
|
|
);
|
|
|
|
await db.$client.execute('SELECT 1;');
|
|
|
|
expect(db.$client).toBeInstanceOf(Client);
|
|
});
|
|
|
|
it('drizzle(string, config)', async () => {
|
|
const db = drizzle(
|
|
process.env['PLANETSCALE_CONNECTION_STRING'],
|
|
{
|
|
schema,
|
|
},
|
|
);
|
|
|
|
await db.$client.execute('SELECT 1;');
|
|
|
|
expect(db.$client).toBeInstanceOf(Client);
|
|
expect(db.query.User).not.toStrictEqual(undefined);
|
|
});
|
|
|
|
it('drizzle({connection: string, ...config})', async () => {
|
|
const db = drizzle({
|
|
connection: process.env['PLANETSCALE_CONNECTION_STRING'],
|
|
schema,
|
|
});
|
|
|
|
await db.$client.execute('SELECT 1;');
|
|
|
|
expect(db.$client).toBeInstanceOf(Client);
|
|
expect(db.query.User).not.toStrictEqual(undefined);
|
|
});
|
|
|
|
it('drizzle({connection: params, ...config})', async () => {
|
|
const db = drizzle({
|
|
connection: {
|
|
url: process.env['PLANETSCALE_CONNECTION_STRING'],
|
|
},
|
|
schema,
|
|
});
|
|
|
|
await db.$client.execute('SELECT 1;');
|
|
|
|
expect(db.$client).toBeInstanceOf(Client);
|
|
expect(db.query.User).not.toStrictEqual(undefined);
|
|
});
|
|
|
|
it('drizzle(client)', async () => {
|
|
const client = new Client({
|
|
url: process.env['PLANETSCALE_CONNECTION_STRING'],
|
|
});
|
|
const db = drizzle(client);
|
|
|
|
await db.$client.execute('SELECT 1;');
|
|
|
|
expect(db.$client).toBeInstanceOf(Client);
|
|
});
|
|
|
|
it('drizzle(client, config)', async () => {
|
|
const client = new Client({
|
|
url: process.env['PLANETSCALE_CONNECTION_STRING'],
|
|
});
|
|
const db = drizzle(client, {
|
|
schema,
|
|
});
|
|
|
|
await db.$client.execute('SELECT 1;');
|
|
|
|
expect(db.$client).toBeInstanceOf(Client);
|
|
expect(db.query.User).not.toStrictEqual(undefined);
|
|
});
|
|
|
|
it('drizzle({client, ...config})', async () => {
|
|
const client = new Client({
|
|
url: process.env['PLANETSCALE_CONNECTION_STRING'],
|
|
});
|
|
const db = drizzle({
|
|
client,
|
|
schema,
|
|
});
|
|
|
|
await db.$client.execute('SELECT 1;');
|
|
|
|
expect(db.$client).toBeInstanceOf(Client);
|
|
expect(db.query.User).not.toStrictEqual(undefined);
|
|
});
|
|
});
|