class TestElasticSearchOutput(unittest.TestCase, ProjectTest): INDEX = "test_index" def setUp(self): self.output_type = "ElasticSearchOutput" self.output_args = { 'source': 'localhost', 'index': TestElasticSearchOutput.INDEX, 'content_field': "abstract" } self.project = TopikProject("test_project", output_type=self.output_type, output_args=self.output_args) self.project.read_input(test_data_path, content_field="abstract", synchronous_wait=30) def tearDown(self): instance = elasticsearch.Elasticsearch("localhost") instance.indices.delete(TestElasticSearchOutput.INDEX) if instance.indices.exists("{}_year_alias_date".format( TestElasticSearchOutput.INDEX)): instance.indices.delete("{}_year_alias_date".format( TestElasticSearchOutput.INDEX)) time.sleep(1)
class TestInMemoryOutput(unittest.TestCase, ProjectTest): def setUp(self): self.output_type = "InMemoryOutput" self.output_args = {} self.project = TopikProject("test_project", output_type=self.output_type, output_args=self.output_args) self.project.read_input(test_data_path, content_field="abstract")
class TestElasticSearchOutput(unittest.TestCase, ProjectTest): INDEX = "test_index" def setUp(self): self.output_type = "ElasticSearchOutput" self.output_args = {'source': 'localhost', 'index': TestElasticSearchOutput.INDEX, 'content_field': "abstract"} self.project = TopikProject("test_project", output_type=self.output_type, output_args=self.output_args) self.project.read_input(test_data_path, content_field="abstract", synchronous_wait=30) def tearDown(self): instance = elasticsearch.Elasticsearch("localhost") instance.indices.delete(TestElasticSearchOutput.INDEX) if instance.indices.exists("{}_year_alias_date".format(TestElasticSearchOutput.INDEX)): instance.indices.delete("{}_year_alias_date".format(TestElasticSearchOutput.INDEX)) time.sleep(1)