diff --git a/pkg/mocks/mockEventproxy.go b/pkg/mocks/mockEventproxy.go
index 71c7fec..50ffbd9 100644
--- a/pkg/mocks/mockEventproxy.go
+++ b/pkg/mocks/mockEventproxy.go
@@ -53,3 +53,13 @@
 	}
 	return nil
 }
+
+// EnableLivenessChannel mocks the EnableLivenessChannel function
+func (me *MockEventProxy) EnableLivenessChannel(ctx context.Context, enable bool) chan bool {
+	return nil
+}
+
+// SendLiveness mocks the SendLiveness function
+func (me *MockEventProxy) SendLiveness(ctx context.Context) error {
+	return nil
+}
diff --git a/pkg/mocks/mockTechprofile.go b/pkg/mocks/mockTechprofile.go
index a016b2c..f150190 100644
--- a/pkg/mocks/mockTechprofile.go
+++ b/pkg/mocks/mockTechprofile.go
@@ -185,9 +185,13 @@
 	return []*tp_pb.TrafficQueue{{}}
 }
 
-// GetGemportIDForPbit to mock techprofile GetGemportIDForPbit method
-func (m MockTechProfile) GetGemportIDForPbit(ctx context.Context, tp interface{}, Dir tp_pb.Direction, pbit uint32) uint32 {
-	return 0
+// GetGemportForPbit to mock techprofile GetGemportForPbit method
+func (m MockTechProfile) GetGemportForPbit(ctx context.Context, tpInst interface{}, Dir tp_pb.Direction, pbit uint32) interface{} {
+	return tp.IGemPortAttribute{
+		GemportID:     1,
+		PbitMap:       "0b11111111",
+		AesEncryption: "false",
+	}
 }
 
 // FindAllTpInstances to mock techprofile FindAllTpInstances method