Merge commit '36c66dd290d3ce6eb1ccd310d0c658d4a32bb8eb' as 'repos/effect'

repos/effect/packages/sql/libsql/test/Client.test.ts

@@ -0,0 +1,106 @@
+import { LibsqlClient } from "@effect/sql-libsql"
+import { assert, describe, layer } from "@effect/vitest"
+import { Effect, Layer } from "effect"
+import { LibsqlContainer } from "./util.ts"
+
+const Migrations = Layer.effectDiscard(
+  LibsqlClient.LibsqlClient.pipe(
+    Effect.andThen((sql) =>
+      Effect.acquireRelease(
+        sql`CREATE TABLE test (id INTEGER PRIMARY KEY, name TEXT)`,
+        () => sql`DROP TABLE test;`.pipe(Effect.ignore)
+      )
+    )
+  )
+)
+
+describe("Client", () => {
+  layer(LibsqlContainer.layerClient, { timeout: "30 seconds" })((it) => {
+    it.effect("should work", () =>
+      Effect.gen(function*() {
+        const sql = yield* LibsqlClient.LibsqlClient
+        let response = yield* sql`INSERT INTO test (name) VALUES ('hello')`
+        assert.deepStrictEqual(response, [])
+        response = yield* sql`SELECT * FROM test`
+        assert.deepStrictEqual(response, [{ id: 1, name: "hello" }])
+        response = yield* sql`SELECT * FROM test`
+        assert.deepStrictEqual(yield* sql`select * from test`.values, [
+          [1, "hello"]
+        ])
+      }).pipe(Effect.provide(Migrations)))
+
+    it.effect("should work with raw", () =>
+      Effect.gen(function*() {
+        const sql = yield* LibsqlClient.LibsqlClient
+        let response: any
+        response = yield* sql`CREATE TABLE test2 (id INTEGER PRIMARY KEY, name TEXT)`.raw
+        yield* Effect.addFinalizer(() => sql`DROP TABLE test2;`.pipe(Effect.ignore))
+        assert.deepStrictEqual(response.toJSON(), {
+          columnTypes: [],
+          columns: [],
+          lastInsertRowid: null,
+          rows: [],
+          rowsAffected: 0
+        })
+        response = yield* sql`INSERT INTO test (name) VALUES ('hello')`.raw
+        assert.deepStrictEqual(response.toJSON(), {
+          columnTypes: [],
+          columns: [],
+          lastInsertRowid: "1",
+          rows: [],
+          rowsAffected: 1
+        })
+        response = yield* sql`SELECT * FROM test`.raw
+        assert.deepStrictEqual(response.toJSON(), {
+          columnTypes: ["INTEGER", "TEXT"],
+          columns: ["id", "name"],
+          lastInsertRowid: null,
+          rows: [[1, "hello"]],
+          rowsAffected: 0
+        })
+      }).pipe(Effect.provide(Migrations)))
+
+    it.effect("withTransaction", () =>
+      Effect.gen(function*() {
+        const sql = yield* LibsqlClient.LibsqlClient
+        yield* sql.withTransaction(sql`INSERT INTO test (name) VALUES ('hello')`)
+        const rows = yield* sql`SELECT * FROM test`
+        assert.deepStrictEqual(rows, [{ id: 1, name: "hello" }])
+      }).pipe(Effect.provide(Migrations)))
+
+    it.effect("withTransaction rollback", () =>
+      Effect.gen(function*() {
+        const sql = yield* LibsqlClient.LibsqlClient
+        yield* sql`INSERT INTO test (name) VALUES ('hello')`.pipe(
+          Effect.andThen(Effect.fail("boom")),
+          sql.withTransaction,
+          Effect.ignore
+        )
+        const rows = yield* sql`SELECT * FROM test`
+        assert.deepStrictEqual(rows, [])
+      }).pipe(Effect.provide(Migrations)))
+
+    it.effect("withTransaction nested", () =>
+      Effect.gen(function*() {
+        const sql = yield* LibsqlClient.LibsqlClient
+        const stmt = sql`INSERT INTO test (name) VALUES ('hello')`
+
+        yield* stmt.pipe(Effect.andThen(() => stmt.pipe(sql.withTransaction)), sql.withTransaction)
+        const rows = yield* sql<{ total_rows: number }>`select count(*) as total_rows FROM test`
+        assert.deepStrictEqual(rows.at(0)?.total_rows, 2)
+      }).pipe(Effect.provide(Migrations)))
+
+    it.effect("withTransaction nested rollback", () =>
+      Effect.gen(function*() {
+        const sql = yield* LibsqlClient.LibsqlClient
+        const stmt = sql`INSERT INTO test (name) VALUES ('hello')`
+
+        yield* stmt.pipe(
+          Effect.andThen(() => stmt.pipe(Effect.andThen(Effect.fail("boom")), sql.withTransaction, Effect.ignore)),
+          sql.withTransaction
+        )
+        const rows = yield* sql<{ total_rows: number }>`select count(*) as total_rows FROM test`
+        assert.deepStrictEqual(rows.at(0)?.total_rows, 1)
+      }).pipe(Effect.provide(Migrations)))
+  })
+})