Untitled

from pro_filer.actions.main_actions import show_preview  # NOQA

def test_show_preview(capsys):
    context = {
        "all_files": ["src/__init__.py", "src/app.py", "src/utils/__init__.py", "sample/foo/foo.py", "sample/bar/bar.py"],
        "all_dirs": ["src", "src/utils", "sample", "sample/foo", "sample/bar"]
    }

    show_preview(context)

    assert context["all_files"] == ["src/__init__.py", "src/app.py", "src/utils/__init__.py", "sample/foo/foo.py", "sample/bar/bar.py"]
    assert context["all_dirs"] == ["src", "src/utils", "sample", "sample/foo", "sample/bar"]
    assert len(context["all_files"]) == 5
    assert len(context["all_dirs"]) == 5
    
    context_message = (
        f"Found 5 files and 5 directories\n"
        f"First 5 files: ['src/__init__.py', 'src/app.py', 'src/utils/__init__.py', 'sample/foo/foo.py', 'sample/bar/bar.py']\n"
        f"First 5 directories: ['src', 'src/utils', 'sample', 'sample/foo', 'sample/bar']\n"
    )

    captured = capsys.readouterr()

    assert captured.out == context_message

def test_show_preview_with_empty_context(capsys):
    context = {
        "all_files": [],
        "all_dirs": []
    }


    show_preview(context)

    assert context["all_files"] == []
    assert context["all_dirs"] == []
    assert len(context["all_files"]) == 0
    assert len(context["all_dirs"]) == 0

    context_message = f'Found 0 files and 0 directories\n'

    captured = capsys.readouterr()

    assert captured.out == context_message

def test_show_preview_with_more_than_5_files_and_dirs(capsys):
    context = {
        "all_files": ["src/__init__.py", "src/app.py", "src/utils/__init__.py", "src/utils/other.py", "src/utils/other2.py", "src/utils/other3.py"],
        "all_dirs": ["src", "src/utils"]
    }

    show_preview(context)

    assert context["all_files"] == ["src/__init__.py", "src/app.py", "src/utils/__init__.py", "src/utils/other.py", "src/utils/other2.py", "src/utils/other3.py"]
    assert context["all_dirs"] == ["src", "src/utils"]
    assert len(context["all_files"]) == 6
    assert len(context["all_dirs"]) == 2

    context_message = (
        f"Found 6 files and 2 directories\n"
        f"First 5 files: ['src/__init__.py', 'src/app.py', 'src/utils/__init__.py', 'src/utils/other.py', 'src/utils/other2.py']\n"
        f"First 5 directories: ['src', 'src/utils']\n"
    )

    captured = capsys.readouterr()

    assert captured.out == context_message