36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import pytest
|
|
from src.gui_2 import App
|
|
from src.models import FileItem
|
|
|
|
def test_view_mode_initialization():
|
|
app = App()
|
|
# Mock imgui
|
|
from imgui_bundle import imgui
|
|
|
|
app.context_files = [
|
|
FileItem(path="test1.py"),
|
|
FileItem(path="test2.py", view_mode="full")
|
|
]
|
|
|
|
# We test the model defaults and the rendering assignment logic indirectly.
|
|
assert app.context_files[0].view_mode == "full" # Default from FileItem Model
|
|
assert app.context_files[1].view_mode == "full"
|
|
|
|
def test_batch_view_mode_change():
|
|
app = App()
|
|
|
|
f1 = FileItem(path="test1.py", view_mode="full")
|
|
f2 = FileItem(path="test2.py", view_mode="full")
|
|
|
|
app.context_files = [f1, f2]
|
|
app.ui_selected_context_files = {"test1.py"}
|
|
|
|
# Simulate clicking "Skeleton" batch button
|
|
for f in app.context_files:
|
|
f_path = f.path if hasattr(f, "path") else str(f)
|
|
if f_path in app.ui_selected_context_files:
|
|
f.view_mode = "skeleton"
|
|
|
|
assert f1.view_mode == "skeleton"
|
|
assert f2.view_mode == "full" # not selected
|