sqlite-to-snake.test.ts — drizzle-orm Source File

Source Code

import Database from 'better-sqlite3';
import { beforeEach, describe, it } from 'vitest';
import { drizzle } from '~/better-sqlite3';
import { relations } from '~/relations';
import { asc, eq, sql } from '~/sql';
import { alias, integer, sqliteTable, text, union } from '~/sqlite-core';

const users = sqliteTable('users', {
	id: integer().primaryKey({ autoIncrement: true }),
	firstName: text().notNull(),
	lastName: text().notNull(),
	// Test that custom aliases remain
	age: integer('AGE'),
});
const usersRelations = relations(users, ({ one }) => ({
	developers: one(developers),
}));
const developers = sqliteTable('developers', {
	userId: integer().primaryKey().references(() => users.id),
	usesDrizzleORM: integer({ mode: 'boolean' }).notNull(),
});
const developersRelations = relations(developers, ({ one }) => ({
	user: one(users, {
		fields: [developers.userId],
		references: [users.id],
	}),
}));
const devs = alias(developers, 'devs');
const schema = { users, usersRelations, developers, developersRelations };

const db = drizzle(new Database(':memory:'), { schema, casing: 'snake_case' });

const usersCache = {
	'public.users.id': 'id',
	'public.users.firstName': 'first_name',
	'public.users.lastName': 'last_name',
	'public.users.AGE': 'age',
};
const developersCache = {
	'public.developers.userId': 'user_id',
	'public.developers.usesDrizzleORM': 'uses_drizzle_orm',
};
const cache = {
	...usersCache,
	...developersCache,
};

const fullName = sql`${users.firstName} || ' ' || ${users.lastName}`.as('name');

describe('sqlite to camel case', () => {
	beforeEach(() => {
		db.dialect.casing.clearCache();
	});

	it('select', ({ expect }) => {
		const query = db
			.select({ name: fullName, age: users.age })
			.from(users)
			.leftJoin(developers, eq(users.id, developers.userId))
			.orderBy(asc(users.firstName));
// ... (187 more lines)