VOL-2098 Support for Kafka liveness testing
* Adds liveness channel to sarama_client and kafka_interadapter proxy.
The liveness channel will push true or false to the channel on
each successful or failed Kafka publish.
* Adds support to make a "liveness publish attempt", which publishes
an empty message on a _liveness channel.
* Adds ServiceStatusNotReady to Probe
* Suppresses multiple Probe.UpdateStatus of the same status
* Adds the ability to attach a Probe to the grpc server, so that
when the probe returns NotReady, the Server responds to requests
with UNAVAILABLE.
Change-Id: I996c719570a50f2f6f397887d10d489608269c3f
diff --git a/pkg/grpc/server_test.go b/pkg/grpc/server_test.go
new file mode 100644
index 0000000..a3afc09
--- /dev/null
+++ b/pkg/grpc/server_test.go
@@ -0,0 +1,106 @@
+/*
+ * Copyright 2018-present Open Networking Foundation
+
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+
+ * http://www.apache.org/licenses/LICENSE-2.0
+
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package grpc
+
+import (
+ "context"
+ "github.com/stretchr/testify/assert"
+ "google.golang.org/grpc"
+ "testing"
+)
+
+// A Mock Probe that returns the Ready member using the IsReady() func
+type MockReadyProbe struct {
+ Ready bool
+}
+
+func (m *MockReadyProbe) IsReady() bool {
+ return m.Ready
+}
+
+// A Mock handler that returns the request as its result
+func MockUnaryHandler(ctx context.Context, req interface{}) (interface{}, error) {
+ _ = ctx
+ return req, nil
+}
+
+func TestNewGrpcServer(t *testing.T) {
+ server := NewGrpcServer("127.0.0.1", 1234, nil, false)
+ assert.NotNil(t, server)
+}
+
+func TestMkServerInterceptorNoProbe(t *testing.T) {
+ server := NewGrpcServer("127.0.0.1", 1234, nil, false)
+ assert.NotNil(t, server)
+
+ f := mkServerInterceptor(server)
+ assert.NotNil(t, f)
+
+ req := "SomeRequest"
+ serverInfo := grpc.UnaryServerInfo{Server: nil, FullMethod: "somemethod"}
+
+ result, err := f(context.Background(),
+ req,
+ &serverInfo,
+ MockUnaryHandler)
+
+ assert.Nil(t, err)
+ assert.Equal(t, "SomeRequest", result)
+}
+
+func TestMkServerInterceptorReady(t *testing.T) {
+ server := NewGrpcServer("127.0.0.1", 1234, nil, false)
+ assert.NotNil(t, server)
+
+ f := mkServerInterceptor(server)
+ assert.NotNil(t, f)
+
+ req := "SomeRequest"
+ serverInfo := grpc.UnaryServerInfo{Server: nil, FullMethod: "somemethod"}
+
+ probe := &MockReadyProbe{Ready: true}
+ server.AttachReadyProbe(probe)
+
+ result, err := f(context.Background(),
+ req,
+ &serverInfo,
+ MockUnaryHandler)
+
+ assert.Nil(t, err)
+ assert.NotNil(t, result)
+}
+
+func TestMkServerInterceptorNotReady(t *testing.T) {
+ server := NewGrpcServer("127.0.0.1", 1234, nil, false)
+ assert.NotNil(t, server)
+
+ f := mkServerInterceptor(server)
+ assert.NotNil(t, f)
+
+ req := "SomeRequest"
+ serverInfo := grpc.UnaryServerInfo{Server: nil, FullMethod: "somemethod"}
+
+ probe := &MockReadyProbe{Ready: false}
+ server.AttachReadyProbe(probe)
+
+ result, err := f(context.Background(),
+ req,
+ &serverInfo,
+ MockUnaryHandler)
+
+ assert.NotNil(t, err)
+ assert.Nil(t, result)
+}