ragflow/sdk/python/test/test_basic.py

from test_sdkbase import TestSdk
import ragflow
from ragflow.ragflow import RAGFLow
import pytest
from unittest.mock import MagicMock
from common import API_KEY, HOST_ADDRESS


class TestBasic(TestSdk):

    def test_version(self):
        print(ragflow.__version__)

    # def test_create_dataset(self):
    #     res = RAGFLow(API_KEY, HOST_ADDRESS).create_dataset('abc')
    #     print(res)
    #
    # def test_delete_dataset(self):
    #     assert RAGFLow('123', 'url').delete_dataset('abc') == 'abc'
    #
    # def test_list_dataset_success(self, ragflow_instance, monkeypatch):
    #     # Mocking the response of requests.get method
    #     mock_response = MagicMock()
    #     mock_response.status_code = 200
    #     mock_response.json.return_value = {'datasets': [{'id': 1, 'name': 'dataset1'}, {'id': 2, 'name': 'dataset2'}]}
    #
    #     # Patching requests.get to return the mock_response
    #     monkeypatch.setattr("requests.get", MagicMock(return_value=mock_response))
    #
    #     # Call the method under test
    #     result = ragflow_instance.list_dataset()
    #
    #     # Assertion
    #     assert result == [{'id': 1, 'name': 'dataset1'}, {'id': 2, 'name': 'dataset2'}]
    #
    # def test_list_dataset_failure(self, ragflow_instance, monkeypatch):
    #     # Mocking the response of requests.get method
    #     mock_response = MagicMock()
    #     mock_response.status_code = 404  # Simulating a failed request
    #
    #     # Patching requests.get to return the mock_response
    #     monkeypatch.setattr("requests.get", MagicMock(return_value=mock_response))
    #
    #     # Call the method under test
    #     result = ragflow_instance.list_dataset()
    #
    #     # Assertion
    #     assert result is None