diff options
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 |