From d447e694b2b4dc5db40f5514b9b60ed53ffd4a7b Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Thu, 14 Mar 2019 00:22:11 -0600
Subject: [PATCH] Set NVMe/SMART test to FAIL if safety check fails

* Fixes issues #49
---
 .bin/Scripts/functions/hw_diags.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/.bin/Scripts/functions/hw_diags.py b/.bin/Scripts/functions/hw_diags.py
index b1a2a79a..331289c8 100644
--- a/.bin/Scripts/functions/hw_diags.py
+++ b/.bin/Scripts/functions/hw_diags.py
@@ -215,11 +215,12 @@ class DiskObj():
     # Done
     return test_running
 
-  def disable_test(self, name, status):
+  def disable_test(self, name, status, test_failed=False):
     """Disable test by name and update status."""
     if name in self.tests:
       self.tests[name].update_status(status)
       self.tests[name].disabled = True
+      self.tests[name].failed = test_failed
 
   def generate_attribute_report(
       self, description=False, timestamp=False):
@@ -450,7 +451,7 @@ class DiskObj():
       for t in ['badblocks', 'I/O Benchmark']:
         self.disable_test(t, 'Denied')
     elif not disk_ok:
-      self.disable_test('NVMe / SMART', 'FAIL')
+      self.disable_test('NVMe / SMART', 'FAIL', test_failed=True)
       for t in ['badblocks', 'I/O Benchmark']:
         self.disable_test(t, 'Denied')