replace the the pretty_print_table call with rich.table.Table in testcases

Author: opswithranjan

cloudsmith_cli/core/download.py

@@ -8,6 +8,8 @@
 import click
 import cloudsmith_api
 import requests
+from rich.console import Console
+from rich.table import Table
 
 from . import keyring, ratelimits, utils
 from .api.exceptions import catch_raise_api_exception
@@ -295,9 +297,6 @@ def resolve_package(
 
 def _display_multiple_packages(packages: List[Dict]) -> None:
     """Display a table of multiple matching packages."""
-    from rich.console import Console
-    from rich.table import Table
-
     click.echo("Multiple packages found:")
     click.echo()
 

cloudsmith_cli/core/tests/test_download.py

@@ -107,10 +107,10 @@ def test_resolve_package_multiple_matches_yes(
         mock_select_best.assert_called_once_with(mock_packages)
 
     @patch("cloudsmith_cli.core.download.list_packages")
-    @patch("cloudsmith_cli.cli.utils.pretty_print_table")
+    @patch("cloudsmith_cli.core.download.Console")
     @patch("click.echo")
     def test_resolve_package_multiple_matches_no_yes(
-        self, mock_echo, mock_pretty_print, mock_list_packages
+        self, mock_echo, mock_console_cls, mock_list_packages
     ):
         """Test package resolution with multiple matches without --yes."""
         mock_packages = [
@@ -127,7 +127,7 @@ def test_resolve_package_multiple_matches_no_yes(
             download.resolve_package("owner", "repo", "test-package", yes=False)
 
         self.assertEqual(cm.exception.exit_code, 3)
-        mock_pretty_print.assert_called_once()
+        mock_console_cls().print.assert_called_once()
 
     @patch("cloudsmith_cli.core.download.list_packages")
     def test_resolve_package_with_filters(self, mock_list_packages):
@@ -677,9 +677,12 @@ def test_resolve_package_error_mentions_download_all(self, mock_list_packages):
 class TestDisplayMultiplePackages(unittest.TestCase):
     """Test _display_multiple_packages function."""
 
-    @patch("cloudsmith_cli.cli.utils.pretty_print_table")
+    @patch("cloudsmith_cli.core.download.Console")
+    @patch("cloudsmith_cli.core.download.Table")
     @patch("click.echo")
-    def test_display_includes_filename_column(self, mock_echo, mock_table):
+    def test_display_includes_filename_column(
+        self, mock_echo, mock_table_cls, mock_console_cls
+    ):
         """Test that the multiple packages table includes filename."""
         packages = [
             {
@@ -694,9 +697,11 @@ def test_display_includes_filename_column(self, mock_echo, mock_table):
 
         download._display_multiple_packages(packages)
 
-        mock_table.assert_called_once()
-        headers = mock_table.call_args[0][0]
-        self.assertIn("Filename", headers)
+        # Verify that add_column was called with "Filename"
+        column_names = [
+            call.args[0] for call in mock_table_cls().add_column.call_args_list
+        ]
+        self.assertIn("Filename", column_names)
 
 
 if __name__ == "__main__":