Fix-6906: Added code-highlight option to disable highlighting optionally

code-highlight has default value to True, this value is used in `TerminalWriter`
in order to control highlighting of the code.

Author: gdhameeja

src/_pytest/_io/__init__.py

@@ -36,4 +36,6 @@ def _highlight(self, source):
         except ImportError:
             return source
         else:
-            return highlight(source, PythonLexer(), TerminalFormatter(bg="dark"))
+            if self.code_highlight:
+                return highlight(source, PythonLexer(), TerminalFormatter(bg="dark"))
+            return source

src/_pytest/config/__init__.py

@@ -1215,6 +1215,11 @@ def create_terminal_writer(config: Config, *args, **kwargs) -> TerminalWriter:
         tw.hasmarkup = True
     if config.option.color == "no":
         tw.hasmarkup = False
+
+    if config.option.code_highlight == "no":
+        tw.code_highlight = False
+    else:
+        tw.code_highlight = True
     return tw
 
 

src/_pytest/terminal.py

@@ -161,6 +161,14 @@ def pytest_addoption(parser):
         choices=["yes", "no", "auto"],
         help="color terminal output (yes/no/auto).",
     )
+    group._addoption(
+        "--code-highlight",
+        action="store",
+        dest="code_highlight",
+        default="yes",
+        choices=["yes", "no"],
+        help="Whether code should be highlighted on the terminal or not",
+    )
 
     parser.addini(
         "console_output_style",

testing/code/test_terminal_writer.py

@@ -18,6 +18,7 @@ def test_code_highlight(has_markup, expected, color_mapping):
     f = StringIO()
     tw = TerminalWriter(f)
     tw.hasmarkup = has_markup
+    tw.code_highlight = True
     tw._write_source(["assert 0"])
     assert f.getvalue().splitlines(keepends=True) == color_mapping.format([expected])
 

testing/test_terminal.py

@@ -2138,3 +2138,24 @@ def test_foo():
                 ]
             )
         )
+
+    def test_code_highlight_false(self, testdir: Testdir, color_mapping) -> None:
+        testdir.makepyfile(
+            """
+            def test_foo():
+                print('''
+                '''); assert 0
+        """
+        )
+        result = testdir.runpytest("--code-highlight=no")
+        color_mapping.requires_ordered_markup(result)
+        result.stdout.fnmatch_lines(
+            color_mapping.format_for_fnmatch(
+                [
+                    "    {kw}def{hl-reset} {function}test_foo{hl-reset}():",
+                    "        {print}print{hl-reset}({str}'''{hl-reset}{str}{hl-reset}",
+                    ">   {str}    {hl-reset}{str}'''{hl-reset}); {kw}assert{hl-reset} {number}0{hl-reset}",
+                    "assert 0{reset}",
+                ]
+            )
+        )