fix: empty plugins sections repr (#1738)

ApeWorX · Nov 17, 2023 · f2511ae · f2511ae
1 parent e06bec7
commit f2511ae
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/src/ape_plugins/utils.py b/src/ape_plugins/utils.py
@@ -441,6 +441,9 @@ def __str__(self) -> str:
         if PluginType.AVAILABLE in self.include and self.metadata.available:
             sections.append(self.metadata.available)
 
+        if not sections:
+            return ""
+
         # Use a single max length for all the sections.
         max_length = max(x.max_name_length for x in sections)
 

diff --git a/tests/functional/test_plugins.py b/tests/functional/test_plugins.py
@@ -115,8 +115,8 @@ def test_is_available(self):
 
 class TestApePluginsRepr:
     def test_str(self, plugin_metadata):
-        plugin_map = ApePluginsRepr(plugin_metadata)
-        actual = str(plugin_map)
+        representation = ApePluginsRepr(plugin_metadata)
+        actual = str(representation)
         expected = f"""
 Installed Plugins
   installed     {VERSION}
@@ -127,8 +127,8 @@ def test_str(self, plugin_metadata):
         assert actual == expected.strip()
 
     def test_str_all_types(self, plugin_metadata):
-        plugin_map = ApePluginsRepr(plugin_metadata, include=list(PluginType))
-        actual = str(plugin_map)
+        representation = ApePluginsRepr(plugin_metadata, include=list(PluginType))
+        actual = str(representation)
         expected = f"""
 Core Plugins
   run
@@ -143,3 +143,8 @@ def test_str_all_types(self, plugin_metadata):
   available
         """
         assert actual == expected.strip()
+
+    def test_str_no_plugins(self):
+        plugins = PluginMetadataList.from_package_names([])
+        representation = ApePluginsRepr(plugins)
+        assert str(representation) == ""