fix next branch following rebase.

Author: igalklebanov

package.json

@@ -110,6 +110,7 @@
   "devDependencies": {
     "@arethetypeswrong/cli": "^0.18.2",
     "@ark/attest": "^0.55.0",
+    "@electric-sql/pglite": "^0.3.14",
     "@types/better-sqlite3": "^7.6.13",
     "@types/chai": "^5.2.3",
     "@types/chai-as-promised": "^8.0.2",

pnpm-lock.yaml

@@ -1,7 +1,9 @@
+import type { DialectAdapter } from '../dialect/dialect-adapter.js'
 import type { CompiledQuery } from '../query-compiler/compiled-query.js'
 import type { QueryCompiler } from '../query-compiler/query-compiler.js'
 import type { Log } from '../util/log.js'
 import { performanceNow } from '../util/performance-now.js'
+import { ConnectionMutex } from './connection-mutex.js'
 import type { DatabaseConnection, QueryResult } from './database-connection.js'
 import type { Driver, TransactionSettings } from './driver.js'
 

src/driver/runtime-driver.ts

@@ -4,6 +4,7 @@ import type { TableNode } from './table-node.js'
 import type { ConstraintNode } from './constraint-node.js'
 import type { ColumnDefinitionNode } from './column-definition-node.js'
 import type { ArrayItemType } from '../util/type-utils.js'
+import type { AddIndexNode } from './add-index-node.js'
 
 export const ON_COMMIT_ACTIONS = ['preserve rows', 'delete rows', 'drop']
 export type OnCommitAction = ArrayItemType<typeof ON_COMMIT_ACTIONS>

src/operation-node/create-table-node.ts

@@ -1,7 +1,7 @@
-import { QueryResult } from '../../driver/database-connection.js'
-import { RootOperationNode } from '../../query-compiler/query-compiler.js'
-import { UnknownRow } from '../../util/type-utils.js'
-import {
+import type { QueryResult } from '../../driver/database-connection.js'
+import type { RootOperationNode } from '../../query-compiler/query-compiler.js'
+import type { UnknownRow } from '../../util/type-utils.js'
+import type {
   KyselyPlugin,
   PluginTransformQueryArgs,
   PluginTransformResultArgs,

src/plugin/safe-null-comparison/safe-null-comparison-plugin.ts

@@ -1,12 +1,17 @@
 import type { Expression } from '../expression/expression.js'
 import { ExpressionWrapper } from '../expression/expression-wrapper.js'
 import { freeze } from '../util/object-utils.js'
-import type { ReferenceExpression } from '../parser/reference-parser.js'
+import {
+  parseReferenceExpression,
+  type ExtractTypeFromReferenceExpression,
+  type ReferenceExpression,
+} from '../parser/reference-parser.js'
 import { CaseNode } from '../operation-node/case-node.js'
 import { WhenNode } from '../operation-node/when-node.js'
 import {
   type ComparisonOperatorExpression,
   type OperandValueExpressionOrList,
+  parseReferentialBinaryOperation,
   parseValueBinaryOperationOrExpression,
 } from '../parser/binary-operation-parser.js'
 import {
@@ -222,7 +227,7 @@ export class CaseWhenBuilder<DB, TB extends keyof DB, W, O>
    * Adds an `else` clause to the `case` statement where the value is a reference to a column.
    *
    * See {@link else} for value-first variant.
-   * 
+   *
    * An `elseRef` call must be followed by an {@link Endable.end} or {@link Endable.endCase} call.
    */
   elseRef<RE extends ReferenceExpression<DB, TB>>(

src/query-builder/case-builder.ts

@@ -7,7 +7,7 @@ import type { OperationNodeSource } from '../operation-node/operation-node-sourc
 import { RenameColumnNode } from '../operation-node/rename-column-node.js'
 import type { CompiledQuery } from '../query-compiler/compiled-query.js'
 import type { Compilable } from '../util/compilable.js'
-import { freeze, noop } from '../util/object-utils.js'
+import { freeze, isString, noop } from '../util/object-utils.js'
 import {
   ColumnDefinitionBuilder,
   type ColumnDefinitionBuilderCallback,
@@ -56,7 +56,7 @@ import {
 } from './check-constraint-builder.js'
 import { RenameConstraintNode } from '../operation-node/rename-constraint-node.js'
 import {
-  ExpressionOrFactory,
+  type ExpressionOrFactory,
   parseExpression,
 } from '../parser/expression-parser.js'
 

src/schema/alter-table-builder.ts

@@ -1,6 +1,6 @@
 import { AddIndexNode } from '../operation-node/add-index-node.js'
-import { IndexType } from '../operation-node/create-index-node.js'
-import { OperationNodeSource } from '../operation-node/operation-node-source.js'
+import type { IndexType } from '../operation-node/create-index-node.js'
+import type { OperationNodeSource } from '../operation-node/operation-node-source.js'
 import { RawNode } from '../operation-node/raw-node.js'
 
 export class CreateTableAddIndexBuilder implements OperationNodeSource {

src/schema/create-table-add-index-builder.ts

@@ -9,7 +9,7 @@ import type { Compilable } from '../util/compilable.js'
 import type { QueryExecutor } from '../query-executor/query-executor.js'
 import { ColumnDefinitionBuilder } from './column-definition-builder.js'
 import type { QueryId } from '../util/query-id.js'
-import { freeze, noop } from '../util/object-utils.js'
+import { freeze, isString, noop } from '../util/object-utils.js'
 import { ForeignKeyConstraintNode } from '../operation-node/foreign-key-constraint-node.js'
 import { ColumnNode } from '../operation-node/column-node.js'
 import {
@@ -31,7 +31,7 @@ import {
   type UniqueConstraintNodeBuilderCallback,
 } from './unique-constraint-builder.js'
 import {
-  ExpressionOrFactory,
+  type ExpressionOrFactory,
   parseExpression,
 } from '../parser/expression-parser.js'
 import {
@@ -45,7 +45,7 @@ import {
 import { AddIndexNode } from '../operation-node/add-index-node.js'
 import {
   CreateTableAddIndexBuilder,
-  CreateTableAddIndexBuilderCallback,
+  type CreateTableAddIndexBuilderCallback,
 } from './create-table-add-index-builder.js'
 
 /**

src/schema/create-table-builder.ts

@@ -337,9 +337,8 @@ for (const dialect of DIALECTS) {
 
         await query.execute()
       })
-    })
 
-      if (dialect === 'postgres') {
+      if (sqlSpec === 'postgres') {
         it('should also add schema to using clause', async () => {
           const query = ctx.db
             .withSchema('mammals')
@@ -363,12 +362,13 @@ for (const dialect of DIALECTS) {
       }
     })
 
-      describe('delete from', () => {
-        it('should add schema', async () => {
-          const query = ctx.db
-            .withSchema('mammals')
-            .deleteFrom('pet')
-            .where('pet.name', '=', 'Doggo')
+    describe('update', () => {
+      it('should add schema', async () => {
+        const query = ctx.db
+          .withSchema('mammals')
+          .updateTable('pet')
+          .where('pet.name', '=', 'Doggo')
+          .set({ species: 'cat' })
 
         testSql(query, dialect, {
           postgres: {