kiwi-code 0.0.42__tar.gz → 0.0.43__tar.gz

{kiwi_code-0.0.42 → kiwi_code-0.0.43}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kiwi-code
-Version: 0.0.42
+Version: 0.0.43
 Summary: A textual-based terminal user interface application
 Project-URL: Homepage, https://meetkiwi.ai
 Project-URL: Repository, https://github.com/jetoslabs/kiwi-code

{kiwi_code-0.0.42 → kiwi_code-0.0.43}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kiwi-code"
-version = "0.0.42"
+version = "0.0.43"
 description = "A textual-based terminal user interface application"
 readme = {file = "README.md", content-type = "text/markdown"}
 requires-python = ">=3.11,<4.0"

{kiwi_code-0.0.42 → kiwi_code-0.0.43}/src/kiwi_tui/screens/dashboard.py

@@ -138,6 +138,7 @@ class DashboardScreen(Screen):
         "69c2180355a89324a9926bc6",
         "6a0625b8ab5f80ba8ac5d012",
         "69e4be0d70c2d7a89197e66e",
+        "6a1c0773c519da8a0fa9b8a4",
     ]
 
     BINDINGS = [

{kiwi_code-0.0.42 → kiwi_code-0.0.43}/tests/test_tui_headless.py

@@ -701,7 +701,7 @@ async def test_tui_continue_run_updates_action_name_from_run(
 async def test_tui_autocode_select_shows_only_special_action_names(
     isolated_home: Path, monkeypatch: pytest.MonkeyPatch
  ) -> None:
-    """/autocode-select should open the picker with only the 3 special actions."""
+    """/autocode-select should open the picker with only the 4 special actions."""
     tokens_path = isolated_home / ".kiwi" / "tokens.json"
     tokens_path.parent.mkdir(parents=True, exist_ok=True)
     tokens_path.write_text(
@@ -720,6 +720,7 @@ async def test_tui_autocode_select_shows_only_special_action_names(
         "69c2180355a89324a9926bc6": "AutoCode Alpha",
         "6a0625b8ab5f80ba8ac5d012": "AutoCode Beta",
         "69e4be0d70c2d7a89197e66e": "AutoCode Gamma",
+        "6a1c0773c519da8a0fa9b8a4": "AutoCode Delta",
     }
 
     from kiwi_tui.screens.dashboard import DashboardScreen
@@ -753,10 +754,11 @@ async def test_tui_autocode_select_shows_only_special_action_names(
 
         assert type(app.screen).__name__ == "IdPickerScreen"
         table = app.screen.query_one("#idpicker-table", DataTable)
-        assert table.row_count == 3
+        assert table.row_count == 4
         assert list(table.get_row_at(0)) == ["1", "AutoCode Alpha"]
         assert list(table.get_row_at(1)) == ["2", "AutoCode Beta"]
         assert list(table.get_row_at(2)) == ["3", "AutoCode Gamma"]
+        assert list(table.get_row_at(3)) == ["4", "AutoCode Delta"]
 
 
 
@@ -853,8 +855,6 @@ async def test_tui_streaming_requests_autofollow_when_viewport_is_near_bottom(
         )
         await pilot.pause()
         assert calls == [{"after_refresh": True}]
-
-
 @pytest.mark.asyncio
 async def test_tui_streaming_does_not_force_scroll_when_user_reads_older_messages(
     isolated_home: Path,
@@ -881,45 +881,271 @@ async def test_tui_streaming_does_not_force_scroll_when_user_reads_older_message
         await pilot.pause()
         assert type(app.screen).__name__ == "DashboardScreen"
         screen = app.screen
-        messages = screen.query_one("#messages")
 
-        for i in range(30):
-            screen.add_message(
-                "\n".join(f"history {i} line {j}" for j in range(4)),
-                "assistant",
+        calls: list[dict] = []
+        monkeypatch = pytest.MonkeyPatch()
+        monkeypatch.setattr(screen, "_messages_is_near_bottom", lambda *args, **kwargs: False)
+        monkeypatch.setattr(
+            screen,
+            "_scroll_messages_to_end",
+            lambda *args, **kwargs: calls.append(dict(kwargs)),
+        )
+        try:
+            widget = screen.update_streaming_message({"blocks": [{"text": "stream start"}]})
+            await pilot.pause()
+            assert widget is not None
+            assert calls == []
+
+            screen.update_streaming_message(
+                {"blocks": [{"text": "\n".join(f"stream line {i}" for i in range(60))}]},
+                widget,
             )
+            await pilot.pause()
+            assert calls == []
+        finally:
+            monkeypatch.undo()
+
+@pytest.mark.asyncio
+async def test_tui_quits_cleanly(isolated_home: Path) -> None:
+    from kiwi_tui.main import AutobotsTUI
+
+    app = AutobotsTUI()
+    async with app.run_test() as pilot:
         await pilot.pause()
-        assert messages.max_scroll_y > 0
+        await pilot.press("ctrl+c")
+        await pilot.pause()
+    # Reaching here without an exception is the assertion.
 
-        widget = screen.update_streaming_message({"blocks": [{"text": "stream start"}]})
+
+@pytest.mark.asyncio
+async def test_tui_redirects_to_login_when_auth_expires(isolated_home: Path, monkeypatch: pytest.MonkeyPatch) -> None:
+    tokens_path = isolated_home / ".kiwi" / "tokens.json"
+    tokens_path.parent.mkdir(parents=True, exist_ok=True)
+    tokens_path.write_text(
+        json.dumps(
+            {
+                "access_token": "test-access-token",
+                "refresh_token": "test-refresh-token",
+                "token_type": "Bearer",
+                "expires_at": None,
+            }
+        ),
+        encoding="utf-8",
+    )
+
+    from autobots_client.api.actions import get_action_v1_actions_id_get
+    monkeypatch.setattr(
+        get_action_v1_actions_id_get,
+        "sync_detailed",
+        lambda *, id, client: SimpleNamespace(
+            status_code=200,
+            parsed=SimpleNamespace(field_id=id, name="AutoCode Default"),
+        ),
+    )
+
+    from kiwi_tui.main import AutobotsTUI
+
+    app = AutobotsTUI()
+    async with app.run_test() as pilot:
         await pilot.pause()
-        assert widget is not None
+        assert type(app.screen).__name__ == "DashboardScreen"
 
-        target_y = max(0, int(messages.max_scroll_y) - 5)
-        messages.scroll_to(y=target_y, animate=False, immediate=True)
+        app._handle_auth_expired("Session expired during test.")
         await pilot.pause()
-        y_before = int(messages.scroll_offset.y)
-        target_before = int(messages.scroll_target_y)
-        assert y_before == target_before
-        assert y_before < int(messages.max_scroll_y)
 
-        screen.update_streaming_message(
-            {"blocks": [{"text": "\n".join(f"stream line {i}" for i in range(60))}]},
-            widget,
+        assert type(app.screen).__name__ == "LoginScreen"
+        assert not tokens_path.exists()
+
+
+@pytest.mark.asyncio
+async def test_tui_run_action_403_redirects_to_login_when_refresh_cannot_run(
+    isolated_home: Path, monkeypatch: pytest.MonkeyPatch
+) -> None:
+    tokens_path = isolated_home / ".kiwi" / "tokens.json"
+    tokens_path.parent.mkdir(parents=True, exist_ok=True)
+    tokens_path.write_text(
+        json.dumps(
+            {
+                "access_token": "test-access-token",
+                "refresh_token": "test-refresh-token",
+                "token_type": "Bearer",
+                "expires_at": None,
+            }
+        ),
+        encoding="utf-8",
+    )
+
+    from autobots_client.api.actions import get_action_v1_actions_id_get
+    monkeypatch.setattr(
+        get_action_v1_actions_id_get,
+        "sync_detailed",
+        lambda *, id, client: SimpleNamespace(
+            status_code=200,
+            parsed=SimpleNamespace(field_id=id, name="AutoCode Default"),
+        ),
+    )
+
+    from kiwi_tui.main import AutobotsTUI
+
+    app = AutobotsTUI()
+    async with app.run_test() as pilot:
+        await pilot.pause()
+        assert type(app.screen).__name__ == "DashboardScreen"
+        screen = app.screen
+
+        monkeypatch.setattr(app, "_refresh_token_if_needed", lambda force=False: False)
+        monkeypatch.setattr(
+            app.autobots_client,
+            "run_action_async",
+            lambda *args, **kwargs: (False, None, "Failed to start action: status 403"),
         )
+
+        screen.run_action_with_polling("hello")
+        await pilot.pause(0.3)
+        await pilot.pause(0.3)
+
+        assert type(app.screen).__name__ == "LoginScreen"
+        assert not tokens_path.exists()
+
+
+@pytest.mark.asyncio
+async def test_tui_run_action_403_redirects_to_login_on_second_failure(
+    isolated_home: Path, monkeypatch: pytest.MonkeyPatch
+) -> None:
+    tokens_path = isolated_home / ".kiwi" / "tokens.json"
+    tokens_path.parent.mkdir(parents=True, exist_ok=True)
+    tokens_path.write_text(
+        json.dumps(
+            {
+                "access_token": "test-access-token",
+                "refresh_token": "test-refresh-token",
+                "token_type": "Bearer",
+                "expires_at": None,
+            }
+        ),
+        encoding="utf-8",
+    )
+
+    from autobots_client.api.actions import get_action_v1_actions_id_get
+    monkeypatch.setattr(
+        get_action_v1_actions_id_get,
+        "sync_detailed",
+        lambda *, id, client: SimpleNamespace(
+            status_code=200,
+            parsed=SimpleNamespace(field_id=id, name="AutoCode Default"),
+        ),
+    )
+
+    from kiwi_tui.main import AutobotsTUI
+
+    app = AutobotsTUI()
+    async with app.run_test() as pilot:
         await pilot.pause()
+        assert type(app.screen).__name__ == "DashboardScreen"
+        screen = app.screen
+
+        monkeypatch.setattr(app, "_refresh_token_if_needed", lambda force=False: True)
+        monkeypatch.setattr(
+            app.autobots_client,
+            "run_action_async",
+            lambda *args, **kwargs: (False, None, "Failed to start action: status 403"),
+        )
+
+        screen.run_action_with_polling("hello")
+        await pilot.pause(0.3)
+        await pilot.pause(0.3)
+
+        assert type(app.screen).__name__ == "LoginScreen"
+        assert not tokens_path.exists()
+
+
+def test_tui_force_refresh_uses_refresh_token_even_when_access_token_not_expired(
+    isolated_home: Path, monkeypatch: pytest.MonkeyPatch
+) -> None:
+    tokens_path = isolated_home / ".kiwi" / "tokens.json"
+    tokens_path.parent.mkdir(parents=True, exist_ok=True)
+    tokens_path.write_text(
+        json.dumps(
+            {
+                "access_token": "test-access-token",
+                "refresh_token": "test-refresh-token",
+                "token_type": "Bearer",
+                "expires_at": "2999-01-01T00:00:00",
+            }
+        ),
+        encoding="utf-8",
+    )
+
+    from kiwi_cli.models import AuthTokens
+    from kiwi_tui.main import AutobotsTUI
+
+    app = AutobotsTUI()
+
+    calls: list[str] = []
+
+    def _fake_refresh(refresh_token: str):
+        calls.append(refresh_token)
+        return True, AuthTokens(
+            access_token="new-access-token",
+            refresh_token="new-refresh-token",
+            token_type="Bearer",
+            expires_at=None,
+        ), "ok"
+
+    monkeypatch.setattr(app.autobots_client, "refresh_token", _fake_refresh)
+
+    assert app._refresh_token_if_needed(force=True) is True
+    assert calls == ["test-refresh-token"]
 
-        assert int(messages.scroll_offset.y) == y_before
-        assert int(messages.scroll_target_y) == target_before
-        assert int(messages.scroll_offset.y) < int(messages.max_scroll_y)
 
 @pytest.mark.asyncio
-async def test_tui_quits_cleanly(isolated_home: Path) -> None:
+async def test_tui_auth_expired_clears_dashboard_draft_and_messages(
+    isolated_home: Path, monkeypatch: pytest.MonkeyPatch
+) -> None:
+    tokens_path = isolated_home / ".kiwi" / "tokens.json"
+    tokens_path.parent.mkdir(parents=True, exist_ok=True)
+    tokens_path.write_text(
+        json.dumps(
+            {
+                "access_token": "test-access-token",
+                "refresh_token": "test-refresh-token",
+                "token_type": "Bearer",
+                "expires_at": None,
+            }
+        ),
+        encoding="utf-8",
+    )
+
+    from autobots_client.api.actions import get_action_v1_actions_id_get
+    monkeypatch.setattr(
+        get_action_v1_actions_id_get,
+        "sync_detailed",
+        lambda *, id, client: SimpleNamespace(
+            status_code=200,
+            parsed=SimpleNamespace(field_id=id, name="AutoCode Default"),
+        ),
+    )
+
+    from textual.containers import VerticalScroll
     from kiwi_tui.main import AutobotsTUI
+    from kiwi_tui.widgets import ChatInput
 
     app = AutobotsTUI()
     async with app.run_test() as pilot:
         await pilot.pause()
-        await pilot.press("ctrl+c")
+        assert type(app.screen).__name__ == "DashboardScreen"
+        dashboard = app.screen
+        chat_input = dashboard.query_one("#chat-input", ChatInput)
+        messages = dashboard.query_one("#messages", VerticalScroll)
+
+        chat_input.value = "hi"
+        dashboard.add_message("hi", "user")
         await pilot.pause()
-    # Reaching here without an exception is the assertion.
+        assert len(list(messages.children)) > 0
+
+        app._handle_auth_expired("Session expired during test.")
+        await pilot.pause()
+
+        assert chat_input.value == ""
+        assert len(list(messages.children)) == 0
+        assert type(app.screen).__name__ == "LoginScreen"

{kiwi_code-0.0.42 → kiwi_code-0.0.43}/uv.lock

@@ -397,7 +397,7 @@ wheels = [
 
 [[package]]
 name = "kiwi-code"
-version = "0.0.42"
+version = "0.0.43"
 source = { editable = "." }
 dependencies = [
     { name = "autobots-client" },