From dd2d9ea4d1292cf4e1d12d05dd5dd5e78e4645e3 Mon Sep 17 00:00:00 2001
From: Josef <j.dabrowski@distreon.com.au>
Date: Wed, 11 Dec 2019 15:26:41 +0800
Subject: [PATCH] complete issue 8

---
 notiframe/Notiframe.py | 31 ++++++++++++++++++-------------
 1 file changed, 18 insertions(+), 13 deletions(-)

diff --git a/notiframe/Notiframe.py b/notiframe/Notiframe.py
index 5c62d48..fbc5b31 100644
--- a/notiframe/Notiframe.py
+++ b/notiframe/Notiframe.py
@@ -8,19 +8,24 @@ class Manager():
     def gen_id(self):
         return uuid.uuid4()
 
+class Test():
+    # Test class to test other classes and their functions.
+    def __init__(self):
+        #instantiate manager class
+        self.manager = Manager()
 
-#do function wrapper here for tests functions, to say "test_function_name" success/failure
-def TEST_manager():
-    manager = Manager()
-    uuid_test = manager.gen_id()
-    print(is_valid_uuid(uuid_test))
+    # Note: Do function wrapper here for tests functions, to say "test_function_name" success/failure
+    def manager_uuid(self):
+        uuid_test = self.manager.gen_id()
+        print(self.is_valid_uuid(uuid_test))
 
-# Test if manager class's gen_id function produces a valid uuid
-def is_valid_uuid(val):
-    try:
-        uuid.UUID(str(val))
-        return True
-    except ValueError:
-        return False
+    # Test if manager class's gen_id function produces a valid uuid
+    def is_valid_uuid(self, val):
+        try:
+            uuid.UUID(str(val))
+            return True
+        except ValueError:
+            return False
 
-TEST_manager()
+test = Test()
+test.manager_uuid()