Spaces:
Sleeping
Sleeping
| """Tests for data models.""" | |
| import pytest | |
| from models import Issue, SearchResult, CodeChange, TestResult, WorkflowResult | |
| def test_issue_creation(): | |
| """Test Issue model creation.""" | |
| issue = Issue( | |
| number=123, | |
| title="Test Issue", | |
| body="Test body", | |
| state="open", | |
| labels=["bug"], | |
| author="testuser", | |
| url="https://github.com/test/repo/issues/123", | |
| ) | |
| assert issue.number == 123 | |
| assert issue.title == "Test Issue" | |
| assert "bug" in issue.labels | |
| def test_search_result_creation(): | |
| """Test SearchResult model creation.""" | |
| result = SearchResult( | |
| file_path="src/main.py", | |
| content="def main():\n pass", | |
| relevance_score=0.95, | |
| ) | |
| assert result.file_path == "src/main.py" | |
| assert result.relevance_score == 0.95 | |
| def test_code_change_creation(): | |
| """Test CodeChange model creation.""" | |
| change = CodeChange( | |
| file_path="src/main.py", | |
| original_content="old code", | |
| new_content="new code", | |
| explanation="Fixed bug", | |
| ) | |
| assert change.file_path == "src/main.py" | |
| assert change.explanation == "Fixed bug" | |
| def test_test_result_creation(): | |
| """Test TestResult model creation.""" | |
| result = TestResult( | |
| success=True, | |
| output="All tests passed", | |
| exit_code=0, | |
| duration=5.2, | |
| ) | |
| assert result.success is True | |
| assert result.duration == 5.2 | |
| def test_workflow_result_creation(): | |
| """Test WorkflowResult model creation.""" | |
| result = WorkflowResult( | |
| success=True, | |
| branch_name="ai-fix/issue-123", | |
| files_changed=["src/main.py"], | |
| pr_url="https://github.com/test/repo/pull/1", | |
| ) | |
| assert result.success is True | |
| assert len(result.files_changed) == 1 | |
| assert result.pr_url is not None | |
| def test_workflow_result_with_error(): | |
| """Test WorkflowResult with error.""" | |
| result = WorkflowResult( | |
| success=False, | |
| error="Something went wrong", | |
| ) | |
| assert result.success is False | |
| assert result.error == "Something went wrong" | |
| assert result.files_changed == [] | |