def __init__(self, aiml_parser):
self._aiml_parser = aiml_parser
template_nodes = aiml_parser.brain.configuration.nodes.template_nodes
self._template_factory = TemplateNodeFactory()
self._template_factory.load_nodes_config_from_file(template_nodes)
def assert_upload_from_file(self, store):
store.upload_from_file(
os.path.dirname(__file__) + os.sep + "data" + os.sep + "nodes" +
os.sep + "template_nodes.conf")
collection = TemplateNodeFactory()
store.load(collection)
self.assertEqual(64, len(collection.nodes))
self.assertTrue(collection.exists("lowercase"))
def __init__(self, aiml_parser=None):
self._aiml_parser = aiml_parser
self.node_lookups = None
template_nodes = None
if aiml_parser is not None:
if aiml_parser._brain is not None:
template_nodes = aiml_parser._brain.configuration.nodes.template_nodes
self._template_factory = TemplateNodeFactory()
self._template_factory.load_nodes_config_from_file(template_nodes)
def test_load_variables(self):
config = FileStorageConfiguration()
config._template_nodes_storage = FileStoreConfiguration(
file=os.path.dirname(__file__) + os.sep + "data" + os.sep +
"nodes" + os.sep + "template_nodes.conf",
fileformat="text",
encoding="utf-8",
delete_on_start=False)
engine = FileStorageEngine(config)
engine.initialise()
store = FileTemplateNodeStore(engine)
collection = TemplateNodeFactory()
store.load(collection)
self.assertEqual(64, len(collection.nodes))
self.assertTrue(collection.exists("lowercase"))
def test_init(self):
factory = TemplateNodeFactory()
self.assertIsNotNone(factory)
self.assertEquals({}, factory._nodes_config)
self.assertEqual("Template", factory._type)
def test_load_nodes_config_from_file_invalid_filename(self):
factory = TemplateNodeFactory()
factory.load_nodes_config_from_file("some_rubbish.txt")
self.assertEquals(54, len(factory._nodes_config))
self.assert_nodes(factory)
def test_load_nodes_config_from_file(self):
factory = TemplateNodeFactory()
factory.load_nodes_config_from_file(os.path.dirname(__file__) + os.sep + "pattern_nodes.conf")
self.assert_nodes(factory)
def load_template_node_factory(self):
template_nodes = self._aiml_parser.brain.configuration.nodes.template_nodes
self._template_factory = TemplateNodeFactory()
self._template_factory.load_nodes_config_from_file(template_nodes)
def _load_template_nodes(self):
self._template_factory = TemplateNodeFactory()
self._template_factory.load(self.bot.client.storage_factory)
