mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-04-22 06:00:00 +08:00
8295979bb2
### What problem does this PR solve? delete SDK repo and edit readme ### Type of change - [x] New Feature (non-breaking change which adds functionality)
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
from test_sdkbase import TestSdk
|
|
import ragflow
|
|
from ragflow.ragflow import RAGFLow
|
|
import pytest
|
|
from unittest.mock import MagicMock
|
|
|
|
|
|
class TestCase(TestSdk):
|
|
|
|
@pytest.fixture
|
|
def ragflow_instance(self):
|
|
# Here we create a mock instance of RAGFlow for testing
|
|
return ragflow.ragflow.RAGFLow('123', 'url')
|
|
|
|
def test_version(self):
|
|
print(ragflow.__version__)
|
|
|
|
def test_create_dataset(self):
|
|
assert ragflow.ragflow.RAGFLow('123', 'url').create_dataset('abc') == 'abc'
|
|
|
|
def test_delete_dataset(self):
|
|
assert ragflow.ragflow.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
|