test_kg_node_service.py 784 B

1234567891011121314151617181920212223242526272829
  1. import pytest
  2. from service.kg_node_service import KGNodeService
  3. from model.kg_node import KGNode
  4. from sqlalchemy.exc import IntegrityError
  5. @pytest.fixture(scope="module")
  6. def kg_node_service():
  7. from db.session import get_db
  8. return KGNodeService(next(get_db()))
  9. @pytest.fixture
  10. def test_node_data():
  11. return {
  12. "name": "测试节点",
  13. "category": "测试类别",
  14. "version": "1.0"
  15. }
  16. class TestPaginatedSearch:
  17. def test_paginated_search(self, kg_node_service, test_node_data):
  18. results = kg_node_service.paginated_search({
  19. 'keyword': '咳嗽',
  20. 'pageNo': 1,
  21. 'limit': 10
  22. })
  23. print(results)
  24. assert len(results['records']) > 0
  25. assert results['pagination']['pageNo'] == 1