zopu-code/repos/effect-smol/packages/effect/test/unstable/persistence/KeyValueStore.test.ts

import { afterEach, describe, it } from "@effect/vitest"
import { assertTrue, deepStrictEqual, strictEqual } from "@effect/vitest/utils"
import type { Layer } from "effect"
import { Effect, Option, Schema } from "effect"
import * as KeyValueStore from "effect/unstable/persistence/KeyValueStore"

export const testLayer = <E>(layer: Layer.Layer<KeyValueStore.KeyValueStore, E>) => {
  const run = <E, A>(effect: Effect.Effect<A, E, KeyValueStore.KeyValueStore>) =>
    Effect.runPromise(Effect.provide(effect, layer))

  afterEach(() =>
    run(Effect.gen(function*() {
      const kv = yield* (KeyValueStore.KeyValueStore)
      yield* kv.clear
    }))
  )

  it("set", () =>
    run(Effect.gen(function*() {
      const kv = yield* (KeyValueStore.KeyValueStore)
      yield* (kv.set("/foo/bar", "bar"))

      const value = yield* (kv.get("/foo/bar"))
      const length = yield* (kv.size)

      strictEqual(value, "bar")
      strictEqual(length, 1)
    })))

  it("get/ missing", () =>
    run(Effect.gen(function*() {
      const kv = yield* (KeyValueStore.KeyValueStore)
      yield* (kv.clear)
      const value = yield* (kv.get("foo"))

      strictEqual(value, undefined)
    })))

  it("remove", () =>
    run(Effect.gen(function*() {
      const kv = yield* (KeyValueStore.KeyValueStore)
      yield* (kv.set("foo", "bar"))
      yield* (kv.remove("foo"))

      const value = yield* (kv.get("foo"))
      const length = yield* (kv.size)

      strictEqual(value, undefined)
      strictEqual(length, 0)
    })))

  it("clear", () =>
    run(Effect.gen(function*() {
      const kv = yield* (KeyValueStore.KeyValueStore)
      yield* (kv.set("foo", "bar"))
      yield* (kv.clear)

      const value = yield* (kv.get("foo"))
      const length = yield* (kv.size)

      strictEqual(value, undefined)
      strictEqual(length, 0)
    })))

  it("modify", () =>
    run(Effect.gen(function*() {
      const kv = yield* (KeyValueStore.KeyValueStore)
      yield* (kv.set("foo", "bar"))

      const value = yield* (kv.modify("foo", (v) => v + "bar"))
      const length = yield* (kv.size)

      strictEqual(value, "barbar")
      strictEqual(length, 1)
    })))

  it("modify - none", () =>
    run(Effect.gen(function*() {
      const kv = yield* (KeyValueStore.KeyValueStore)

      const value = yield* (kv.modify("foo", (v) => v + "bar"))
      const length = yield* (kv.size)

      strictEqual(value, undefined)
      strictEqual(length, 0)
    })))
}

describe("KeyValueStore / layerMemory", () => testLayer(KeyValueStore.layerMemory))

describe("KeyValueStore / prefix", () => {
  it("prefixes the keys", () =>
    Effect.gen(function*() {
      const store = yield* (KeyValueStore.KeyValueStore)
      const prefixed = KeyValueStore.prefix(store, "prefix/")

      yield* (prefixed.set("foo", "bar"))
      yield* (prefixed.modify("foo", (v) => v + "bar"))

      strictEqual(yield* (prefixed.get("foo")), "barbar")
      assertTrue(yield* (prefixed.has("foo")))

      strictEqual(yield* (store.get("prefix/foo")), "barbar")
      assertTrue(yield* (store.has("prefix/foo")))
    }).pipe(
      Effect.provide(KeyValueStore.layerMemory),
      Effect.runPromise
    ))
})

describe("toSchemaStore", () => {
  class User extends Schema.Class<User>("User")({
    name: Schema.String,
    age: Schema.Number
  }) {}

  it("encodes & decodes", () =>
    Effect.gen(function*() {
      const store = yield* KeyValueStore.KeyValueStore
      const schemaStore = KeyValueStore.toSchemaStore(store, User)
      yield* (schemaStore.set("foo", new User({ name: "foo", age: 42 })))
      yield* (schemaStore.modify("foo", (user) => new User({ ...user, age: 43 })))
      const value = yield* schemaStore.get("foo")

      assertTrue(Option.isSome(value))
      strictEqual(value.value.name, "foo")
      strictEqual(value.value.age, 43)
    }).pipe(
      Effect.provide(KeyValueStore.layerMemory),
      Effect.runPromise
    ))

  it("prefix", () =>
    Effect.gen(function*() {
      const store = yield* KeyValueStore.KeyValueStore
      const schemaStore = KeyValueStore.toSchemaStore(store, User)
      yield* (schemaStore.set("prefix/foo", new User({ name: "foo", age: 42 })))
      const value = yield* schemaStore.get("prefix/foo")

      if (Option.isSome(value)) {
        strictEqual(value.value.name, "foo")
        strictEqual(value.value.age, 42)
      }
    }).pipe(
      Effect.provide(KeyValueStore.layerMemory),
      Effect.runPromise
    ))

  it("json compliant", () =>
    Effect.gen(function*() {
      const store = yield* KeyValueStore.KeyValueStore
      const schema = Schema.Struct({
        a: Schema.Date
      })
      const schemaStore = KeyValueStore.toSchemaStore(store, schema)
      yield* (schemaStore.set("foo", { a: new Date(0) }))
      const value = yield* schemaStore.get("foo")
      assertTrue(Option.isSome(value))
      deepStrictEqual(value.value.a, new Date(0))
    }).pipe(
      Effect.provide(KeyValueStore.layerMemory),
      Effect.runPromise
    ))
})