Skip to content

Commit

Permalink
Refactored the storage adapter tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@kraenhansen
kraenhansen committed Sep 6, 2024
1 parent 7c486df commit 62831d3
Showing 1 changed file with 26 additions and 62 deletions.
88 changes: 26 additions & 62 deletions packages/automerge-repo/src/helpers/tests/storage-adapter-tests.ts
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import { describe, expect, it } from "vitest"
import { describe, expect, beforeEach, test } from "vitest"

import type { StorageAdapterInterface } from "../../storage/StorageAdapterInterface.js"

Expand All @@ -8,72 +8,58 @@ const PAYLOAD_C = () => new Uint8Array([2, 111, 74, 131, 236, 96, 142, 193])

const LARGE_PAYLOAD = new Uint8Array(100000).map(() => Math.random() * 256)

export function runStorageAdapterTests(_setup: SetupFn, title?: string): void {
const setup = async () => {
const { adapter, teardown = NO_OP } = await _setup()
return { adapter, teardown }
}
type AdapterTestContext = {
adapter: StorageAdapterInterface;
};

const it = test<AdapterTestContext>

export function runStorageAdapterTests(setup: SetupFn, title?: string): void {

beforeEach<AdapterTestContext>(async (ctx) => {
const { adapter, teardown = NO_OP } = await setup()
ctx.adapter = adapter
return teardown
})

describe(`Storage adapter acceptance tests ${
title ? `(${title})` : ""
}`, () => {
describe("load", () => {
it("should return undefined if there is no data", async () => {
const { adapter, teardown } = await setup()

it("should return undefined if there is no data", async ({ adapter }) => {
const actual = await adapter.load(["AAAAA", "sync-state", "xxxxx"])
expect(actual).toBeUndefined()

teardown()
})
})

describe("save and load", () => {
it("should return data that was saved", async () => {
const { adapter, teardown } = await setup()

it("should return data that was saved", async ({ adapter }) => {
await adapter.save(["storage-adapter-id"], PAYLOAD_A())
const actual = await adapter.load(["storage-adapter-id"])
expect(actual).toStrictEqual(PAYLOAD_A())

teardown()
})

it("should work with composite keys", async () => {
const { adapter, teardown } = await setup()

it("should work with composite keys", async ({ adapter }) => {
await adapter.save(["AAAAA", "sync-state", "xxxxx"], PAYLOAD_A())
const actual = await adapter.load(["AAAAA", "sync-state", "xxxxx"])
expect(actual).toStrictEqual(PAYLOAD_A())

teardown()
})

it("should work with a large payload", async () => {
const { adapter, teardown } = await setup()

it("should work with a large payload", async ({ adapter }) => {
await adapter.save(["AAAAA", "sync-state", "xxxxx"], LARGE_PAYLOAD)
const actual = await adapter.load(["AAAAA", "sync-state", "xxxxx"])
expect(actual).toStrictEqual(LARGE_PAYLOAD)

teardown()
})
})

describe("loadRange", () => {
it("should return an empty array if there is no data", async () => {
const { adapter, teardown } = await setup()

it("should return an empty array if there is no data", async ({ adapter }) => {
expect(await adapter.loadRange(["AAAAA"])).toStrictEqual([])

teardown()
})
})

describe("save and loadRange", () => {
it("should return all the data that matches the key", async () => {
const { adapter, teardown } = await setup()

it("should return all the data that matches the key", async ({ adapter }) => {
await adapter.save(["AAAAA", "sync-state", "xxxxx"], PAYLOAD_A())
await adapter.save(["AAAAA", "snapshot", "yyyyy"], PAYLOAD_B())
await adapter.save(["AAAAA", "sync-state", "zzzzz"], PAYLOAD_C())
Expand All @@ -92,13 +78,9 @@ export function runStorageAdapterTests(_setup: SetupFn, title?: string): void {
{ key: ["AAAAA", "sync-state", "zzzzz"], data: PAYLOAD_C() },
])
)

teardown()
})

it("should only load values that match they key", async () => {
const { adapter, teardown } = await setup()

it("should only load values that match they key", async ({ adapter }) => {
await adapter.save(["AAAAA", "sync-state", "xxxxx"], PAYLOAD_A())
await adapter.save(["BBBBB", "sync-state", "zzzzz"], PAYLOAD_C())

Expand All @@ -113,46 +95,34 @@ export function runStorageAdapterTests(_setup: SetupFn, title?: string): void {
{ key: ["BBBBB", "sync-state", "zzzzz"], data: PAYLOAD_C() },
])
)

teardown()
})
})

describe("save and remove", () => {
it("after removing, should be empty", async () => {
const { adapter, teardown } = await setup()

it("after removing, should be empty", async ({ adapter }) => {
await adapter.save(["AAAAA", "snapshot", "xxxxx"], PAYLOAD_A())
await adapter.remove(["AAAAA", "snapshot", "xxxxx"])

expect(await adapter.loadRange(["AAAAA"])).toStrictEqual([])
expect(
await adapter.load(["AAAAA", "snapshot", "xxxxx"])
).toBeUndefined()

teardown()
})
})

describe("save and save", () => {
it("should overwrite data saved with the same key", async () => {
const { adapter, teardown } = await setup()

it("should overwrite data saved with the same key", async ({ adapter }) => {
await adapter.save(["AAAAA", "sync-state", "xxxxx"], PAYLOAD_A())
await adapter.save(["AAAAA", "sync-state", "xxxxx"], PAYLOAD_B())

expect(await adapter.loadRange(["AAAAA", "sync-state"])).toStrictEqual([
{ key: ["AAAAA", "sync-state", "xxxxx"], data: PAYLOAD_B() },
])

teardown()
})
})

describe("removeRange", () => {
it("should remove a range of records", async () => {
const { adapter, teardown } = await setup()

it("should remove a range of records", async ({ adapter }) => {
await adapter.save(["AAAAA", "sync-state", "xxxxx"], PAYLOAD_A())
await adapter.save(["AAAAA", "snapshot", "yyyyy"], PAYLOAD_B())
await adapter.save(["AAAAA", "sync-state", "zzzzz"], PAYLOAD_C())
Expand All @@ -162,13 +132,9 @@ export function runStorageAdapterTests(_setup: SetupFn, title?: string): void {
expect(await adapter.loadRange(["AAAAA"])).toStrictEqual([
{ key: ["AAAAA", "snapshot", "yyyyy"], data: PAYLOAD_B() },
])

teardown()
})

it("should not remove records that don't match", async () => {
const { adapter, teardown } = await setup()

it("should not remove records that don't match", async ({ adapter }) => {
await adapter.save(["AAAAA", "sync-state", "xxxxx"], PAYLOAD_A())
await adapter.save(["BBBBB", "sync-state", "zzzzz"], PAYLOAD_B())

Expand All @@ -178,8 +144,6 @@ export function runStorageAdapterTests(_setup: SetupFn, title?: string): void {
expect(actual).toStrictEqual([
{ key: ["BBBBB", "sync-state", "zzzzz"], data: PAYLOAD_B() },
])

teardown()
})
})
})
Expand All @@ -189,5 +153,5 @@ const NO_OP = () => {}

export type SetupFn = () => Promise<{
adapter: StorageAdapterInterface
teardown?: () => void
teardown?: () => void | Promise<void>
}>

0 comments on commit 62831d3

Please sign in to comment.