Use the disk currently in use by SONiC for the 'show platform ssdheal…

…th' command.

show/platform.py

@@ -104,7 +104,15 @@ def psustatus(index, json, verbose):
 def ssdhealth(device, verbose, vendor):
     """Show SSD Health information"""
     if not device:
-        device = os.popen("lsblk -o NAME,TYPE -p | grep disk").readline().strip().split()[0]
+        # Looking for the SSD disk currently used by the SONiC.
+
+        base_device = os.popen("lsblk -l -o NAME,TYPE,MOUNTPOINT -p | grep -w '/host'").readline().strip().split()[0]
+
+        # Extract the base device name, handling both 'p' and non-'p' partitioning schemes
+        # Example : /dev/sda1 => /dev/sda
+        #           /dev/nvme0n1p2 => /dev/nvme0n1
+        #           /dev/mmcblk0p1 => /dev/mmcblk0
+        device = base_device.rstrip("0123456789").split("p")[0]
     cmd = ['sudo', 'ssdutil', '-d', str(device)]
     options = ["-v"] if verbose else []
     options += ["-e"] if vendor else []

tests/show_test.py

@@ -693,7 +693,7 @@ def test_ssdhealth(self, mock_popen, mock_run_command):
         print(result.exit_code)
         print(result.output)
         assert result.exit_code == 0
-        mock_popen.assert_called_once_with('lsblk -o NAME,TYPE -p | grep disk')
+        mock_popen.assert_called_once_with("lsblk -l -o NAME,TYPE,MOUNTPOINT -p | grep -w '/host'")
         mock_run_command.assert_called_once_with(['sudo', 'ssdutil', '-d', '/dev/sda', '-v', '-e'], display_cmd=True)
 
     @patch('utilities_common.cli.run_command')