mirror of
https://github.com/coredns/coredns.git
synced 2025-11-02 02:03:13 -05:00
99800a687c
This revert 17d807f0 and re-adds the metadata plugin as a plugin that
just sets a label to a value function.
Add package documentation on how to use the metadata package. Make it
clear that any caching is up to the Func implemented.
There are now - no in tree users. We could add the request metadata by
default under names that copy request.Request, i.e
request/ip - remote IP
request/port - remote port
Variables.go has been deleted.
Signed-off-by: Miek Gieben <miek@miek.nl>
62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
package metadata
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/coredns/coredns/plugin/test"
|
|
"github.com/coredns/coredns/request"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
type testProvider map[string]Func
|
|
|
|
func (tp testProvider) Metadata(ctx context.Context, state request.Request) context.Context {
|
|
for k, v := range tp {
|
|
SetValueFunc(ctx, k, v)
|
|
}
|
|
return ctx
|
|
}
|
|
|
|
type testHandler struct{ ctx context.Context }
|
|
|
|
func (m *testHandler) Name() string { return "test" }
|
|
|
|
func (m *testHandler) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
|
|
m.ctx = ctx
|
|
return 0, nil
|
|
}
|
|
|
|
func TestMetadataServeDNS(t *testing.T) {
|
|
expectedMetadata := []testProvider{
|
|
testProvider{"test/key1": func() string { return "testvalue1" }},
|
|
testProvider{"test/key2": func() string { return "two" }, "test/key3": func() string { return "testvalue3" }},
|
|
}
|
|
// Create fake Providers based on expectedMetadata
|
|
providers := []Provider{}
|
|
for _, e := range expectedMetadata {
|
|
providers = append(providers, e)
|
|
}
|
|
|
|
next := &testHandler{} // fake handler which stores the resulting context
|
|
m := Metadata{
|
|
Zones: []string{"."},
|
|
Providers: providers,
|
|
Next: next,
|
|
}
|
|
|
|
ctx := context.TODO()
|
|
m.ServeDNS(ctx, &test.ResponseWriter{}, new(dns.Msg))
|
|
nctx := next.ctx
|
|
|
|
for _, expected := range expectedMetadata {
|
|
for label, expVal := range expected {
|
|
val := ValueFunc(nctx, label)
|
|
if val() != expVal() {
|
|
t.Errorf("Expected value %s for %s, but got %s", expVal(), label, val())
|
|
}
|
|
}
|
|
}
|
|
}
|