mirror of
https://github.com/coredns/coredns.git
synced 2025-10-29 17:24:20 -04:00
34 lines
698 B
Go
34 lines
698 B
Go
|
package trace
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/mholt/caddy"
|
||
|
|
)
|
||
|
|
|
||
|
|
// CreateTestTrace creates a trace middleware to be used in tests
|
||
|
|
func CreateTestTrace(config string) (*caddy.Controller, *trace, error) {
|
||
|
|
c := caddy.NewTestController("dns", config)
|
||
|
|
m, err := traceParse(c)
|
||
|
|
return c, m, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestTrace(t *testing.T) {
|
||
|
|
_, m, err := CreateTestTrace(`trace`)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Error parsing test input: %s", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if m.Name() != "trace" {
|
||
|
|
t.Errorf("Wrong name from GetName: %s", m.Name())
|
||
|
|
}
|
||
|
|
err = m.OnStartup()
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Error starting tracing middleware: %s", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if m.Tracer() == nil {
|
||
|
|
t.Errorf("Error, no tracer created")
|
||
|
|
}
|
||
|
|
}
|