Test getting job status.
(mock_get)
| 173 | |
| 174 | @patch('simstudio.requests.Session.get') |
| 175 | def test_get_job_status_success(mock_get): |
| 176 | """Test getting job status.""" |
| 177 | mock_response = Mock() |
| 178 | mock_response.ok = True |
| 179 | mock_response.json.return_value = { |
| 180 | "success": True, |
| 181 | "taskId": "task-123", |
| 182 | "status": "completed", |
| 183 | "metadata": { |
| 184 | "startedAt": "2024-01-01T00:00:00Z", |
| 185 | "completedAt": "2024-01-01T00:01:00Z", |
| 186 | "duration": 60000 |
| 187 | }, |
| 188 | "output": {"result": "done"} |
| 189 | } |
| 190 | mock_response.headers.get.return_value = None |
| 191 | mock_get.return_value = mock_response |
| 192 | |
| 193 | client = SimStudioClient(api_key="test-api-key", base_url="https://test.sim.ai") |
| 194 | result = client.get_job_status("task-123") |
| 195 | |
| 196 | assert result["taskId"] == "task-123" |
| 197 | assert result["status"] == "completed" |
| 198 | assert result["output"]["result"] == "done" |
| 199 | mock_get.assert_called_once_with("https://test.sim.ai/api/jobs/task-123") |
| 200 | |
| 201 | |
| 202 | @patch('simstudio.requests.Session.get') |
nothing calls this directly
no test coverage detected