summaryrefslogtreecommitdiffstats
path: root/SharedModules/Utils/atfutils.py
blob: 980c286e1457cf4f11fc58db1ecf69ae059fc656 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
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']