mirror of
https://github.com/simstudioai/sim.git
synced 2026-01-14 09:27:58 -05:00
ae43381d84
* feat(domain): drop the * change all references for Sim Studio to Sim * change back license and notice * lint --------- Co-authored-by: waleedlatif <waleedlatif@waleedlatifs-MacBook-Pro.local>
97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
"""
|
|
Tests for the Sim Python SDK
|
|
"""
|
|
|
|
import pytest
|
|
from unittest.mock import Mock, patch
|
|
from simstudio import SimStudioClient, SimStudioError, WorkflowExecutionResult, WorkflowStatus
|
|
|
|
|
|
def test_simstudio_client_initialization():
|
|
"""Test SimStudioClient initialization."""
|
|
client = SimStudioClient(api_key="test-api-key", base_url="https://test.sim.ai")
|
|
assert client.api_key == "test-api-key"
|
|
assert client.base_url == "https://test.sim.ai"
|
|
|
|
|
|
def test_simstudio_client_default_base_url():
|
|
"""Test SimStudioClient with default base URL."""
|
|
client = SimStudioClient(api_key="test-api-key")
|
|
assert client.api_key == "test-api-key"
|
|
assert client.base_url == "https://sim.ai"
|
|
|
|
|
|
def test_set_api_key():
|
|
"""Test setting a new API key."""
|
|
client = SimStudioClient(api_key="test-api-key")
|
|
client.set_api_key("new-api-key")
|
|
assert client.api_key == "new-api-key"
|
|
|
|
|
|
def test_set_base_url():
|
|
"""Test setting a new base URL."""
|
|
client = SimStudioClient(api_key="test-api-key")
|
|
client.set_base_url("https://new.sim.ai/")
|
|
assert client.base_url == "https://new.sim.ai"
|
|
|
|
|
|
def test_set_base_url_strips_trailing_slash():
|
|
"""Test that base URL strips trailing slash."""
|
|
client = SimStudioClient(api_key="test-api-key")
|
|
client.set_base_url("https://test.sim.ai/")
|
|
assert client.base_url == "https://test.sim.ai"
|
|
|
|
|
|
@patch('simstudio.requests.Session.get')
|
|
def test_validate_workflow_returns_false_on_error(mock_get):
|
|
"""Test that validate_workflow returns False when request fails."""
|
|
mock_get.side_effect = SimStudioError("Network error")
|
|
|
|
client = SimStudioClient(api_key="test-api-key")
|
|
result = client.validate_workflow("test-workflow-id")
|
|
|
|
assert result is False
|
|
mock_get.assert_called_once_with("https://sim.ai/api/workflows/test-workflow-id/status")
|
|
|
|
|
|
def test_simstudio_error():
|
|
"""Test SimStudioError creation."""
|
|
error = SimStudioError("Test error", "TEST_CODE", 400)
|
|
assert str(error) == "Test error"
|
|
assert error.code == "TEST_CODE"
|
|
assert error.status == 400
|
|
|
|
|
|
def test_workflow_execution_result():
|
|
"""Test WorkflowExecutionResult data class."""
|
|
result = WorkflowExecutionResult(
|
|
success=True,
|
|
output={"data": "test"},
|
|
metadata={"duration": 1000}
|
|
)
|
|
assert result.success is True
|
|
assert result.output == {"data": "test"}
|
|
assert result.metadata == {"duration": 1000}
|
|
|
|
|
|
def test_workflow_status():
|
|
"""Test WorkflowStatus data class."""
|
|
status = WorkflowStatus(
|
|
is_deployed=True,
|
|
deployed_at="2023-01-01T00:00:00Z",
|
|
is_published=False,
|
|
needs_redeployment=False
|
|
)
|
|
assert status.is_deployed is True
|
|
assert status.deployed_at == "2023-01-01T00:00:00Z"
|
|
assert status.is_published is False
|
|
assert status.needs_redeployment is False
|
|
|
|
|
|
@patch('simstudio.requests.Session.close')
|
|
def test_context_manager(mock_close):
|
|
"""Test SimStudioClient as context manager."""
|
|
with SimStudioClient(api_key="test-api-key") as client:
|
|
assert client.api_key == "test-api-key"
|
|
# Should close without error
|
|
mock_close.assert_called_once() |