From d7b15e82834e733f9562ee16bbcd9d37816336b9 Mon Sep 17 00:00:00 2001
From: delldubey <Utkarsh.Dubey@dell.com>
Date: Tue, 25 Jun 2024 14:17:12 +0530
Subject: [PATCH] fixes unit tests for NVMe (#64)

---
 mock/mock.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mock/mock.go b/mock/mock.go
index 38b0162..9bdd97c 100644
--- a/mock/mock.go
+++ b/mock/mock.go
@@ -2349,6 +2349,7 @@ func newInitiator(initiatorID string, initiatorName string, initiatorType string
 		InitiatorType:        initiatorType,
 		FCID:                 "0",
 		IPAddress:            "192.168.1.175",
+		Host:                 hostID,
 		HostID:               hostID,
 		HostGroupIDs:         []string{},
 		LoggedIn:             true,
@@ -2464,6 +2465,7 @@ func AddHost(hostID string, hostType string, initiatorIDs []string) (*types.Host
 		for k, v := range Data.InitiatorIDToInitiator {
 			if v.InitiatorID == initID {
 				Data.InitiatorIDToInitiator[k].HostID = hostID
+				Data.InitiatorIDToInitiator[k].Host = hostID
 				break
 			}
 		}