31 lines
727 B
Python
31 lines
727 B
Python
from server.knowledge_base.kb_service.pg_kb_service import PGKBService
|
|
from server.knowledge_base.migrate import create_tables
|
|
from server import KnowledgeFile
|
|
|
|
kbService = PGKBService("test")
|
|
|
|
test_kb_name = "test"
|
|
test_file_name = "../../knowledge_base/test/content/README.md"
|
|
testKnowledgeFile = KnowledgeFile(test_file_name, test_kb_name)
|
|
search_content = "如何启动api服务"
|
|
|
|
|
|
def test_init():
|
|
create_tables()
|
|
|
|
|
|
def test_create_db():
|
|
assert kbService.create_kb()
|
|
|
|
|
|
def test_add_doc():
|
|
assert kbService.add_doc(testKnowledgeFile)
|
|
|
|
|
|
def test_search_db():
|
|
result = kbService.search_docs(search_content)
|
|
assert len(result) > 0
|
|
def test_delete_doc():
|
|
assert kbService.delete_doc(testKnowledgeFile)
|
|
|