From 9bcbf880bb5252b023797ee1efd371ee9c34fd50 Mon Sep 17 00:00:00 2001 From: 尹姜谊 Date: Tue, 16 Apr 2024 20:28:05 +0800 Subject: Modify: 适配24.04知识库 MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- test/test_knowledgebase_tool.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'test/test_knowledgebase_tool.py') diff --git a/test/test_knowledgebase_tool.py b/test/test_knowledgebase_tool.py index f809872..4e9544e 100644 --- a/test/test_knowledgebase_tool.py +++ b/test/test_knowledgebase_tool.py @@ -14,7 +14,11 @@ class TestKnowledgeApi(unittest.TestCase): def setUp(self): config = Config().config - self.knowledgebase_tool = KnowledgeApi(config) + self.knowledgebase_tool = KnowledgeApi(config['knowledgebase']) def test_get_knowledgebase(self): - print(self.knowledgebase_tool.get_knowledgebase_count(knowledge_id=6, q="c_time >= '2023-10-26 08:12:47' AND addr_format = 'Single'")) + print(self.knowledgebase_tool.get_knowledgebase_count(knowledge_id=16, q="c_time >= '2023-10-26 08:12:47' AND addr_format = 'Single'")) + + def test_file_import(self): + self.knowledgebase_tool.file_import('/Users/joy/work/iie/project/cyber_narrator/CN/3-新功能研发/vpn-detection/3.项目代码/CN-Code/vpn-finder-plugins/tmp_ip.csv', + action='update') \ No newline at end of file -- cgit v1.2.3