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
100
|
import os.path
import unittest
# PYCOMPAT
import six
from six import StringIO
from sos.utilities import grep, checksum, get_hash_name, is_executable, sos_get_command_output, find, tail, shell_out
import sos
TEST_DIR = os.path.dirname(__file__)
class GrepTest(unittest.TestCase):
def test_file_obj(self):
test_s = "\n".join(['this is only a test', 'there are only two lines'])
test_fo = StringIO(test_s)
matches = grep(".*test$", test_fo)
self.assertEquals(matches, ['this is only a test\n'])
def test_real_file(self):
matches = grep(".*unittest$", __file__.replace(".pyc", ".py"))
self.assertEquals(matches, ['import unittest\n'])
def test_open_file(self):
matches = grep(".*unittest$", open(__file__.replace(".pyc", ".py")))
self.assertEquals(matches, ['import unittest\n'])
def test_grep_multiple_files(self):
matches = grep(".*unittest$", __file__.replace(".pyc", ".py"), "does_not_exist.txt")
self.assertEquals(matches, ['import unittest\n'])
class TailTest(unittest.TestCase):
def test_tail(self):
t = tail("tests/tail_test.txt", 10)
self.assertEquals(t, six.b("last line\n"))
def test_tail_too_many(self):
t = tail("tests/tail_test.txt", 200)
expected = open("tests/tail_test.txt", "r").read()
self.assertEquals(t, six.b(expected))
class ChecksumTest(unittest.TestCase):
def test_simple_hash(self):
self.assertEquals(checksum(StringIO('this is a test'), algorithm="sha256"),
'2e99758548972a8e8822ad47fa1017ff72f06f3ff6a016851f45c398732bc50c')
def test_hash_loading(self):
# not the greatest test, since we are asking the policy to pick for us
name = get_hash_name()
self.assertTrue(name in ('md5', 'sha256'))
class ExecutableTest(unittest.TestCase):
def test_nonexe_file(self):
path = os.path.join(TEST_DIR, 'utility_tests.py')
self.assertFalse(is_executable(path))
def test_exe_file(self):
path = os.path.join(TEST_DIR, 'test_exe.py')
self.assertTrue(is_executable(path))
def test_exe_file_abs_path(self):
self.assertTrue(is_executable("/usr/bin/timeout"))
def test_output(self):
path = os.path.join(TEST_DIR, 'test_exe.py')
ret, out, junk = sos_get_command_output(path)
self.assertEquals(ret, 0)
self.assertEquals(out, "executed\n")
def test_output_non_exe(self):
path = os.path.join(TEST_DIR, 'utility_tests.py')
ret, out, junk = sos_get_command_output(path)
self.assertEquals(ret, 127)
self.assertEquals(out, "")
def test_shell_out(self):
path = os.path.join(TEST_DIR, 'test_exe.py')
self.assertEquals("executed\n", shell_out(path))
class FindTest(unittest.TestCase):
def test_find_leaf(self):
leaves = find("leaf", TEST_DIR)
self.assertTrue(any(name.endswith("leaf") for name in leaves))
def test_too_shallow(self):
leaves = find("leaf", TEST_DIR, max_depth=1)
self.assertFalse(any(name.endswith("leaf") for name in leaves))
def test_not_in_pattern(self):
leaves = find("leaf", TEST_DIR, path_pattern="tests/path")
self.assertFalse(any(name.endswith("leaf") for name in leaves))
|
