import io

from rich.console import Console
from rich.text import Text


class Foo:
    def __rich__(self):
        return Text("Foo")


def test_rich_case():
    foo = Foo()
    console = Console(file=io.StringIO())
    console.print(foo)
    assert console.file.getvalue() == "Foo\n"