diff --git a/main_test.go b/main_test.go
index 1192ff6..362a135 100644
--- a/main_test.go
+++ b/main_test.go
@@ -18,6 +18,8 @@
 import (
 	"context"
 	"errors"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
 	"testing"
 
 	"github.com/opencord/voltha-lib-go/v2/pkg/kafka"
@@ -216,3 +218,11 @@
 	}
 	return errors.New("invalid topic")
 }
+
+func (kc *mockKafkaClient) SendLiveness() error {
+	return status.Error(codes.Unimplemented, "SendLiveness")
+}
+
+func (kc *mockKafkaClient) EnableLivenessChannel(enable bool) chan bool {
+	return nil
+}