import pytest
from unittest.mock import MagicMock, patch, call
from src import gui_2
from src.gui_2 import App

@pytest.fixture
def app_instance():
 with (
  patch('src.app_controller.AppController.load_config', return_value={'ai': {'provider': 'gemini', 'model': 'gemini-2.5-flash-lite'}, 'projects': {}}),
  patch('src.app_controller.AppController.save_config'),
  patch('src.gui_2.project_manager'),
  patch('src.gui_2.session_logger'),
  patch('src.gui_2.immapp.run'),
  patch('src.app_controller.AppController._load_active_project'),
  patch('src.app_controller.AppController._fetch_models'),
  patch.object(App, '_load_fonts'),
  patch.object(App, '_post_init'),
  patch('src.app_controller.AppController._prune_old_logs'),
  patch('src.app_controller.AppController.start_services'),
  patch('src.api_hooks.HookServer'),
  patch('src.ai_client.set_provider'),
  patch('src.ai_client.reset_session')
 ):
  app = App()
  app.project = {
   "discussion": {
    "active": "main",
    "discussions": {
     "main": {"history": []},
     "main_take_1": {"history": []},
     "main_take_2": {"history": []}
    }
   }
  }
  app.active_discussion = "main"
  app.is_viewing_prior_session = False
  app.ui_disc_new_name_input = ""
  app.ui_disc_truncate_pairs = 1
  yield app

def test_render_discussion_tabs(app_instance):
 """Verify that _render_discussion_panel uses tabs for discussions."""
 with patch('src.gui_2.imgui') as mock_imgui, \
  patch('src.gui_2.imscope') as mock_imscope:
  mock_imgui.collapsing_header.return_value = True
  mock_imgui.begin_combo.return_value = False
  mock_imgui.input_text.return_value = (False, "")
  mock_imgui.input_int.return_value = (False, 0)
  mock_imgui.drag_int.return_value = (False, 0)
  mock_imgui.input_text_multiline.return_value = (False, "")
  mock_imgui.button.return_value = False
  mock_imgui.checkbox.return_value = (False, False)
  mock_imgui.begin_child.return_value = True
  mock_imgui.selectable.return_value = (False, False)
  mock_imgui.ListClipper.return_value.step.return_value = False
  mock_imgui.begin_tab_bar.return_value = True

  mock_imscope.window.return_value.__enter__.return_value = (True, True)
  mock_imscope.child.return_value.__enter__.return_value = True
  mock_imscope.table.return_value.__enter__.return_value = True
  mock_imscope.tree_node_ex.return_value.__enter__.return_value = True
  mock_imscope.tab_item.return_value.__enter__.return_value = (True, True)
  mock_imscope.style_color.return_value.__enter__.return_value = None
  mock_imscope.style_var.return_value.__enter__.return_value = None

  mock_imgui.begin_tab_item.return_value = (True, True)

  gui_2.render_discussion_panel(app_instance)

  mock_imgui.begin_tab_bar.assert_called_with("discussion_takes_tabs")
  assert mock_imscope.tab_item.call_count >= 3, f"Expected at least 3 tab items via imscope.tab_item, got {mock_imscope.tab_item.call_count}"

def test_switching_discussion_via_tabs(app_instance):
 """Verify that clicking a tab switches the discussion."""
 with patch('src.gui_2.imgui') as mock_imgui, \
  patch('src.gui_2.imscope') as mock_imscope, \
  patch('src.app_controller.AppController._switch_discussion') as mock_switch:
  mock_imgui.collapsing_header.return_value = True
  mock_imgui.begin_combo.return_value = False
  mock_imgui.input_text.return_value = (False, "")
  mock_imgui.input_int.return_value = (False, 0)
  mock_imgui.drag_int.return_value = (False, 0)
  mock_imgui.input_text_multiline.return_value = (False, "")
  mock_imgui.button.return_value = False
  mock_imgui.checkbox.return_value = (False, False)
  mock_imgui.begin_child.return_value = True
  mock_imgui.selectable.return_value = (False, False)
  mock_imgui.ListClipper.return_value.step.return_value = False
  mock_imgui.begin_tab_bar.return_value = True

  mock_imscope.window.return_value.__enter__.return_value = (True, True)
  mock_imscope.child.return_value.__enter__.return_value = True
  mock_imscope.table.return_value.__enter__.return_value = True
  mock_imscope.tree_node_ex.return_value.__enter__.return_value = True
  mock_imscope.tab_item.return_value.__enter__.return_value = (True, True)
  mock_imscope.style_color.return_value.__enter__.return_value = None
  mock_imscope.style_var.return_value.__enter__.return_value = None

  def begin_tab_item_side_effect(label, p_open=None, flags=None):
   if "main_take_1" in label:
    return (True, True)
   return (False, True)

  mock_imgui.begin_tab_item.side_effect = begin_tab_item_side_effect

  gui_2.render_discussion_panel(app_instance)

  assert mock_switch.called, f"Expected _switch_discussion to be called"
