Spaces:
Sleeping
Sleeping
| import sys | |
| import os | |
| import unittest | |
| from unittest.mock import patch, MagicMock | |
| # 添加 src 目录到模块搜索路径，以便可以导入 src 目录中的模块 | |
| sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../src'))) | |
| from github_client import GitHubClient # 导入要测试的 GitHubClient 类 | |
| class TestGitHubClient(unittest.TestCase): | |
| def setUp(self): | |
| """ | |
| 在每个测试方法之前运行，初始化测试环境。 | |
| """ | |
| self.token = "fake_token" # 使用一个虚拟的 GitHub API 令牌 | |
| self.client = GitHubClient(self.token) # 使用该令牌初始化 GitHubClient 实例 | |
| self.repo = "DjangoPeng/openai-quickstart" # 要测试的仓库名称 | |
| def test_fetch_commits(self, mock_get): | |
| """ | |
| 测试 fetch_commits 方法是否正确获取提交记录。 | |
| """ | |
| # 模拟 API 响应 | |
| mock_response = MagicMock() | |
| mock_response.json.return_value = [{"sha": "abc123", "commit": {"message": "Initial commit"}}] | |
| mock_response.status_code = 200 | |
| mock_get.return_value = mock_response # 将模拟的响应赋值给 mock_get | |
| # 调用 fetch_commits 方法并进行断言检查 | |
| commits = self.client.fetch_commits(self.repo) | |
| self.assertEqual(len(commits), 1) # 检查返回的提交记录数量是否为 1 | |
| self.assertEqual(commits[0]['sha'], "abc123") # 检查返回的提交记录 SHA 值 | |
| self.assertEqual(commits[0]['commit']['message'], "Initial commit") # 检查提交记录中的消息 | |
| def test_fetch_issues(self, mock_get): | |
| """ | |
| 测试 fetch_issues 方法是否正确获取关闭的问题。 | |
| """ | |
| # 模拟 API 响应 | |
| mock_response = MagicMock() | |
| mock_response.json.return_value = [{"number": 1, "title": "Fix bug"}] | |
| mock_response.status_code = 200 | |
| mock_get.return_value = mock_response # 将模拟的响应赋值给 mock_get | |
| # 调用 fetch_issues 方法并进行断言检查 | |
| issues = self.client.fetch_issues(self.repo) | |
| self.assertEqual(len(issues), 1) # 检查返回的关闭问题数量是否为 1 | |
| self.assertEqual(issues[0]['number'], 1) # 检查问题编号是否正确 | |
| self.assertEqual(issues[0]['title'], "Fix bug") # 检查问题标题是否正确 | |
| def test_fetch_pull_requests(self, mock_get): | |
| """ | |
| 测试 fetch_pull_requests 方法是否正确获取拉取请求。 | |
| """ | |
| # 模拟 API 响应 | |
| mock_response = MagicMock() | |
| mock_response.json.return_value = [{"number": 42, "title": "Add new feature"}] | |
| mock_response.status_code = 200 | |
| mock_get.return_value = mock_response # 将模拟的响应赋值给 mock_get | |
| # 调用 fetch_pull_requests 方法并进行断言检查 | |
| pull_requests = self.client.fetch_pull_requests(self.repo) | |
| self.assertEqual(len(pull_requests), 1) # 检查返回的拉取请求数量是否为 1 | |
| self.assertEqual(pull_requests[0]['number'], 42) # 检查拉取请求的编号是否正确 | |
| self.assertEqual(pull_requests[0]['title'], "Add new feature") # 检查拉取请求的标题是否正确 | |
| def test_export_daily_progress(self, mock_get): | |
| """ | |
| 测试 export_daily_progress 方法是否正确导出每日进度报告。 | |
| """ | |
| # 模拟 API 响应 | |
| mock_response = MagicMock() | |
| mock_response.json.return_value = [] | |
| mock_response.status_code = 200 | |
| mock_get.return_value = mock_response # 将模拟的响应赋值给 mock_get | |
| # 调用 export_daily_progress 方法并进行断言检查 | |
| file_path = self.client.export_daily_progress(self.repo) | |
| self.assertTrue(file_path.endswith('.md')) # 检查生成的文件路径是否以 .md 结尾 | |
| def test_export_progress_by_date_range(self, mock_get): | |
| """ | |
| 测试 export_progress_by_date_range 方法是否正确导出指定日期范围内的进度报告。 | |
| """ | |
| # 模拟 API 响应 | |
| mock_response = MagicMock() | |
| mock_response.json.return_value = [] | |
| mock_response.status_code = 200 | |
| mock_get.return_value = mock_response # 将模拟的响应赋值给 mock_get | |
| # 调用 export_progress_by_date_range 方法并进行断言检查 | |
| file_path = self.client.export_progress_by_date_range(self.repo, days=7) | |
| self.assertTrue(file_path.endswith('.md')) # 检查生成的文件路径是否以 .md 结尾 | |
| if __name__ == '__main__': | |
| unittest.main() | |