Ad Unit Test for the changes in code

gonvme_mock.go

@@ -20,6 +20,8 @@ import (
 	"errors"
 	"fmt"
 	"strconv"
+
+	log "github.com/sirupsen/logrus"
 )
 
 const (
@@ -53,6 +55,27 @@ type MockNVMe struct {
 	NVMeType
 }
 
+type CommandRunner interface {
+	RunCommand(name string, args ...string) (string, error)
+}
+
+type MockCommandRunner struct {
+	Output string
+	Err    error
+}
+
+func (m *MockCommandRunner) RunCommand(name string, args ...string) (string, error) {
+	// Return pre-defined output or error for testing
+	return m.Output, m.Err
+}
+
+func RunCommandWithRunner(runner CommandRunner, command []string) (string, error) {
+	if len(command) == 0 {
+		return "", errors.New("command is empty")
+	}
+	return runner.RunCommand(command[0], command[1:]...)
+}
+
 // NewMockNVMe - returns a mock NVMe client
 func NewMockNVMe(opts map[string]string) *MockNVMe {
 	nvme := MockNVMe{
@@ -62,6 +85,24 @@ func NewMockNVMe(opts map[string]string) *MockNVMe {
 		},
 	}
 
+	mockRunner := &MockCommandRunner{
+		Output: string("/sbin/nvme"),
+		Err:    nil,
+	}
+
+	command := []string{"chroot", "which", "nvme"}
+	output, err := RunCommandWithRunner(mockRunner, command)
+
+	if err != nil{
+		log.Errorf("Empty command")
+	}
+
+    // Assertions
+    expectedPath := "/sbin/nvme"
+    if output != expectedPath {
+        log.Errorf("expected %v, got %v", expectedPath, output)
+    }
+
 	return &nvme
 }