def test_simplequery_whitelist_work_test_es_result(self): dummy_doc_generate = DummyDocumentsGenerate() nbr_generated_documents = 5 all_doc = dummy_doc_generate.create_documents(nbr_generated_documents) whitelisted_document = dummy_doc_generate.generate_document( {"hostname": "whitelist_hostname"}) all_doc.append(whitelisted_document) self.test_es.add_multiple_docs(all_doc) # Run analyzer self.test_settings.change_configuration_path( config_file_simplequery_test_whitelist) analyzer = AnalyzerFactory.create(use_case_simplequery_dummy_test) analyzer.evaluate_model() nbr_outliers = 0 for elem in es._scan(): if "outliers" in elem["_source"]: nbr_outliers += 1 self.assertEqual(nbr_outliers, nbr_generated_documents)
def _generate_documents(self, nbr_generate): dummy_doc_gen = DummyDocumentsGenerate() all_doc = dummy_doc_gen.create_documents(nbr_generate) self.test_es.add_multiple_docs(all_doc)