"""Tests for rich module."""
import sys

from subprocess_tee import run
from subprocess_tee.rich import ConsoleEx


def test_rich_console_ex() -> None:
    """Validate that ConsoleEx can capture output from print() calls."""
    console = ConsoleEx(record=True, redirect=True)
    console.print("alpha")
    print("beta")
    sys.stdout.write("gamma\n")
    sys.stderr.write("delta\n")
    proc = run("echo 123")
    assert proc.stdout == "123\n"
    text = console.export_text()
    assert text == "alpha\nbeta\ngamma\ndelta\n123\n"

"""Tests for rich module."""
import sys

from subprocess_tee import run
from subprocess_tee.rich import ConsoleEx


def test_rich_console_ex() -> None:
    """Validate that ConsoleEx can capture output from print() calls."""
    console = ConsoleEx(record=True, redirect=True)
    console.print("alpha")
    print("beta")
    sys.stdout.write("gamma\n")
    sys.stderr.write("delta\n")
    proc = run("echo 123")
    assert proc.stdout == "123\n"
    text = console.export_text()
    assert text == "alpha\nbeta\ngamma\ndelta\n123\n"


def test_rich_console_ex_ansi() -> None:
    """Validate that ANSI sent to sys.stdout does not become garbage in record."""
    print()
    console = ConsoleEx(force_terminal=True, record=True, redirect=True)
    console.print("[green]this from Console.print()[/green]", style="red")
    proc = run(r'echo -e "\033[31mred\033[0m"')
    assert proc.returncode == 0
    assert proc.stdout == "\x1b[31mred\x1b[0m\n"

    # validate that what rich recorded is the same as what the subprocess produced
    text = console.export_text(clear=False)
    assert "\x1b[31mred\x1b[0m\n" in text

    # validate that html export also contains at least the "red" text
    html = console.export_html(clear=False)
    assert "red" in html

Add testing of rich html_export
Add testing of rich html_export

  """Tests for rich module."""
  import sys
  
  from subprocess_tee import run
  from subprocess_tee.rich import ConsoleEx
  
  
  def test_rich_console_ex() -> None:
      """Validate that ConsoleEx can capture output from print() calls."""
      console = ConsoleEx(record=True, redirect=True)
      console.print("alpha")
      print("beta")
      sys.stdout.write("gamma\n")
      sys.stderr.write("delta\n")
      proc = run("echo 123")
      assert proc.stdout == "123\n"
      text = console.export_text()
      assert text == "alpha\nbeta\ngamma\ndelta\n123\n"
  
+ 
+ def test_rich_console_ex_ansi() -> None:
+     """Validate that ANSI sent to sys.stdout does not become garbage in record."""
+     print()
+     console = ConsoleEx(force_terminal=True, record=True, redirect=True)
+     console.print("[green]this from Console.print()[/green]", style="red")
+     proc = run(r'echo -e "\033[31mred\033[0m"')
+     assert proc.returncode == 0
+     assert proc.stdout == "\x1b[31mred\x1b[0m\n"
+ 
+     # validate that what rich recorded is the same as what the subprocess produced
+     text = console.export_text(clear=False)
+     assert "\x1b[31mred\x1b[0m\n" in text
+ 
+     # validate that html export also contains at least the "red" text
+     html = console.export_html(clear=False)
+     assert "red" in html
+ 
Add testing of rich html_export
## Code Before:
"""Tests for rich module."""
import sys

from subprocess_tee import run
from subprocess_tee.rich import ConsoleEx


def test_rich_console_ex() -> None:
    """Validate that ConsoleEx can capture output from print() calls."""
    console = ConsoleEx(record=True, redirect=True)
    console.print("alpha")
    print("beta")
    sys.stdout.write("gamma\n")
    sys.stderr.write("delta\n")
    proc = run("echo 123")
    assert proc.stdout == "123\n"
    text = console.export_text()
    assert text == "alpha\nbeta\ngamma\ndelta\n123\n"

## Instruction:
Add testing of rich html_export
## Code After:
"""Tests for rich module."""
import sys

from subprocess_tee import run
from subprocess_tee.rich import ConsoleEx


def test_rich_console_ex() -> None:
    """Validate that ConsoleEx can capture output from print() calls."""
    console = ConsoleEx(record=True, redirect=True)
    console.print("alpha")
    print("beta")
    sys.stdout.write("gamma\n")
    sys.stderr.write("delta\n")
    proc = run("echo 123")
    assert proc.stdout == "123\n"
    text = console.export_text()
    assert text == "alpha\nbeta\ngamma\ndelta\n123\n"


def test_rich_console_ex_ansi() -> None:
    """Validate that ANSI sent to sys.stdout does not become garbage in record."""
    print()
    console = ConsoleEx(force_terminal=True, record=True, redirect=True)
    console.print("[green]this from Console.print()[/green]", style="red")
    proc = run(r'echo -e "\033[31mred\033[0m"')
    assert proc.returncode == 0
    assert proc.stdout == "\x1b[31mred\x1b[0m\n"

    # validate that what rich recorded is the same as what the subprocess produced
    text = console.export_text(clear=False)
    assert "\x1b[31mred\x1b[0m\n" in text

    # validate that html export also contains at least the "red" text
    html = console.export_html(clear=False)
    assert "red" in html
