28 lines
719 B
Python
28 lines
719 B
Python
import unittest
|
|
from utils.Macro import Macro
|
|
|
|
|
|
class TestMacro(unittest.TestCase):
|
|
def setUp(self):
|
|
self.macroParser = Macro()
|
|
|
|
def test_parseMacros(self):
|
|
line = 'hello {@this} is {@my_macro} and {@macro_with_numbers123}'
|
|
symbols = self.macroParser.getSymbols(line)
|
|
|
|
self.assertEqual(3, len(symbols))
|
|
self.assertTrue('this' in symbols)
|
|
self.assertTrue('my_macro' in symbols)
|
|
self.assertTrue('macro_with_numbers123' in symbols)
|
|
|
|
def test_getName(self):
|
|
line = '{@macro_name}'
|
|
name = self.macroParser.getMacroName(line)
|
|
|
|
self.assertEqual('macro_name', name)
|
|
|
|
def test_getMacro(self):
|
|
line = 'some_name'
|
|
macro = self.macroParser.getMacroByName(line)
|
|
|
|
self.assertEqual('{@some_name}', macro) |