Mirrors/coredns
1
0
Fork 0
mirror of https://github.com/coredns/coredns.git synced 2025-10-31 10:13:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

middleware/proxy: silence gprc errors

Browse Source 
Add discard logger
This commit is contained in:
Miek Gieben
2017-06-02 14:39:15 +00:00
parent d684dedfd3
commit 702826c0b7
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
middleware/proxy/grpc_test.go
View File

@@ -3,6 +3,8 @@ package proxy
import (
"testing"
"time"
"google.golang.org/grpc/grpclog"
)
func pool() []*UpstreamHost {
@@ -17,6 +19,8 @@ func pool() []*UpstreamHost {
}
func TestStartupShutdown(t *testing.T) {
grpclog.SetLogger(discard{})
upstream := &staticUpstream{
from: ".",
Hosts: pool(),
@@ -28,7 +32,7 @@ func TestStartupShutdown(t *testing.T) {
g := newGrpcClient(nil, upstream)
upstream.ex = g
p := &Proxy{Trace: nil}
p := &Proxy{}
p.Upstreams = &[]Upstream{upstream}
err := g.OnStartup(p)
@@ -52,3 +56,13 @@ func TestStartupShutdown(t *testing.T) {
t.Errorf("Shutdown didn't remove conns, found %d", len(g.conns))
}
}
// discard is a Logger that outputs nothing.
type discard struct{}
func (d discard) Fatal(args ...interface{}) {}
func (d discard) Fatalf(format string, args ...interface{}) {}
func (d discard) Fatalln(args ...interface{}) {}
func (d discard) Print(args ...interface{}) {}
func (d discard) Printf(format string, args ...interface{}) {}
func (d discard) Println(args ...interface{}) {}