From b71bca45777d0c6af571c37f22de5b2f97ac40e6 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Tue, 3 Dec 2019 16:31:26 -0700
Subject: [PATCH] Updated disk_attribute_check()

---
 scripts/wk/hw/diags.py | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/scripts/wk/hw/diags.py b/scripts/wk/hw/diags.py
index dc98af20..09a39197 100644
--- a/scripts/wk/hw/diags.py
+++ b/scripts/wk/hw/diags.py
@@ -495,10 +495,19 @@ def cpu_mprime_test(state, test_objects):
 def disk_attribute_check(state, test_objects):
   """Disk attribute check."""
   LOG.info('Disk Attribute Check')
-  #TODO: at
-  LOG.debug('%s, %s', state, test_objects)
-  std.print_warning('TODO: at')
-  std.pause()
+  for test in test_objects:
+    if not test.dev.attributes:
+      # No NVMe/SMART data
+      test.set_status('N/A')
+      continue
+
+    if test.dev.check_attributes():
+      test.set_status('Passed')
+    else:
+      test.set_status('Failed')
+
+  # Done
+  state.update_progress_pane()
 
 
 def disk_io_benchmark(state, test_objects):