0da291f8a6
This reverts commit f8707f994af2582f6dc58190106946efeb43bf05. (cherry picked from commit 7a9bef0166951a61bc7094514a20471ae45f6090)
66 lines
2.4 KiB
Python
66 lines
2.4 KiB
Python
import tempfile
|
|
import unittest
|
|
|
|
from clang.cindex import File, Rewriter, SourceLocation, SourceRange, TranslationUnit
|
|
|
|
|
|
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; }")
|