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
|
import unittest
import tempfile
from clang.cindex import (
Rewriter,
TranslationUnit,
File,
SourceLocation,
SourceRange,
)
class TestRewrite(unittest.TestCase):
code = """int main() { return 0; }"""
def setUp(self):
self.tmp = tempfile.NamedTemporaryFile(suffix=".cpp", buffering=0)
self.tmp.write(TestRewrite.code.encode("utf-8"))
self.tmp.flush()
self.tu = TranslationUnit.from_source(self.tmp.name)
self.rew = Rewriter.create(self.tu)
self.file = File.from_name(self.tu, self.tmp.name)
def tearDown(self):
self.tmp.close()
def get_content(self) -> str:
with open(self.tmp.name, "r", encoding="utf-8") as f:
return f.read()
def test_replace(self):
rng = SourceRange.from_locations(
SourceLocation.from_position(self.tu, self.file, 1, 5),
SourceLocation.from_position(self.tu, self.file, 1, 9),
)
self.rew.replace_text(rng, "MAIN")
self.rew.overwrite_changed_files()
self.assertEqual(self.get_content(), "int MAIN() { return 0; }")
def test_replace_shorter(self):
rng = SourceRange.from_locations(
SourceLocation.from_position(self.tu, self.file, 1, 5),
SourceLocation.from_position(self.tu, self.file, 1, 9),
)
self.rew.replace_text(rng, "foo")
self.rew.overwrite_changed_files()
self.assertEqual(self.get_content(), "int foo() { return 0; }")
def test_replace_longer(self):
rng = SourceRange.from_locations(
SourceLocation.from_position(self.tu, self.file, 1, 5),
SourceLocation.from_position(self.tu, self.file, 1, 9),
)
self.rew.replace_text(rng, "patatino")
self.rew.overwrite_changed_files()
self.assertEqual(self.get_content(), "int patatino() { return 0; }")
def test_insert(self):
pos = SourceLocation.from_position(self.tu, self.file, 1, 5)
self.rew.insert_text_before(pos, "ro")
self.rew.overwrite_changed_files()
self.assertEqual(self.get_content(), "int romain() { return 0; }")
def test_remove(self):
rng = SourceRange.from_locations(
SourceLocation.from_position(self.tu, self.file, 1, 5),
SourceLocation.from_position(self.tu, self.file, 1, 9),
)
self.rew.remove_text(rng)
self.rew.overwrite_changed_files()
self.assertEqual(self.get_content(), "int () { return 0; }")
|
