aboutsummaryrefslogtreecommitdiff
path: root/middleware/proxy/grpc_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'middleware/proxy/grpc_test.go')
-rw-r--r--middleware/proxy/grpc_test.go71
1 files changed, 0 insertions, 71 deletions
diff --git a/middleware/proxy/grpc_test.go b/middleware/proxy/grpc_test.go
deleted file mode 100644
index dcde7cc0e..000000000
--- a/middleware/proxy/grpc_test.go
+++ /dev/null
@@ -1,71 +0,0 @@
-package proxy
-
-import (
- "testing"
- "time"
-
- "github.com/coredns/coredns/middleware/pkg/healthcheck"
-
- "google.golang.org/grpc/grpclog"
-)
-
-func pool() []*healthcheck.UpstreamHost {
- return []*healthcheck.UpstreamHost{
- {
- Name: "localhost:10053",
- },
- {
- Name: "localhost:10054",
- },
- }
-}
-
-func TestStartupShutdown(t *testing.T) {
- grpclog.SetLogger(discard{})
-
- upstream := &staticUpstream{
- from: ".",
- HealthCheck: healthcheck.HealthCheck{
- Hosts: pool(),
- FailTimeout: 10 * time.Second,
- Future: 60 * time.Second,
- MaxFails: 1,
- },
- }
- g := newGrpcClient(nil, upstream)
- upstream.ex = g
-
- p := &Proxy{}
- p.Upstreams = &[]Upstream{upstream}
-
- err := g.OnStartup(p)
- if err != nil {
- t.Errorf("Error starting grpc client exchanger: %s", err)
- return
- }
- if len(g.clients) != len(pool()) {
- t.Errorf("Expected %d grpc clients but found %d", len(pool()), len(g.clients))
- }
-
- err = g.OnShutdown(p)
- if err != nil {
- t.Errorf("Error stopping grpc client exchanger: %s", err)
- return
- }
- if len(g.clients) != 0 {
- t.Errorf("Shutdown didn't remove clients, found %d", len(g.clients))
- }
- if len(g.conns) != 0 {
- 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{}) {}