Feature/singlestore integration tests

drizzle-kit/src/api.ts

@@ -2,7 +2,7 @@ import { randomUUID } from 'crypto';
 import { LibSQLDatabase } from 'drizzle-orm/libsql';
 import type { MySql2Database } from 'drizzle-orm/mysql2';
 import { PgDatabase } from 'drizzle-orm/pg-core';
-import { SingleStore2Database } from 'drizzle-orm/singlestore';
+import { SingleStoreDriverDatabase } from 'drizzle-orm/singlestore';
 import {
 	columnsResolver,
 	enumsResolver,
@@ -394,7 +394,7 @@ export const generateSingleStoreMigration = async (
 
 export const pushSingleStoreSchema = async (
 	imports: Record<string, unknown>,
-	drizzleInstance: SingleStore2Database<any>,
+	drizzleInstance: SingleStoreDriverDatabase<any>,
 	databaseName: string,
 ) => {
 	const { applySingleStoreSnapshotsDiff } = await import('./snapshotsDiffer');

drizzle-orm/src/monodriver.ts

@@ -30,7 +30,7 @@ import type { NeonDatabase } from './neon-serverless/index.ts';
 import type { NodePgDatabase } from './node-postgres/index.ts';
 import type { PlanetScaleDatabase } from './planetscale-serverless/index.ts';
 import type { PostgresJsDatabase } from './postgres-js/index.ts';
-import type { SingleStore2Database, SingleStore2DrizzleConfig } from './singlestore/driver.ts';
+import type { SingleStoreDriverDatabase, SingleStoreDriverDrizzleConfig } from './singlestore/driver.ts';
 import type { TiDBServerlessDatabase } from './tidb-serverless/index.ts';
 import type { DrizzleConfig } from './utils.ts';
 import type { VercelPgDatabase } from './vercel-postgres/index.ts';
@@ -116,7 +116,7 @@ type ClientDrizzleInstanceMap<TSchema extends Record<string, any>> = {
 	d1: DrizzleD1Database<TSchema>;
 	'bun:sqlite': BunSQLiteDatabase<TSchema>;
 	'better-sqlite3': DrizzleBetterSQLite3Database<TSchema>;
-	singlestore: SingleStore2Database<TSchema>;
+	singlestore: SingleStoreDriverDatabase<TSchema>;
 };
 
 type ClientInstanceMap = {
@@ -138,7 +138,7 @@ type ClientInstanceMap = {
 	d1: D1Database;
 	'bun:sqlite': BunDatabase;
 	'better-sqlite3': BetterSQLite3Database;
-	singlestore: SingleStore2Database;
+	singlestore: SingleStoreDriverDatabase;
 };
 
 type InitializerParams = {
@@ -220,6 +220,7 @@ export async function drizzle<
 			: TClient extends 'neon-serverless' ? DrizzleConfig<TSchema> & {
 					ws?: any;
 				}
+			: TClient extends 'singlestore' ? SingleStoreDriverDrizzleConfig<TSchema>
 			: DrizzleConfig<TSchema>),
 ): Promise<DetermineClient<TClient, TSchema>> {
 	const { connection, ws, ...drizzleConfig } = params as typeof params & {
@@ -394,7 +395,7 @@ export async function drizzle<
 			const instance = createPool(connection as Mysql2Config);
 			const { drizzle } = await import('./mysql2');
 
-			const db = drizzle(instance, drizzleConfig as SingleStore2DrizzleConfig) as any;
+			const db = drizzle(instance, drizzleConfig as SingleStoreDriverDrizzleConfig) as any;
 			db.$client = instance;
 
 			return db;

drizzle-orm/src/monomigrator.ts

@@ -12,7 +12,7 @@ import type { NeonDatabase } from './neon-serverless/index.ts';
 import type { NodePgDatabase } from './node-postgres/index.ts';
 import type { PlanetScaleDatabase } from './planetscale-serverless/index.ts';
 import type { PostgresJsDatabase } from './postgres-js/index.ts';
-import type { SingleStore2Database } from './singlestore/driver.ts';
+import type { SingleStoreDriverDatabase } from './singlestore/driver.ts';
 import type { TiDBServerlessDatabase } from './tidb-serverless/index.ts';
 import type { VercelPgDatabase } from './vercel-postgres/index.ts';
 
@@ -31,7 +31,7 @@ export async function migrate(
 		| PostgresJsDatabase<any>
 		| VercelPgDatabase<any>
 		| TiDBServerlessDatabase<any>
-		| SingleStore2Database<any>,
+		| SingleStoreDriverDatabase<any>,
 	config:
 		| string
 		| MigrationConfig,
@@ -102,10 +102,10 @@ export async function migrate(
 
 			return migrate(db as VercelPgDatabase, config as string | MigrationConfig);
 		}
-		case 'SingleStore2Database': {
+		case 'SingleStoreDriverDatabase': {
 			const { migrate } = await import('./singlestore/migrator');
 
-			return migrate(db as SingleStore2Database, config as MigrationConfig);
+			return migrate(db as SingleStoreDriverDatabase, config as MigrationConfig);
 		}
 	}
 }

drizzle-orm/src/singlestore-core/db.ts

@@ -24,7 +24,6 @@ import type { OptimizeTableArgument } from './query-builders/optimizeTable.types
 import { RelationalQueryBuilder } from './query-builders/query.ts';
 import type { SelectedFields } from './query-builders/select.types.ts';
 import type {
-	Mode,
 	PreparedQueryHKTBase,
 	SingleStoreQueryResultHKT,
 	SingleStoreQueryResultKind,
@@ -61,7 +60,6 @@ export class SingleStoreDatabase<
 		/** @internal */
 		readonly session: SingleStoreSession<any, any, any, any>,
 		schema: RelationalSchemaConfig<TSchema> | undefined,
-		protected readonly mode: Mode,
 	) {
 		this._ = schema
 			? {
@@ -86,7 +84,6 @@ export class SingleStoreDatabase<
 						columns,
 						dialect,
 						session,
-						this.mode,
 					);
 			}
 		}