fix: show tags (#1058)

* fix: show tags in listing

Signed-off-by: Henry Schreiner <henryfs@princeton.edu>

* refactor: simply a bit

Signed-off-by: Henry Schreiner <henryfs@princeton.edu>

* tests: update for tags

Signed-off-by: Henry Schreiner <henryfs@princeton.edu>

---------

Signed-off-by: Henry Schreiner <henryfs@princeton.edu>

Author: henryiii

nox/tasks.py

@@ -273,28 +273,19 @@ def _produce_listing(manifest: Manifest, global_config: Namespace) -> None:
     skipped_color = parse_colors("white") if global_config.color else ""
 
     for session, selected in manifest.list_all_sessions():
-        output = "{marker} {color}{session}{reset}"
-
-        if selected:
-            marker = "*"
-            color = selected_color
-        else:
-            marker = "-"
-            color = skipped_color
-
-        if session.description is not None:
-            output += " -> {description}"
-
-        print(
-            output.format(
-                color=color,
-                reset=reset,
-                session=session.friendly_name,
-                description=session.description,
-                marker=marker,
-            )
+        marker = "*" if selected else "-"
+        color = selected_color if selected else skipped_color
+        tag_color = (
+            (parse_colors("purple") if selected else parse_colors("light_purple"))
+            if global_config.color
+            else ""
         )
 
+        tags = f" {tag_color}[{','.join(session.tags)}]{reset}" if session.tags else ""
+        description = f" -> {session.description}" if session.description else ""
+
+        print(f"{marker} {color}{session.friendly_name}{reset}{tags}{description}")
+
     print(
         f"\nsessions marked with {selected_color}*{reset} are selected, sessions marked"
         f" with {skipped_color}-{reset} are skipped."

tests/test_tasks.py

@@ -451,7 +451,10 @@ def test_honor_list_request(
     manifest = mock.create_autospec(Manifest)
     manifest.module_docstring = module_docstring
     manifest.list_all_sessions.return_value = [
-        (argparse.Namespace(friendly_name="foo", description=description), True)
+        (
+            argparse.Namespace(friendly_name="foo", description=description, tags=[]),
+            True,
+        )
     ]
     return_value = tasks.honor_list_request(manifest, global_config=config)
     assert return_value == 0
@@ -466,8 +469,8 @@ def test_honor_list_request_skip_and_selected(
     manifest = mock.create_autospec(Manifest)
     manifest.module_docstring = None
     manifest.list_all_sessions.return_value = [
-        (argparse.Namespace(friendly_name="foo", description=None), True),
-        (argparse.Namespace(friendly_name="bar", description=None), False),
+        (argparse.Namespace(friendly_name="foo", description=None, tags=[]), True),
+        (argparse.Namespace(friendly_name="bar", description=None, tags=[]), False),
     ]
     return_value = tasks.honor_list_request(manifest, global_config=config)
     assert return_value == 0
@@ -487,8 +490,8 @@ def test_honor_list_request_prints_docstring_if_present(
     manifest = mock.create_autospec(Manifest)
     manifest.module_docstring = "Hello I'm a docstring"
     manifest.list_all_sessions.return_value = [
-        (argparse.Namespace(friendly_name="foo", description=None), True),
-        (argparse.Namespace(friendly_name="bar", description=None), False),
+        (argparse.Namespace(friendly_name="foo", description=None, tags=[]), True),
+        (argparse.Namespace(friendly_name="bar", description=None, tags=[]), False),
     ]
 
     return_value = tasks.honor_list_request(manifest, global_config=config)
@@ -508,8 +511,8 @@ def test_honor_list_request_doesnt_print_docstring_if_not_present(
     manifest = mock.create_autospec(Manifest)
     manifest.module_docstring = None
     manifest.list_all_sessions.return_value = [
-        (argparse.Namespace(friendly_name="foo", description=None), True),
-        (argparse.Namespace(friendly_name="bar", description=None), False),
+        (argparse.Namespace(friendly_name="foo", description=None, tags=[]), True),
+        (argparse.Namespace(friendly_name="bar", description=None, tags=[]), False),
     ]
 
     return_value = tasks.honor_list_request(manifest, global_config=config)