26 lines
539 B
Go
26 lines
539 B
Go
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")
|
|
}
|
|
}
|