72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
from unittest.mock import MagicMock, patch
|
|
|
|
from aider_gitea.gitea_client import GiteaClient
|
|
|
|
|
|
class TestGiteaClientPRComments:
|
|
def setup_method(self):
|
|
"""Set up test fixtures."""
|
|
self.client = GiteaClient('https://gitea.example.com', 'fake-token')
|
|
self.owner = 'test-owner'
|
|
self.repo = 'test-repo'
|
|
self.pull_number = 123
|
|
|
|
@patch('requests.Session.get')
|
|
def test_get_pull_request_comments(self, mock_get):
|
|
"""Test retrieving comments for a pull request."""
|
|
# Mock response data
|
|
mock_response = MagicMock()
|
|
mock_response.json.return_value = [
|
|
{
|
|
'id': 1,
|
|
'body': 'This is a test comment',
|
|
'user': {'login': 'test-user'},
|
|
'created_at': '2023-01-01T00:00:00Z',
|
|
},
|
|
{
|
|
'id': 2,
|
|
'body': 'Another test comment',
|
|
'user': {'login': 'another-user'},
|
|
'created_at': '2023-01-02T00:00:00Z',
|
|
},
|
|
]
|
|
mock_get.return_value = mock_response
|
|
|
|
# Call the method
|
|
comments = self.client.get_pull_request_comments(
|
|
self.owner, self.repo, self.pull_number,
|
|
)
|
|
|
|
# Verify the request was made correctly
|
|
mock_get.assert_called_once_with(
|
|
f'{self.client.gitea_url}/repos/{self.owner}/{self.repo}/pulls/{self.pull_number}/comments',
|
|
)
|
|
|
|
# Verify the response was processed correctly
|
|
assert len(comments) == 2
|
|
assert comments[0]['id'] == 1
|
|
assert comments[0]['body'] == 'This is a test comment'
|
|
assert comments[1]['id'] == 2
|
|
assert comments[1]['body'] == 'Another test comment'
|
|
|
|
@patch('requests.Session.get')
|
|
def test_get_pull_request_comments_empty(self, mock_get):
|
|
"""Test retrieving comments for a pull request with no comments."""
|
|
# Mock response data
|
|
mock_response = MagicMock()
|
|
mock_response.json.return_value = []
|
|
mock_get.return_value = mock_response
|
|
|
|
# Call the method
|
|
comments = self.client.get_pull_request_comments(
|
|
self.owner, self.repo, self.pull_number,
|
|
)
|
|
|
|
# Verify the request was made correctly
|
|
mock_get.assert_called_once_with(
|
|
f'{self.client.gitea_url}/repos/{self.owner}/{self.repo}/pulls/{self.pull_number}/comments',
|
|
)
|
|
|
|
# Verify the response was processed correctly
|
|
assert len(comments) == 0
|