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
|
#!/usr/bin/python
import unittest
from subprocess import Popen, PIPE
import difflib
import uuid
import os,sys,time
import argparse
import tempfile
verbose = 0
def diff_files(fromfile,tofile):
n = 3 #context lines
fromdate = time.ctime(os.stat(fromfile).st_mtime)
todate = time.ctime(os.stat(tofile).st_mtime)
fromlines = open(fromfile, 'U').readlines()
tolines = open(tofile, 'U').readlines()
diff = difflib.unified_diff(fromlines, tolines, fromfile, tofile,
fromdate, todate, n=n)
l = list(diff)
if l :
if verbose == 2:
sys.stdout.writelines(l)
return False
else :
return True
def test_application_stdout(self,expected_file,cmd):
print " ".join(cmd)
(output,output_file) = tempfile.mkstemp()
if verbose == 2:
print " ".join(cmd)
p = Popen(cmd, stdout=output)
p.communicate()
d = diff_files(output_file,expected_file)
#output.close()
os.remove(output_file)
self.assertTrue(d)
def test_application_file(self,expected_file,output_file,cmd):
print " ".join(cmd)
p = Popen(cmd, stdout=sys.stdout)
p.communicate()
d = diff_files(output_file,expected_file)
self.assertTrue(d)
class ApplicationTests(unittest.TestCase):
def test_reduced_dist(self):
expected_file = "tests/reduced-dist"
cmd = [ "./reduced_dist.native", "--deb-native-arch=amd64",
"./tests/sid-amd64-packages-20121001.bz2",
"./tests/sid-sources-20121001.bz2"
]
test_application_stdout(self,expected_file,cmd)
def test_basebuildsystem(self):
expected_file = "tests/min-cross-sources.list"
cmd = ["./basebuildsystem.native", "--deb-native-arch=amd64",
"./tests/sid-amd64-packages-20121001.bz2",
"./tests/sid-sources-20121001.bz2"
]
test_application_file(self,expected_file,"min-cross-sources.list",cmd)
def main():
global verbose
parser = argparse.ArgumentParser(description='description of you program')
parser.add_argument('-v', '--verbose', action='store_const', const=2)
parser.add_argument('-d', '--debug', action='store_true')
parser.add_argument('-pwd', type=str, nargs=1, help="dose root directory")
args = parser.parse_args()
verbose = args.verbose
suite = unittest.TestLoader().loadTestsFromTestCase(ApplicationTests)
unittest.TextTestRunner(verbosity=args.verbose).run(suite)
if __name__ == '__main__':
main()
|
