package store

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

func TestMemorySnapshotMethodsRespectCancellation(t *testing.T) {
	m := NewMemory()
	m.SeedDemo()
	tenant, _, module, _, _ := m.DemoIDs()
	ctx, cancel := context.WithCancel(context.Background())
	cancel()
	if snap, ok, err := m.GetModulePrefixSnapshot(ctx, tenant, module); snap != nil || ok || !errors.Is(err, context.Canceled) {
		t.Fatalf("get: %+v %v %v", snap, ok, err)
	}
	if err := m.SetModulePrefixSnapshot(ctx, tenant, module, "hash", []PrefixRow{{Prefix: "192.0.2.0/24"}}); !errors.Is(err, context.Canceled) {
		t.Fatalf("set: %v", err)
	}
	if err := m.DeleteModulePrefixSnapshot(ctx, tenant, module); !errors.Is(err, context.Canceled) {
		t.Fatalf("delete: %v", err)
	}
	if err := m.SetModuleInputHash(ctx, tenant, module, "hash"); !errors.Is(err, context.Canceled) {
		t.Fatalf("input hash: %v", err)
	}
	if rows, err := m.ListModuleSnapshotSourceRows(ctx, tenant, module, "ip_range"); rows != nil || !errors.Is(err, context.Canceled) {
		t.Fatalf("source rows: %+v %v", rows, err)
	}
	if snap, ok, err := m.GetModulePrefixSnapshot(context.Background(), tenant, module); snap != nil || ok || err != nil {
		t.Fatalf("cancelled write committed: %+v %v %v", snap, ok, err)
	}
}