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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
#!/usr/bin/python
import os
import sys
import subprocess, re, shutil, glob
import gettext
_ = gettext.lgettext
COLOR_BLACK = "\033[00m"
COLOR_RED = "\033[1;31m"
PRESCRIPTS = """
patch -s < ks.p
patch -s < conf.p
sudo mv /etc/mic/mic.conf /etc/mic/orig.conf
sudo mv test.conf /etc/mic/mic.conf
"""
POSTSCRIPTS = """
sudo mv -f /etc/mic/orig.conf /etc/mic/mic.conf
"""
def PrepEnv(cases_dir, case, work_env):
"""prepare working env"""
for one in glob.glob(os.path.join(cases_dir, 'base', '*')):
shutil.copy(one, work_env)
for other in glob.glob(os.path.join(cases_dir, 'test-'+case, '*')):
shutil.copy(other, work_env)
def ImgCheck(work_env):
"""check image generate"""
genImage = False
for root, dirs, files in os.walk(work_env):
for name in files:
#add raw check support and XXX.tar file check support
m = re.match(r'.*\.(img|raw|iso|usbimg|tar)', name) or re.match(r'system-release',name)
if m:
genImage = True
break
return genImage
def RunandCheck(object, work_env):
"""run mic-image-creator command and check something"""
ret = False
cwd = os.getcwd()
os.chdir(work_env)
os.system(PRESCRIPTS)
#set value of "expect"
expect = None
if "expect" in os.listdir(work_env):
exp_f = open('expect', 'r')
exp = exp_f.read()
if len(exp) > 0:
expect = exp.strip()
exp_f.close()
#set cmdline
opt_f = open('options','r')
mic_cmd = opt_f.read().strip()
if mic_cmd.find('-h')!=-1 or mic_cmd.find('help')!=-1 or mic_cmd.find('?')!=-1:
args = mic_cmd
else:
args = mic_cmd+' test.ks'
print args
log = open('miclog','w')
proc = subprocess.Popen(args,stdout = log ,stderr=subprocess.PIPE,shell=True)
errorinfo = proc.communicate()[1]
log.close()
mic_cmd_msg = None
miclog_f = open('miclog','r')
miclog_tuple = miclog_f.read()
if len(miclog_tuple) > 0:
mic_cmd_msg = miclog_tuple.strip()
#check
if expect:
if errorinfo.find(expect) != -1 or mic_cmd_msg.find(expect) != -1 :#FIXME
ret =True
else:
proc.wait()
ret = ImgCheck(work_env)
os.system(POSTSCRIPTS)
os.chdir(cwd)
try:
object.assertTrue(ret)
except object.failureException:
if expect:
''' Used to update help expect info automaticlly.
path = object._testMethodName
path = path.replace('_','-',1)
os.unlink('%s/mic_cases/%s/expect' % (cwd,path))
fp = open('%s/mic_cases/%s/expect' % (cwd,path),'w')
fp.write(mic_cmd_msg)
fp.close()
'''
raise object.failureException(_("Expect and mic out msg are not constant\n%sExpect:%s\n\nMic out msg:%s%s") %(COLOR_RED,expect,mic_cmd_msg,COLOR_BLACK))
else:
raise object.failureException(_("%s%s%s") %(COLOR_RED,errorinfo,COLOR_BLACK))
|
