def setUp(self):
self.replacer = ThriftConstReplacer()
class ThriftConstReplacerTestCase(unittest.TestCase):
def setUp(self):
self.replacer = ThriftConstReplacer()
def test_parse_line_SimpleLine_ReturnAsIsNoConstsSet(self):
line = "// abcdefg"
output = self.replacer.parse_line(line)
self.assertEqual(line, output)
self.assertEqual({}, self.replacer.get_consts())
def test_parse_line_ConstantDeclaration_ConstsSet(self):
const = '"abc"'
line = "const string a = " + const + ";"
output = self.replacer.parse_line(line)
self.assertEqual(line, output)
self.assertEqual({'a': const}, self.replacer.get_consts())
def test_parse_line_ConstantDeclarationThenUsed_ConstReplaced(self):
const = '"abc"'
line = "const string a = " + const
self.replacer.parse_line(line)
line2 = "a + xyz"
output = self.replacer.parse_line(line2)
self.assertEqual('"abc" + xyz', output)
def test_parse_line_ConstantDeclarationThenUsedWithCommas_ConstReplaced(
self
):
const = '"abc"'
line = "const string a = " + const
self.replacer.parse_line(line)
line2 = "a, + xyz"
output = self.replacer.parse_line(line2)
self.assertEqual('"abc", + xyz', output)
def test_parse_line_ValueHasSpace_ConstReplaced(self):
const = '"abc def"'
line = "const string a = " + const
self.replacer.parse_line(line)
line2 = "a + xyz"
output = self.replacer.parse_line(line2)
self.assertEqual('"abc def" + xyz', output)
def test_parse_line_ValueAtSeparateLine_ConstReplaced(self):
const = '"abc def"'
line1 = "const string a ="
line2 = const
self.replacer.parse_line(line1)
self.replacer.parse_line(line2)
line3 = "a + xyz"
output = self.replacer.parse_line(line3)
self.assertEqual('"abc def" + xyz', output)