From b162c99d6e5aa5f3897f0c7791edbe94246e02a7 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Thu, 7 Nov 2019 20:40:50 -0700
Subject: [PATCH] Added BaseObj() class

---
 scripts/wk/hw/obj.py | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/scripts/wk/hw/obj.py b/scripts/wk/hw/obj.py
index 3bba408d..495051de 100644
--- a/scripts/wk/hw/obj.py
+++ b/scripts/wk/hw/obj.py
@@ -48,9 +48,24 @@ class SMARTNotSupportedError(TypeError):
 
 
 # Classes
-class CpuRam():
+class BaseObj():
+  """Base object for tracking device data."""
+  def __init__(self):
+    self.tests = OrderedDict()
+
+  def all_tests_passed(self):
+    """Check if all tests passed, returns bool."""
+    return all([results.passed for results in self.tests.values()])
+
+  def any_test_failed(self):
+    """Check if any test failed, returns bool."""
+    return any([results.failed for results in self.tests.values()])
+
+
+class CpuRam(BaseObj):
   """Object for tracking CPU & RAM specific data."""
   def __init__(self):
+    super().__init__()
     self.description = 'Unknown'
     self.details = {}
     self.ram_total = 'Unknown'
@@ -127,9 +142,10 @@ class CpuRam():
       ]
 
 
-class Disk():
+class Disk(BaseObj):
   """Object for tracking disk specific data."""
   def __init__(self, path):
+    super().__init__()
     self.attributes = {}
     self.description = 'Unknown'
     self.details = {}