Unit test updated for NVME fix (#43).

petersulyok · Dec 12, 2024 · aa0686d · aa0686d
1 parent 9e03b63
commit aa0686d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_00_data.py b/test/test_00_data.py
@@ -101,7 +101,7 @@ def create_hd_temp_files(self, count: int, temp_list: List[float] = None, wildch
             # NVME disk type.
             elif hd_types[i] == self.HT_NVME:
                 new_dir = self.td_dir + '/sys/class/nvme/'
-                new_path = new_dir + "nvme" + str(nvme_counter) + '/' + "nvme" + str(nvme_counter) + 'n1'
+                new_path = new_dir + "nvme" + str(nvme_counter) + '/' + "nvme" + str(nvme_counter) + 'n1/device'
                 os.makedirs(new_path, exist_ok=True)
 
                 real_path = new_path + '/hwmon' + str(i) + '/'
@@ -238,7 +238,7 @@ def get_random_str(count: int) -> str:
                     f.write(str(' '))
                 # Create a link with device name.
                 dev_name = os.path.join(dev_dir, 'nvme-Samsung_SSD_870_PRO_1TB_' + get_random_str(8))
-                os.symlink('../../nvme' + str(nvme_counter), dev_name)
+                os.symlink('../../nvme' + str(nvme_counter) + 'n1', dev_name)
                 hd_names = hd_names + dev_name + separator
                 nvme_counter += 1