package store

import (
	"context"
	"errors"
	"testing"
	"time"
)

func TestPruneRevisionsBeforeCancelledContext(t *testing.T) {
	m := NewMemory()
	m.SeedDemo()
	tenant, _, _, rev, _ := m.DemoIDs()
	ctx, cancel := context.WithCancel(context.Background())
	cancel()
	if _, err := m.PruneRevisionsBefore(ctx, tenant, time.Now().Add(time.Hour)); !errors.Is(err, context.Canceled) {
		t.Fatalf("err=%v", err)
	}
	if _, err := m.GetRevision(tenant, rev); err != nil {
		t.Fatal(err)
	}
}