WIP: multiple colors with terminal summary_stats

Ref: pytest-dev#5060

Author: blueyed

src/_pytest/terminal.py

@@ -864,14 +864,40 @@ def _outrep_summary(self, rep):
 
     def summary_stats(self):
         session_duration = time.time() - self._sessionstarttime
-        (line, color) = build_summary_stats_line(self.stats)
-        msg = "%s in %.2f seconds" % (line, session_duration)
-        markup = {color: True, "bold": True}
+        (parts, main_color) = build_summary_stats_line(self.stats)
+        line_parts = []
+        fullwidth = self._tw.fullwidth
+        for text, markup in parts:
+            with_markup = self._tw.markup(text, **markup)
+            fullwidth += len(with_markup) - len(text)
+            line_parts.append(with_markup)
+        msg = ", ".join(line_parts)
+
+        main_markup = {main_color: True, "bold": True}
+        duration = " in %.2f seconds" % session_duration
+        duration_with_markup = self._tw.markup(duration, **main_markup)
+        fullwidth += len(duration_with_markup) - len(duration)
+        msg += duration_with_markup
+
+        markup_for_end_sep = self._tw.markup("", **main_markup)
+        if markup_for_end_sep.endswith("\x1b[0m"):
+            markup_for_end_sep = markup_for_end_sep[:-4]
+        fullwidth += len(markup_for_end_sep)
+        msg += markup_for_end_sep
 
         if self.verbosity >= 0:
-            self.write_sep("=", msg, **markup)
+            self.write_sep("=", msg, fullwidth=fullwidth, **main_markup)
         if self.verbosity == -1:
-            self.write_line(msg, **markup)
+            self.write_line(msg, **main_markup)
+
+
+_color_for_type = {
+    "failed": "red",
+    "error": "red",
+    "warnings": "yellow",
+    "passed": "green",
+}
+_color_for_type_default = "yellow"
 
 
 def build_summary_stats_line(stats):
@@ -884,30 +910,32 @@ def build_summary_stats_line(stats):
             if found_type:  # setup/teardown reports have an empty key, ignore them
                 known_types.append(found_type)
                 unknown_type_seen = True
+
+    # main color
+    if "failed" in stats or "error" in stats:
+        main_color = "red"
+    elif "warnings" in stats or unknown_type_seen:
+        main_color = "yellow"
+    elif "passed" in stats:
+        main_color = "green"
+    else:
+        main_color = "yellow"
+
     parts = []
     for key in known_types:
         reports = stats.get(key, None)
         if reports:
             count = sum(
                 1 for rep in reports if getattr(rep, "count_towards_summary", True)
             )
-            parts.append("%d %s" % (count, key))
-
-    if parts:
-        line = ", ".join(parts)
-    else:
-        line = "no tests ran"
+            color = _color_for_type.get(key, _color_for_type_default)
+            markup = {color: True, "bold": color == main_color}
+            parts.append(("%d %s" % (count, key), markup))
 
-    if "failed" in stats or "error" in stats:
-        color = "red"
-    elif "warnings" in stats or unknown_type_seen:
-        color = "yellow"
-    elif "passed" in stats:
-        color = "green"
-    else:
-        color = "yellow"
+    if not parts:
+        parts = [("no tests ran", {_color_for_type_default: True})]
 
-    return line, color
+    return parts, main_color
 
 
 def _plugin_nameversions(plugininfo):