test: update widget tests for APKG export components

tests/widgets_test.py

@@ -93,31 +93,30 @@ def test_voice_map_instruct_strings_format():
 
 
 def test_enable_phase2_returns_tuple(_mock_gradio):
-    """_enable_phase2() returns tuple of (Checkbox, Checkbox, Button, Dropdown, Button, CSS).
-
-    Export button is VISIBLE but DISABLED here — it is enabled after Phase 2 completes
-    (when _current_cards is populated) via _on_media_generation_complete().
-    """
+    """_enable_phase2() returns tuple of (Checkbox, Checkbox, Button, Dropdown, Button, Button, File, File, CSS)."""
     from frontend.ui.widgets import _enable_phase2
 
     result = _enable_phase2()
     assert isinstance(result, tuple)
-    assert len(result) == 6
+    assert len(result) == 9
 
 
 def test_reset_to_idle_returns_tuple(_mock_gradio):
-    """_reset_to_idle() returns tuple with interactive=False, disabled CSS string, export button, and file component."""
+    """_reset_to_idle() returns tuple with interactive=False, disabled CSS string, export buttons, and file components."""
     from frontend.ui.widgets import _reset_to_idle
 
     result = _reset_to_idle()
     assert isinstance(result, tuple)
-    assert len(result) == 8
+    assert len(result) == 10
     # Element at index 5 should be a CSS string (non-empty)
     assert isinstance(result[5], str)
     assert len(result[5]) > 0
-    # Elements at indices 6, 7 should be the export button and file component mocks
-    assert isinstance(result[6], MagicMock)  # export_btn
-    assert isinstance(result[7], MagicMock)  # export_file
+    # Elements at indices 6, 7 should be the export buttons
+    assert isinstance(result[6], MagicMock)  # export_csv_btn
+    assert isinstance(result[7], MagicMock)  # export_apkg_btn
+    # Elements at indices 8, 9 should be the file components
+    assert isinstance(result[8], MagicMock)  # export_file
+    assert isinstance(result[9], MagicMock)  # export_apkg_file
 
 
 def test_reset_to_idle_disabled_css_content(_mock_gradio):
@@ -125,6 +124,7 @@ def test_reset_to_idle_disabled_css_content(_mock_gradio):
     from frontend.ui.widgets import _reset_to_idle
 
     result = _reset_to_idle()
+    # CSS is at index 5 (after generate_text_btn, images_toggle, audio_toggle, generate_cards_btn, voice_dropdown)
     css = result[5]
     assert "europalex-btn-disabled" in css or "toggle-images" in css
     assert "#voice-dropdown" in css
@@ -153,12 +153,12 @@ def test_enable_language_dropdown_on_audio_false(_mock_gradio):
 
 
 def test_restore_generate_cards_button_returns_tuple(_mock_gradio):
-    """_restore_generate_cards_button() returns tuple of (Button, Button)."""
+    """_restore_generate_cards_button() returns tuple of (Button, Button, Button)."""
     from frontend.ui.widgets import _restore_generate_cards_button
 
     result = _restore_generate_cards_button()
     assert isinstance(result, tuple)
-    assert len(result) == 2
+    assert len(result) == 3
 
 
 def test_restore_generate_cards_button_makes_button_visible_interactive(_mock_gradio):
@@ -166,33 +166,38 @@ def test_restore_generate_cards_button_makes_button_visible_interactive(_mock_gr
     from frontend.ui.widgets import _restore_generate_cards_button
 
     _restore_generate_cards_button()
-    # gr.Button was called twice: first for generate_cards_btn, then for export_btn
+    # gr.Button was called three times: generate_cards_btn, export_csv_btn, export_apkg_btn
     calls = _mock_gradio.Button.call_args_list
-    assert len(calls) == 2
+    assert len(calls) == 3
     first_call_kwargs = calls[0].kwargs if calls[0].kwargs else calls[0][1]
     assert first_call_kwargs.get("visible") is True
     assert first_call_kwargs.get("interactive") is True
 
 
 def test_restore_generate_cards_button_export_stays_disabled(_mock_gradio):
-    """Export button stays visible but disabled."""
+    """Export buttons stay visible but disabled."""
     from frontend.ui.widgets import _restore_generate_cards_button
 
     _restore_generate_cards_button()
     calls = _mock_gradio.Button.call_args_list
-    assert len(calls) == 2
+    assert len(calls) == 3
+    # Second call: export_csv_btn (disabled)
     second_call_kwargs = calls[1].kwargs if calls[1].kwargs else calls[1][1]
     assert second_call_kwargs.get("visible") is True
     assert second_call_kwargs.get("interactive") is False
+    # Third call: export_apkg_btn (disabled)
+    third_call_kwargs = calls[2].kwargs if calls[2].kwargs else calls[2][1]
+    assert third_call_kwargs.get("visible") is True
+    assert third_call_kwargs.get("interactive") is False
 
 
 def test_restore_generate_cards_button_only_returns_tuple(_mock_gradio):
-    """_restore_generate_cards_button_only() returns tuple of 3 (Button, Button, Button)."""
+    """_restore_generate_cards_button_only() returns tuple of 4 (Button, Button, Button, Button)."""
     from frontend.ui.widgets import _restore_generate_cards_button_only
 
     result = _restore_generate_cards_button_only()
     assert isinstance(result, tuple)
-    assert len(result) == 3
+    assert len(result) == 4
 
 
 def test_restore_generate_cards_button_only_hides_generate_text(_mock_gradio):
@@ -201,7 +206,7 @@ def test_restore_generate_cards_button_only_hides_generate_text(_mock_gradio):
 
     _restore_generate_cards_button_only()
     calls = _mock_gradio.Button.call_args_list
-    assert len(calls) == 3
+    assert len(calls) == 4
     first_call_kwargs = calls[0].kwargs if calls[0].kwargs else calls[0][1]
     assert first_call_kwargs.get("visible") is False
 
@@ -212,7 +217,7 @@ def test_restore_generate_cards_button_only_restores_generate_cards(_mock_gradio
 
     _restore_generate_cards_button_only()
     calls = _mock_gradio.Button.call_args_list
-    assert len(calls) == 3
+    assert len(calls) == 4
     second_call_kwargs = calls[1].kwargs if calls[1].kwargs else calls[1][1]
     assert second_call_kwargs.get("visible") is True
     assert second_call_kwargs.get("interactive") is True