diff options
| author | 尹姜谊 <[email protected]> | 2024-11-08 14:42:50 +0800 |
|---|---|---|
| committer | 尹姜谊 <[email protected]> | 2024-11-08 14:42:50 +0800 |
| commit | c057aff33d329f918bad57c8de5705f841a9495e (patch) | |
| tree | 15c25b451afde91775fea215c7c2be0f4e6f806c /test/test_knowledgebase_tool.py | |
| parent | ee5a5dba40817632ed32d8d86313bb45def60100 (diff) | |
| parent | c0d48d9b8b55926fcaf38c7a126c67ea01e03dbf (diff) | |
修改：适配24.08版本知识库库表结构变化24.08
Diffstat (limited to 'test/test_knowledgebase_tool.py')
| -rw-r--r-- | test/test_knowledgebase_tool.py | 8 |
1 files changed, 6 insertions, 2 deletions
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 |