Move compare functions into Test base
authorJack Miller <jack@codezen.org>
Sun, 23 Nov 2014 22:16:01 +0000 (16:16 -0600)
committerJack Miller <jack@codezen.org>
Mon, 24 Nov 2014 17:57:03 +0000 (11:57 -0600)
tests/base.py
tests/test-config-function.py

index c5cd8c5..d602d45 100644 (file)
@@ -1,3 +1,5 @@
+from canto_next.remote import access_dict
+
 from threading import Lock
 import traceback
 import logging
@@ -92,7 +94,26 @@ class Test(object):
     def __init__(self, name):
         self.name = name
         self.run()
-    
+
+    def compare_flags(self, value):
+        if self.flags != value:
+            raise Exception("Expected flags %d - got %d" % (value, self.flags))
+
+    def compare_config(self, config, var, evalue):
+        ok, got = access_dict(config, var)
+        if not ok:
+            raise Exception("Couldn't get %s?" % var)
+        if got != evalue:
+            raise Exception("Expected %s == %s - got %s" % (var, evalue, got))
+
+    def compare_var(self, var, evalue):
+        if hasattr(self, var):
+            val = getattr(self, var)
+            if val != evalue:
+                raise Exception("Expected self.%s == %s - got %s" % (var, evalue, val))
+        else:
+            raise Exception("Couldn't get self.%s?" % var)
+
     def run(self):
         print("STARTING %s\n" % self.name)
 
index 425e222..53cbc01 100755 (executable)
@@ -7,7 +7,6 @@ from canto_curses.main import CANTO_PROTOCOL_COMPATIBLE
 from canto_curses.config import config
 
 from canto_next.hooks import on_hook
-from canto_next.remote import access_dict
 
 OPT_CHANGE = 1
 TAG_OPT_CHANGE = 2
@@ -54,25 +53,6 @@ class TestConfigFunction(Test):
     def on_eval_tags_changed(self):
         self.flags |= EVAL_TAGS
 
-    def compare_flags(self, value):
-        if self.flags != value:
-            raise Exception("Expected flags %d - got %d" % (value, self.flags))
-
-    def compare_config(self, config, var, evalue):
-        ok, got = access_dict(config, var)
-        if not ok:
-            raise Exception("Couldn't get %s?" % var)
-        if got != evalue:
-            raise Exception("Expected %s == %s - got %s" % (var, evalue, got))
-
-    def compare_var(self, var, evalue):
-        if hasattr(self, var):
-            val = getattr(self, var)
-            if val != evalue:
-                raise Exception("Expected self.%s == %s - got %s" % (var, evalue, val))
-        else:
-            raise Exception("Couldn't get self.%s?" % var)
-
     def check(self):
         script = {
             'VERSION' : { '*' : ('VERSION', CANTO_PROTOCOL_COMPATIBLE) },