Enhance API response handling and update datetime formatting

- Improved the GitHub API client to handle empty response content more robustly by checking for whitespace.
- Updated the `format_datetime` function to accept both datetime objects and strings, allowing for more flexible input handling.
- Refactored test cases to run synchronously, ensuring compatibility with the updated testing framework.

Author: SakuraPuare

github_stats_analyzer/api.py

@@ -110,7 +110,7 @@ async def github_request(self, method: str, endpoint: str, **kwargs) -> Tuple[in
 
                 # Return response
                 if response.status_code < 400:
-                    return response.status_code, response.json() if response.content else None
+                    return response.status_code, response.json() if response.content and response.content.strip() else None
                 else:
                     logger.warning(f"GitHub API error: {response.status_code} - {response.text}")
                     return response.status_code, None

github_stats_analyzer/utils.py

@@ -53,11 +53,11 @@ def should_exclude_repo(repo_name: str, languages: Dict[str, int], excluded_lang
 
     return False
 
-def format_datetime(dt: Optional[datetime], output_format: str = "%Y-%m-%d") -> str:
-    """Format a datetime object to a string.
+def format_datetime(dt: Optional[datetime | str], output_format: str = "%Y-%m-%d") -> str:
+    """Format a datetime object or string to a string.
     
     Args:
-        dt: Datetime object to format
+        dt: Datetime object or string to format
         output_format: Output format string
         
     Returns:
@@ -67,6 +67,11 @@ def format_datetime(dt: Optional[datetime], output_format: str = "%Y-%m-%d") ->
         return "Unknown"
 
     try:
+        # If dt is already a string, return it directly
+        if isinstance(dt, str):
+            return dt
+        
+        # If dt is a datetime object, format it
         return dt.strftime(output_format)
     except Exception as e:
         logger.warning(f"Could not format date: {e}")

tests/run_all_tests.py

@@ -29,8 +29,9 @@ def run_unit_tests():
     suite = unittest.TestSuite()
 
     # 添加测试用例
-    suite.addTest(unittest.makeSuite(TestGitHubStatsAnalyzer))
-    suite.addTest(unittest.makeSuite(TestGitHubAPIClient))
+    loader = unittest.TestLoader()
+    suite.addTest(loader.loadTestsFromTestCase(TestGitHubStatsAnalyzer))
+    suite.addTest(loader.loadTestsFromTestCase(TestGitHubAPIClient))
 
     # 运行测试
     runner = unittest.TextTestRunner(verbosity=2)

tests/test_analyzer.py

@@ -61,7 +61,7 @@ def test_init_with_params(self, mock_api_client):
     @patch('github_stats_analyzer.analyzer.GitHubStatsAnalyzer.fetch_user_repos')
     @patch('github_stats_analyzer.analyzer.GitHubStatsAnalyzer.process_repos')
     @patch('github_stats_analyzer.analyzer.GitHubStatsAnalyzer.calculate_language_percentages')
-    async def test_analyze(self, mock_calc, mock_process, mock_fetch):
+    def test_analyze(self, mock_calc, mock_process, mock_fetch):
         """测试分析方法"""
         # 创建模拟仓库列表
         mock_repos = [
@@ -85,8 +85,9 @@ async def test_analyze(self, mock_calc, mock_process, mock_fetch):
             excluded_languages=self.excluded_languages
         )
 
-        # 调用分析方法
-        await analyzer.analyze()
+        # 创建事件循环并运行异步方法
+        loop = asyncio.get_event_loop()
+        loop.run_until_complete(analyzer.analyze())
 
         # 验证方法调用
         mock_fetch.assert_called_once()
@@ -196,11 +197,12 @@ def test_init(self, mock_client):
         self.assertIsNotNone(client.client)
 
     @patch('httpx.AsyncClient.get')
-    async def test_get_user_repos_basic(self, mock_get):
+    def test_get_user_repos_basic(self, mock_get):
         """测试获取用户仓库 (基本访问级别)"""
         # 创建模拟响应
         mock_response = MagicMock()
         mock_response.status_code = 200
+        mock_response.content = b'{"some": "content"}'
         mock_response.json.return_value = [
             {
                 "name": "repo1",
@@ -217,8 +219,9 @@ async def test_get_user_repos_basic(self, mock_get):
         # 创建客户端实例
         client = GitHubAPIClient(AccessLevel.BASIC)
 
-        # 调用方法
-        repos = await client.get_user_repos("test-user")
+        # 创建事件循环并运行异步方法
+        loop = asyncio.get_event_loop()
+        repos = loop.run_until_complete(client.get_user_repos("test-user"))
 
         # 验证结果
         self.assertEqual(len(repos), 1)