blob: 0978752bb88614045ac903905c3cfbdbbb852361 (
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
#!/usr/bin/python
#
# Copyright 2007 Google Inc. Released under the GPL v2
"""This module defines the unittests for the utils
"""
__author__ = """stutsman@google.com (Ryan Stutsman)"""
import os
import sys
import os.path
import unittest
# Adjust the path so Python can find the autoserv modules
src = os.path.abspath("%s/.." % (os.path.dirname(sys.argv[0]),))
if src not in sys.path:
sys.path.insert(1, src)
import utils
class UtilsTestCase(unittest.TestCase):
def setUp(self):
pass
def tearDown(self):
pass
def testGetWithOpenFile(self):
tmpdir = utils.get_tmp_dir()
tmppath = os.path.join(tmpdir, 'testfile')
tmpfile = file(tmppath, 'w')
print >> tmpfile, 'Test string'
tmpfile.close()
tmpfile = file(tmppath)
newtmppath = utils.get(tmpfile)
self.assertEqual(file(newtmppath).read(), 'Test string\n')
def testGetWithHTTP(self):
# Yeah, this test is a bad idea, oh well
url = 'http://www.kernel.org/pub/linux/kernel/README'
tmppath = utils.get(url)
f = file(tmppath)
f.readline()
self.assertTrue('Linux' in f.readline().split())
def testGetWithPath(self):
path = utils.get('/proc/cpuinfo')
self.assertTrue(file(path).readline().startswith('processor'))
def testGetWithString(self):
path = utils.get('/tmp loves rabbits!')
self.assertTrue(file(path).readline().startswith('/tmp loves'))
def testGetWithDir(self):
tmpdir = utils.get_tmp_dir()
origpath = os.path.join(tmpdir, 'testGetWithDir')
os.mkdir(origpath)
dstpath = utils.get(origpath)
self.assertTrue(dstpath.endswith('/'))
self.assertTrue(os.path.isdir(dstpath))
def suite():
return unittest.TestLoader().loadTestsFromTestCase(UtilsTestCase)
if __name__ == '__main__':
unittest.TextTestRunner(verbosity=2).run(suite())
|
