fix SEA test

Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>

Author: vikrantpuppala

scripts/dependency_manager.py

@@ -9,6 +9,7 @@
 import sys
 import argparse
 from packaging.specifiers import SpecifierSet
+from packaging.requirements import Requirement
 from pathlib import Path
 
 class DependencyManager:
@@ -170,13 +171,13 @@ def generate_requirements(self, version_type="min", include_optional=False):
 
         # First add main requirements
         for req in requirements:
-            package_name = req.split('>=')[0].split('==')[0].split('<')[0]
+            package_name = Requirement(req).name
             seen_packages.add(package_name)
             final_requirements.append(req)
 
         # Then add transitive requirements that don't conflict
         for req in transitive_requirements:
-            package_name = req.split('>=')[0].split('==')[0].split('<')[0]
+            package_name = Requirement(req).name
             if package_name not in seen_packages:
                 final_requirements.append(req)
 

tests/unit/test_sea_http_client.py

@@ -96,7 +96,8 @@ def test_make_request_success(self, mock_get_auth_headers, sea_http_client):
         # Setup mock response
         mock_response = Mock()
         mock_response.status = 200
-        mock_response.json.return_value = {"result": "success"}
+        # Mock response.data.decode() to return a valid JSON string
+        mock_response.data.decode.return_value = '{"result": "success"}'
         mock_response.__enter__ = Mock(return_value=mock_response)
         mock_response.__exit__ = Mock(return_value=None)