package dbx

import (
	"context"
	"testing"
)

func TestInjectExtractQuerier(t *testing.T) {
	ctx := context.Background()
	fallback := &Node{name: "fallback"}

	// No querier in context → returns fallback
	got := ExtractQuerier(ctx, fallback)
	if got != fallback {
		t.Error("expected fallback when no querier in context")
	}

	// Inject querier → extract it
	injected := &Node{name: "injected"}
	ctx = InjectQuerier(ctx, injected)
	got = ExtractQuerier(ctx, fallback)
	if got != injected {
		t.Error("expected injected querier")
	}
}