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
|
# Unit tests for sedsed
import unittest
import sys
import os
# Make ../sedsed.py importable
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
import sedsed # pylint: disable=wrong-import-position
# Mock to avoid the print() call from the original
sedsed.fatal_error = sys.exit
class TestSedsed(unittest.TestCase): # pylint: disable=unused-variable
def test_read_file_not_found(self):
with self.assertRaises(SystemExit):
sedsed.read_file("404.txt")
def test_read_file_ok(self):
output = sedsed.read_file(__file__)
self.assertEqual(output[0], "# Unit tests for sedsed")
def test_write_file_use_dir(self):
with self.assertRaises(SystemExit):
some_dir = os.path.dirname(__file__)
sedsed.write_file(some_dir, ["line1"])
def test_write_file_ok(self):
temp_file = __file__ + ".tmp"
sedsed.write_file(temp_file, ["line1"])
self.assertTrue(os.path.isfile(temp_file))
os.remove(temp_file)
def test_lastaddr_should_be_empty(self):
"""
The "y" command should not save or set lastaddr.
Numeric or `$` addresses should not save lastaddr.
"""
data = [
("y/a/x/", "s///"),
("s/a/x/", "y///"),
("1x", "s///"),
("1,5x", "s///"),
("1~5x", "s///"),
("1,~5x", "s///"),
("1,+5x", "s///"),
("$x", "s///"),
]
expected = ""
for script in data:
result = sedsed.parse(script)
self.assertEqual(result[-1]["lastaddr"], expected, msg=script)
def test_lastaddr_should_be_set(self):
"""
Any /.../ or s/...// should trigger the saving of lastaddr.
Any empty address // or s//foo should have its lastaddr set.
"""
data = [
("s/foo1/x/", "s///"),
("/foo2/x", "s///"),
("s/foo3/x/", "//p"),
("/foo4/x", "//p"),
]
for index, script in enumerate(data, start=1):
expected = "/foo%s/" % index
result = sedsed.parse(script)
self.assertEqual(result[-1]["lastaddr"], expected, msg=script)
def test_lastaddr_should_have_flags(self):
"""
Any address flags should also be set in lastaddr.
"""
data = [
("/foo1/Ix", "s///"),
("/foo2/Mx", "s///"),
("/foo3/MIx", "s///"),
("/foo4/x", "s///"),
]
for script in data:
result = sedsed.parse(script)
expected = script[0][:-1] # remove "x" command
self.assertEqual(result[-1]["lastaddr"], expected, msg=script)
def test_lastaddr_should_have_escaped_delimiter(self):
"""
If the address delimiter is not the slash, it should be escaped in lastaddr.
"""
data = [("s@foo@x@", "s///"), ("\\@foo@x", "s///")]
expected = "\\@foo@"
for script in data:
result = sedsed.parse(script)
self.assertEqual(result[-1]["lastaddr"], expected, msg=script)
if __name__ == "__main__":
unittest.main()
|
