def testGetFormatterInitCreate(self): """test the render for creating a new formatter init file""" mapper = formatter_init_mapping.FormatterInitMapping( fake_mapping_helper.FakeMappingHelper("template path")) data = init_data_model.InitDataModel(self.plugin_name, True) actual = mapper.GetRenderedTemplate(data) expected = "fake string formatter_init_template.jinja2" self.assertEqual(expected, actual)
def testGetParserInitEdit(self): """test the render for editing a existing parser init file""" mapper = parser_init_mapping.ParserInitMapping( fake_mapping_helper.FakeMappingHelper("template path")) data = init_data_model.InitDataModel(self.plugin_name, False) actual = mapper.GetRenderedTemplate(data) expected = "fake string parser_init_template.jinja2" self.assertEqual(expected, actual)
def testGenerateSQLitePluginWithExistingInit(self): """test the output of a generation of a sqlite plugin""" fake_handler = fake_file_handler.FakeFileHandler() mapping_helper = fake_mapping_helper.FakeMappingHelper( self.template_path) init_formatter_mapper = fake_sqliteplugin_mapping.FakeSQLitePluginMapper( mapping_helper) init_parser_mapper = fake_sqliteplugin_mapping.FakeSQLitePluginMapper( mapping_helper) parser_mapper = fake_sqliteplugin_mapping.FakeSQLitePluginMapper( mapping_helper) formatter_mapper = fake_sqliteplugin_mapping.FakeSQLitePluginMapper( mapping_helper) parser_test_mapper = fake_sqliteplugin_mapping.FakeSQLitePluginMapper( mapping_helper) formatter_test_mapper = ( fake_sqliteplugin_mapping.FakeSQLitePluginMapper(mapping_helper)) fake_database_information = ( fake_sqlite_database_information.FakeSQLiteDatabaseInformation([])) with tempfile.TemporaryDirectory() as tmpdir: fake_path_helper = ( fake_sqlite_plugin_path_helper.FakeSQLitePluginPathHelper( self.template_path, 'test', 'db')) path = os.path.join(tmpdir, 'testfile') generator = sqlite_generator.SQLiteGenerator( tmpdir, 'test', 'test', [], output_handler_file.OutputHandlerFile( path, file_handler.FileHandler()), self.plugin_helper, fake_path_helper) generator.init_formatter_exists = True generator.init_parser_exists = True generator.GenerateSQLitePlugin( tmpdir, fake_handler, init_formatter_mapper, init_parser_mapper, parser_mapper, formatter_mapper, parser_test_mapper, formatter_test_mapper, mapping_helper, fake_database_information) expected = ( 'create testcreate testcreate testcreate testcopy testedit ' 'testedit test') actual = self._ReadFromFile(path) self.assertEqual(expected, actual)