@powersync/service-module-mysql 0.7.4 → 0.9.0

package/test/src/util.ts

@@ -1,11 +1,16 @@
 import * as types from '@module/types/types.js';
-import { getMySQLVersion, isVersionAtLeast } from '@module/utils/mysql-utils.js';
+import { createRandomServerId, getMySQLVersion, isVersionAtLeast } from '@module/utils/mysql-utils.js';
 import * as mongo_storage from '@powersync/service-module-mongodb-storage';
 import * as postgres_storage from '@powersync/service-module-postgres-storage';
 import mysqlPromise from 'mysql2/promise';
 import { env } from './env.js';
 import { describe, TestOptions } from 'vitest';
 import { TestStorageFactory } from '@powersync/service-core';
+import { MySQLConnectionManager } from '@module/replication/MySQLConnectionManager.js';
+import { BinLogEventHandler, BinLogListener, Row, SchemaChange } from '@module/replication/zongji/BinLogListener.js';
+import { TableMapEntry } from '@powersync/mysql-zongji';
+import * as common from '@module/common/common-index.js';
+import { TablePattern } from '@powersync/service-sync-rules';
 
 export const TEST_URI = env.MYSQL_TEST_URI;
 
@@ -52,3 +57,84 @@ export async function clearTestDb(connection: mysqlPromise.Connection) {
     }
   }
 }
+
+export async function createTestDb(connectionManager: MySQLConnectionManager, dbName: string) {
+  await connectionManager.query(`DROP DATABASE IF EXISTS ${dbName}`);
+  await connectionManager.query(`CREATE DATABASE IF NOT EXISTS ${dbName}`);
+}
+
+export async function getFromGTID(connectionManager: MySQLConnectionManager) {
+  const connection = await connectionManager.getConnection();
+  const fromGTID = await common.readExecutedGtid(connection);
+  connection.release();
+
+  return fromGTID;
+}
+
+export interface CreateBinlogListenerParams {
+  connectionManager: MySQLConnectionManager;
+  eventHandler: BinLogEventHandler;
+  sourceTables: TablePattern[];
+  startGTID?: common.ReplicatedGTID;
+}
+export async function createBinlogListener(params: CreateBinlogListenerParams): Promise<BinLogListener> {
+  let { connectionManager, eventHandler, sourceTables, startGTID } = params;
+
+  if (!startGTID) {
+    startGTID = await getFromGTID(connectionManager);
+  }
+
+  return new BinLogListener({
+    connectionManager: connectionManager,
+    eventHandler: eventHandler,
+    startGTID: startGTID!,
+    sourceTables: sourceTables,
+    serverId: createRandomServerId(1)
+  });
+}
+
+export class TestBinLogEventHandler implements BinLogEventHandler {
+  rowsWritten = 0;
+  rowsUpdated = 0;
+  rowsDeleted = 0;
+  commitCount = 0;
+  schemaChanges: SchemaChange[] = [];
+  lastKeepAlive: string | undefined;
+
+  unpause: ((value: void | PromiseLike<void>) => void) | undefined;
+  private pausedPromise: Promise<void> | undefined;
+
+  pause() {
+    this.pausedPromise = new Promise((resolve) => {
+      this.unpause = resolve;
+    });
+  }
+
+  async onWrite(rows: Row[], tableMap: TableMapEntry) {
+    if (this.pausedPromise) {
+      await this.pausedPromise;
+    }
+    this.rowsWritten = this.rowsWritten + rows.length;
+  }
+
+  async onUpdate(afterRows: Row[], beforeRows: Row[], tableMap: TableMapEntry) {
+    this.rowsUpdated = this.rowsUpdated + afterRows.length;
+  }
+
+  async onDelete(rows: Row[], tableMap: TableMapEntry) {
+    this.rowsDeleted = this.rowsDeleted + rows.length;
+  }
+
+  async onCommit(lsn: string) {
+    this.commitCount++;
+  }
+
+  async onSchemaChange(change: SchemaChange) {
+    this.schemaChanges.push(change);
+  }
+  async onTransactionStart(options: { timestamp: Date }) {}
+  async onRotate() {}
+  async onKeepAlive(lsn: string) {
+    this.lastKeepAlive = lsn;
+  }
+}