[VOL-2193] Create mocks for Kafka Client and Etcd
This commit consists of:
1) A kafka client mock that implements the kafka client interface
under voltha-lib-go/pkg/kafka/client.go
2) An embedded Etcd server that runs in-process and represents an
Etcd server.
Change-Id: I52a36132568e08c596bb4136918bebcb654a3b99
diff --git a/pkg/mocks/kafka_client_test.go b/pkg/mocks/kafka_client_test.go
new file mode 100644
index 0000000..f4707f6
--- /dev/null
+++ b/pkg/mocks/kafka_client_test.go
@@ -0,0 +1,100 @@
+/*
+ * Copyright 2019-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 mocks
+
+import (
+ "github.com/opencord/voltha-lib-go/v2/pkg/kafka"
+ ic "github.com/opencord/voltha-protos/v2/go/inter_container"
+ "github.com/stretchr/testify/assert"
+ "testing"
+ "time"
+)
+
+func TestKafkaClientImplementsKafkaClientIf(t *testing.T) {
+ client := NewKafkaClient()
+
+ if _, ok := interface{}(client).(kafka.Client); !ok {
+ t.Error("mock kafka client does not implement voltha-lib-go/v2/pkg/kafka/Client interface")
+ }
+}
+
+func TestKafkaClientCreateTopic(t *testing.T) {
+ cTkc := NewKafkaClient()
+ topic := kafka.Topic{Name: "myTopic"}
+ err := cTkc.CreateTopic(&topic, 1, 1)
+ assert.Nil(t, err)
+ err = cTkc.CreateTopic(&topic, 1, 1)
+ assert.NotNil(t, err)
+}
+
+func TestKafkaClientDeleteTopic(t *testing.T) {
+ cTkc := NewKafkaClient()
+ topic := kafka.Topic{Name: "myTopic"}
+ err := cTkc.DeleteTopic(&topic)
+ assert.Nil(t, err)
+}
+
+func TestKafkaClientSubscribeSend(t *testing.T) {
+ cTkc := NewKafkaClient()
+ topic := kafka.Topic{Name: "myTopic"}
+ ch, err := cTkc.Subscribe(&topic)
+ assert.Nil(t, err)
+ assert.NotNil(t, ch)
+ testCh := make(chan bool)
+ maxWait := 5 * time.Millisecond
+ msg := &ic.InterContainerMessage{
+ Header: &ic.Header{Id: "1234", ToTopic: topic.Name},
+ Body: nil,
+ }
+ timer := time.NewTimer(maxWait)
+ defer timer.Stop()
+ go func() {
+ select {
+ case val, ok := <-ch:
+ assert.True(t, ok)
+ assert.Equal(t, val, msg)
+ testCh <- true
+ case <-timer.C:
+ testCh <- false
+ }
+ }()
+ err = cTkc.Send(msg, &topic)
+ assert.Nil(t, err)
+ res := <-testCh
+ assert.True(t, res)
+}
+
+func TestKafkaClientUnSubscribe(t *testing.T) {
+ cTkc := NewKafkaClient()
+ topic := kafka.Topic{Name: "myTopic"}
+ ch, err := cTkc.Subscribe(&topic)
+ assert.Nil(t, err)
+ assert.NotNil(t, ch)
+ err = cTkc.UnSubscribe(&topic, ch)
+ assert.Nil(t, err)
+}
+
+func TestKafkaClientStop(t *testing.T) {
+ cTkc := NewKafkaClient()
+ topic := kafka.Topic{Name: "myTopic"}
+ ch, err := cTkc.Subscribe(&topic)
+ assert.Nil(t, err)
+ assert.NotNil(t, ch)
+ err = cTkc.UnSubscribe(&topic, ch)
+ assert.Nil(t, err)
+ cTkc.Stop()
+}