1 files changed, 58 insertions, 0 deletions
diff --git a/libs/utils/atfutils.py b/libs/utils/atfutils.py
new file mode 100644
index 0000000..980c286
--- /dev/null
+++ b/libs/utils/atfutils.py
@@ -0,0 +1,58 @@
+"""atfutils module contains  general purpose wrappers
+
+*) assert_success
+*) assert_failure
+*) print-stdout
+*) print_stderr
+*) set_active_volume
+"""
+
+import re
+import ssh
+from atfglobals import GlobalObj
+
+def assert_success(**arguments):
+    """
+    """     
+    if not arguments["exitstatus"]:
+        return 0
+    else:
+        return 1
+
+def assert_failure(**arguments):
+    """
+    """        
+    if arguments["exitstatus"]:
+        return 0
+    else:
+        return 1
+
+def print_stdout(stdoutdata):
+    """
+    """
+    if not stdoutdata == None:
+        for data in stdoutdata:
+            print data
+
+def print_stderr(stderrdata):
+    if not stderrdata == None:
+        for data in stderrdata:
+            print data
+
+def set_active_volume(volumekey):
+    """
+    """
+    env = GlobalObj.getTestenvObj()
+    return_status = env.setActiveVolume(volumekey)
+    if return_status:
+        print "Unable to set Active Volume. '%s' Not defined in TestEnvironment"\
+              % volumekey
+    return return_status
+
+
+
+__all__ = ['assert_success',
+           'assert_failure',
+           'print-stdout',
+           'print_stderr',
+           'set_active_volume']