feat: add durable client command spool
This commit is contained in:
@@ -0,0 +1,182 @@
|
||||
package spool
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/sha256"
|
||||
"errors"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/rvbox/rvbox/internal/domain"
|
||||
)
|
||||
|
||||
func TestStoreDurableIdentityAndSingleInstance_HP_CLIENT_01(t *testing.T) {
|
||||
t.Parallel()
|
||||
ctx := context.Background()
|
||||
directory := filepath.Join(t.TempDir(), "spool")
|
||||
store := openTestStore(t, ctx, directory, DefaultTombstoneLimit)
|
||||
identity := store.ClientInstanceID()
|
||||
if identity == (domain.UUID{}) {
|
||||
t.Fatal("client instance ID is zero")
|
||||
}
|
||||
if _, err := Open(ctx, Options{DataDir: directory, BusyTimeout: time.Second}); !errors.Is(err, ErrAlreadyOpen) {
|
||||
t.Fatalf("second Open error = %v, want ErrAlreadyOpen", err)
|
||||
}
|
||||
if err := store.Close(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
reopened := openTestStore(t, ctx, directory, DefaultTombstoneLimit)
|
||||
if got := reopened.ClientInstanceID(); got != identity {
|
||||
t.Fatalf("reopened client instance ID = %s, want %s", got, identity)
|
||||
}
|
||||
}
|
||||
|
||||
func TestStoreRejectsCorruptIdentity_BH_CLIENT_01(t *testing.T) {
|
||||
t.Parallel()
|
||||
directory := t.TempDir()
|
||||
if err := os.Chmod(directory, 0o700); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := os.WriteFile(filepath.Join(directory, "client-instance-id"), []byte("not-an-id\n"), 0o600); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
_, err := Open(context.Background(), Options{DataDir: directory, BusyTimeout: time.Second})
|
||||
if !errors.Is(err, ErrIdentityCorrupt) {
|
||||
t.Fatalf("Open corrupt identity error = %v, want ErrIdentityCorrupt", err)
|
||||
}
|
||||
data, err := os.ReadFile(filepath.Join(directory, "client-instance-id"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if string(data) != "not-an-id\n" {
|
||||
t.Fatalf("corrupt identity was rewritten as %q", data)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSpoolAcceptanceSequencingAndAck_HP_CLIENT_07(t *testing.T) {
|
||||
t.Parallel()
|
||||
ctx := context.Background()
|
||||
directory := filepath.Join(t.TempDir(), "spool")
|
||||
store := openTestStore(t, ctx, directory, DefaultTombstoneLimit)
|
||||
issue := testUUID(t, "019c46f1-1d02-7000-8000-000000000001")
|
||||
command := testCommand(issue, []byte("immutable command"))
|
||||
acceptedAt := time.Date(2026, time.September, 6, 12, 0, 0, 0, time.UTC)
|
||||
accepted, err := store.AcceptCommand(ctx, command, acceptedAt)
|
||||
if err != nil || accepted.Duplicate {
|
||||
t.Fatalf("first acceptance = %#v, %v", accepted, err)
|
||||
}
|
||||
duplicate, err := store.AcceptCommand(ctx, command, acceptedAt.Add(time.Second))
|
||||
if err != nil || !duplicate.Duplicate || duplicate.Command.Revision != command.Revision {
|
||||
t.Fatalf("duplicate acceptance = %#v, %v", duplicate, err)
|
||||
}
|
||||
first, err := store.AppendEvent(ctx, issue, EventInput{Kind: 1, Compression: 1, RawBytes: 5, Payload: []byte("first"), CreatedAt: acceptedAt})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
second, err := store.AppendEvent(ctx, issue, EventInput{Kind: 2, Compression: 1, RawBytes: 6, Payload: []byte("second"), CreatedAt: acceptedAt.Add(time.Nanosecond)})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if first.LocalOrdinal != 1 || first.EventSeq != 0 || second.LocalOrdinal != 2 || second.EventSeq != 0 {
|
||||
t.Fatalf("unassigned events = %#v, %#v", first, second)
|
||||
}
|
||||
assigned, err := store.AssignSendWindow(ctx, issue, 1, 1<<20)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(assigned) != 1 || assigned[0].LocalOrdinal != 1 || assigned[0].EventSeq != 1 {
|
||||
t.Fatalf("first send window = %#v", assigned)
|
||||
}
|
||||
assigned, err = store.AssignSendWindow(ctx, issue, 1, 1<<20)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(assigned) != 2 || assigned[1].LocalOrdinal != 2 || assigned[1].EventSeq != 2 {
|
||||
t.Fatalf("replayed send window = %#v", assigned)
|
||||
}
|
||||
if err := store.Ack(ctx, issue, 1); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
pending, err := store.PendingEvents(ctx, issue)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(pending) != 1 || pending[0].EventSeq != 2 || string(pending[0].Payload) != "second" {
|
||||
t.Fatalf("pending after cumulative ack = %#v", pending)
|
||||
}
|
||||
if err := store.Ack(ctx, issue, 3); !errors.Is(err, ErrInvalidEventAck) {
|
||||
t.Fatalf("ack beyond send window error = %v, want ErrInvalidEventAck", err)
|
||||
}
|
||||
if err := store.Close(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
store = openTestStore(t, ctx, directory, DefaultTombstoneLimit)
|
||||
pending, err = store.PendingEvents(ctx, issue)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(pending) != 1 || pending[0].EventSeq != 2 {
|
||||
t.Fatalf("pending after reopen = %#v", pending)
|
||||
}
|
||||
}
|
||||
|
||||
func TestTerminalCleanupTombstonesAndConflicts_BH_CLIENT_02(t *testing.T) {
|
||||
t.Parallel()
|
||||
ctx := context.Background()
|
||||
store := openTestStore(t, ctx, filepath.Join(t.TempDir(), "spool"), 1)
|
||||
first := testCommand(testUUID(t, "019c46f1-1d02-7000-8000-000000000011"), []byte("first"))
|
||||
second := testCommand(testUUID(t, "019c46f1-1d02-7000-8000-000000000012"), []byte("second"))
|
||||
now := time.Date(2026, time.September, 6, 12, 0, 0, 0, time.UTC)
|
||||
for _, command := range []Command{first, second} {
|
||||
if _, err := store.AcceptCommand(ctx, command, now); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := store.MarkTerminal(ctx, command.IssueUUID, 5); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := store.CleanupTerminal(ctx, command.IssueUUID, now); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
now = now.Add(time.Nanosecond)
|
||||
}
|
||||
if _, err := store.AcceptCommand(ctx, second, now); !errors.Is(err, ErrAlreadyExecuted) {
|
||||
t.Fatalf("exact tombstone replay error = %v, want ErrAlreadyExecuted", err)
|
||||
}
|
||||
conflict := second
|
||||
conflict.ImmutableSHA256 = sha256.Sum256([]byte("changed"))
|
||||
if _, err := store.AcceptCommand(ctx, conflict, now); !errors.Is(err, ErrCommandConflict) {
|
||||
t.Fatalf("conflicting tombstone replay error = %v, want ErrCommandConflict", err)
|
||||
}
|
||||
if _, err := store.AcceptCommand(ctx, first, now); err != nil {
|
||||
t.Fatalf("oldest tombstone was not rotated: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func openTestStore(t *testing.T, ctx context.Context, directory string, tombstoneLimit uint64) *Store {
|
||||
t.Helper()
|
||||
store, err := Open(ctx, Options{DataDir: directory, BusyTimeout: time.Second, TombstoneLimit: tombstoneLimit})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
t.Cleanup(func() {
|
||||
if err := store.Close(); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
})
|
||||
return store
|
||||
}
|
||||
|
||||
func testUUID(t *testing.T, value string) domain.UUID {
|
||||
t.Helper()
|
||||
parsed, err := domain.ParseUUIDv7(value)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
return parsed
|
||||
}
|
||||
|
||||
func testCommand(issue domain.UUID, immutable []byte) Command {
|
||||
return Command{IssueUUID: issue, ImmutableSHA256: sha256.Sum256(immutable), Revision: 1, Phase: 1}
|
||||
}
|
||||
Reference in New Issue
Block a user