Skip to content

Commit

Permalink
Changed: updated for the refactored get_smart_data() method.
Browse files Browse the repository at this point in the history
  • Loading branch information
@petersulyok
petersulyok committed Mar 3, 2024
1 parent cb8b459 commit 893f419
Showing 1 changed file with 29 additions and 38 deletions.
67 changes: 29 additions & 38 deletions test/test_disksmart.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,56 +12,47 @@ class DiskSmartDataTest(unittest.TestCase):
def test_find_smart_attribute_by_id(self):
"""Unit test for find_smart_attribute_by_id() method."""
my_tsd = TestSmartData()
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(5), 0, "find_smart_attribute_by_id 1")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(9), 1, "find_smart_attribute_by_id 2")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(12), 2, "find_smart_attribute_by_id 3")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(177), 3, "find_smart_attribute_by_id 4")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(179), 4, "find_smart_attribute_by_id 5")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(181), 5, "find_smart_attribute_by_id 6")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(182), 6, "find_smart_attribute_by_id 7")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(183), 7, "find_smart_attribute_by_id 8")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(187), 8, "find_smart_attribute_by_id 9")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(190), 9, "find_smart_attribute_by_id 10")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(195), 10, "find_smart_attribute_by_id 11")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(199), 11, "find_smart_attribute_by_id 12")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(235), 12, "find_smart_attribute_by_id 13")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(241), 13, "find_smart_attribute_by_id 14")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_id(253), -1, "find_smart_attribute_by_id 15")
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_id(1), 0,
"find_smart_attribute_by_id 1")
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_id(5), 3,
"find_smart_attribute_by_id 2")
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_id(9), 5,
"find_smart_attribute_by_id 3")
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_id(12), 7,
"find_smart_attribute_by_id 4")
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_id(187), 8,
"find_smart_attribute_by_id 5")
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_id(188), 9,
"find_smart_attribute_by_id 6")
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_id(195), 14,
"find_smart_attribute_by_id 7")
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_id(199), 17,
"find_smart_attribute_by_id 8")
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_id(200), 18,
"find_smart_attribute_by_id 9")
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_id(241), 20,
"find_smart_attribute_by_id 10")
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_id(253), -1,
"find_smart_attribute_by_id 11")
del my_tsd

def test_find_smart_attribute_by_name(self):
"""Unit test for find_smart_attribute_by_id() method."""
my_tsd = TestSmartData()
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("Reallocated_Sector"), 0,
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_name("Reallocated_Sector"), 3,
"find_smart_attribute_by_name 1")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("Power_On_Hours"), 1,
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_name("Power_On_Hours"), 5,
"find_smart_attribute_by_name 2")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("Power_Cycle_Count"), 2,
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_name("Power_Cycle_Count"), 7,
"find_smart_attribute_by_name 3")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("Wear_Leveling_Count"), 3,
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_name("Airflow_Temperature"), 10,
"find_smart_attribute_by_name 4")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("Used_Rsvd_Blk_Cnt_Tot"), 4,
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_name("Hardware_ECC_Recovered"), 14,
"find_smart_attribute_by_name 5")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("Program_Fail_Cnt_Total"), 5,
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_name("Total_LBAs_Written"), 20,
"find_smart_attribute_by_name 6")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("Erase_Fail_Count_Total"), 6,
self.assertEqual(my_tsd.tsd[1].result.find_smart_attribute_by_name("nonexistent_attribute"), -1,
"find_smart_attribute_by_name 7")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("Runtime_Bad_Block"), 7,
"find_smart_attribute_by_name 8")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("Uncorrectable_Error"), 8,
"find_smart_attribute_by_name 9")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("Airflow_Temperature"), 9,
"find_smart_attribute_by_name 10")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("ECC_Error_Rate"), 10,
"find_smart_attribute_by_name 12")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("CRC_Error_Count"), 11,
"find_smart_attribute_by_name 13")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("POR_Recovery_Count"), 12,
"find_smart_attribute_by_name 14")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("Total_LBAs_Written"), 13,
"find_smart_attribute_by_name 15")
self.assertEqual(my_tsd.smart_results[0].find_smart_attribute_by_name("nonexistent_attribute"), -1,
"find_smart_attribute_by_name 16")
del my_tsd


Expand Down

0 comments on commit 893f419

Please sign in to comment.