#standard libraries import unittest, doctest #local libraries from pyogp.lib.base.data import msg_tmpl from pyogp.lib.base.message.message_template import MessageTemplate, MessageTemplateBlock, MessageTemplateVariable from pyogp.lib.base.message.message_template_parser import MessageTemplateParser from pyogp.lib.base.message.message_template_builder import MessageTemplateBuilder from pyogp.lib.base.message.message_template_dict import TemplateDictionary class TestTemplateReader(unittest.TestCase): def tearDown(self): pass def setUp(self): self.template_file = msg_tmpl parser = MessageTemplateParser(self.template_file) self.template_list = parser.message_templates self.template_dict = TemplateDictionary(self.template_list) self.builder = MessageTemplateBuilder(self.template_dict) def test_suite(): from unittest import TestSuite, makeSuite suite = TestSuite() suite.addTest(makeSuite(TestTemplateReader)) return suite