119 lines · 4.4 KB · TypeScript
RawDownload
import * as NodeServices from "@effect/platform-node/NodeServices";import { assert, it } from "@effect/vitest";import * as Effect from "effect/Effect";import * as FileSystem from "effect/FileSystem";import * as Path from "effect/Path";import * as SqlClient from "effect/unstable/sql/SqlClient"; import * as NodeSqliteClient from "../src/persistence/NodeSqliteClient.ts";import { runSqliteState } from "./t3-sqlite-state.ts"; const createFixtureDatabase = Effect.fn("createSqliteStateFixtureDatabase")(function* (  baseDir: string,) {  const fs = yield* FileSystem.FileSystem;  const path = yield* Path.Path;  const stateDir = path.join(baseDir, "userdata");  const databasePath = path.join(stateDir, "state.sqlite");  yield* fs.makeDirectory(stateDir, { recursive: true });  yield* Effect.gen(function* () {    const sql = yield* SqlClient.SqlClient;    yield* sql`CREATE TABLE fixtures (id INTEGER PRIMARY KEY, label TEXT NOT NULL)`;    yield* sql`INSERT INTO fixtures (id, label) VALUES (1, 'existing')`;  }).pipe(Effect.provide(NodeSqliteClient.layer({ filename: databasePath })));}); it.layer(NodeServices.layer)("t3-sqlite-state", (it) => {  it.effect("reports each invalid SQL source with a specific error", () =>    Effect.gen(function* () {      const fs = yield* FileSystem.FileSystem;      const baseDir = yield* fs.makeTempDirectoryScoped({ prefix: "t3-sqlite-state-input-" });       const multipleSources = yield* runSqliteState({        operation: "query",        baseDir,        sql: "SELECT 1",        file: "fixture.sql",      }).pipe(Effect.flip);      assert.equal(multipleSources._tag, "SqliteStateMultipleSqlSourcesError");      assert.equal(multipleSources.message, "Provide only one of --sql or --file.");       const missingSource = yield* runSqliteState({ operation: "query", baseDir }).pipe(        Effect.flip,      );      assert.equal(missingSource._tag, "SqliteStateMissingSqlSourceError");      assert.equal(missingSource.message, "Provide one of --sql or --file.");       const emptySql = yield* runSqliteState({        operation: "query",        baseDir,        sql: "   ",      }).pipe(Effect.flip);      assert.equal(emptySql._tag, "SqliteStateEmptySqlError");      assert.equal(emptySql.message, "SQL input is empty.");    }),  );   it.effect("queries an isolated database through Effect SQL", () =>    Effect.gen(function* () {      const fs = yield* FileSystem.FileSystem;      const baseDir = yield* fs.makeTempDirectoryScoped({ prefix: "t3-sqlite-state-query-" });      yield* createFixtureDatabase(baseDir);       const result = yield* runSqliteState({        operation: "query",        baseDir,        sql: "SELECT id, label FROM fixtures",      });       assert.equal(result.operation, "query");      if (result.operation === "query") {        assert.deepStrictEqual(result.rows, [{ id: 1, label: "existing" }]);      }    }),  );   it.effect("backs up isolated state before writes and refuses the shared home", () =>    Effect.gen(function* () {      const fs = yield* FileSystem.FileSystem;      const path = yield* Path.Path;      const baseDir = yield* fs.makeTempDirectoryScoped({ prefix: "t3-sqlite-state-exec-" });      yield* createFixtureDatabase(baseDir);       const mutation = yield* runSqliteState({        operation: "exec",        baseDir,        sql: "INSERT INTO fixtures (id, label) VALUES (2, 'seeded')",      });      assert.equal(mutation.operation, "exec");      if (mutation.operation === "exec") {        assert.equal((yield* fs.stat(mutation.backup)).mode & 0o777, 0o600);      }       const error = yield* runSqliteState(        {          operation: "exec",          baseDir,          sql: "DELETE FROM fixtures",        },        { sharedHome: baseDir },      ).pipe(Effect.flip);      assert.equal(error._tag, "SqliteStateSharedHomeMutationError");       const aliasParent = yield* fs.makeTempDirectoryScoped({        prefix: "t3-sqlite-state-alias-",      });      const aliasBaseDir = path.join(aliasParent, "shared-home-alias");      yield* fs.symlink(baseDir, aliasBaseDir);      const aliasError = yield* runSqliteState(        {          operation: "exec",          baseDir: aliasBaseDir,          sql: "DELETE FROM fixtures",        },        { sharedHome: baseDir },      ).pipe(Effect.flip);      assert.equal(aliasError._tag, "SqliteStateSharedHomeMutationError");    }),  );});