odysseus/tests/test_document_library_mobile_footer.py

"""Mobile Documents-library footer remains focused and explicit."""

import json
import subprocess
from pathlib import Path


ROOT = Path(__file__).resolve().parents[1]
SOURCE = (ROOT / "static/js/documentLibrary.js").read_text(encoding="utf-8")
STYLE = (ROOT / "static/style.css").read_text(encoding="utf-8")


def test_mobile_footer_exposes_open_and_more_only():
    assert "doclib-expanded-open-btn" in SOURCE
    assert "doclib-expanded-mobile-more" in SOURCE
    assert "label: 'Open in new chat'" in SOURCE
    assert "'Open in original' : 'Open document'" in SOURCE
    assert "label: 'Export file'" in SOURCE
    assert "'Restore document' : 'Archive document'" in SOURCE
    assert "label: 'Delete document'" in SOURCE

    mobile_css = STYLE.split("The Documents preview footer only exposes Open and More", 1)[1]
    mobile_css = mobile_css.split("/* Chat top bar", 1)[0]
    for hidden_action in (
        ".doclib-expanded-delete-btn",
        ".doclib-expanded-archive-btn",
        ".doclib-expanded-clone-btn",
        ".doclib-expanded-export-btn",
    ):
        assert hidden_action in mobile_css
    assert ".doclib-expanded-mobile-more" in mobile_css
    assert "display: inline-flex" in mobile_css
    assert "box-sizing: border-box" in mobile_css
    assert "max-width: 100%" in mobile_css


def test_open_in_new_chat_forces_a_distinct_target_session():
    body = SOURCE.split("async function libraryImportDocument", 1)[1].split(
        "// ---- Library bulk operations", 1
    )[0]
    assert "{ newSession = false }" in body
    assert "let sessionId = newSession ? null" in body
    assert "sessionModule.createDirectChat" in body
    assert "await sessionModule.materializePendingSession()" in body
    assert "newSession ? 'Document opened in new chat' : 'Document opened here'" in body


def test_mobile_open_in_new_chat_copies_to_materialized_session():
    script = r"""
      import { chromium } from 'playwright';
      const browser = await chromium.launch({ headless: true });
      const page = await browser.newPage({ viewport: { width: 390, height: 844 }, hasTouch: true });
      await page.goto('http://127.0.0.1:7011/static/js/documentStats.js');
      await page.setContent('<link rel="stylesheet" href="/static/style.css?v=20260831richtexttools91"><div id="toast"></div><div id="chat-container"></div><div id="sidebar"></div>');
      const state = await page.evaluate(async () => {
        let currentSession = 'current-chat';
        let createDirectCalls = 0;
        let postedBody = null;
        const originalFetch = window.fetch;
        window.fetch = async (input, init = {}) => {
          const url = String(input);
          const response = value => new Response(JSON.stringify(value), {
            status: 200, headers: { 'Content-Type': 'application/json' },
          });
          if (url.includes('/api/documents/library')) return response({
            documents: [{ id: 'source-doc', title: 'Source', language: 'markdown', preview: 'Preview', session_id: 'original-chat' }],
            total: 1, languages: { markdown: 1 }, session_count: 1,
          });
          if (url.endsWith('/api/document/source-doc')) return response({
            id: 'source-doc', title: 'Source', language: 'markdown', current_content: '# Source', session_id: 'original-chat',
          });
          if (url.endsWith('/api/document') && init.method === 'POST') {
            postedBody = JSON.parse(init.body);
            return response({ id: 'copied-doc', language: 'markdown', current_content: '# Source', session_id: postedBody.session_id });
          }
          if (url.includes('/api/sessions') || url.includes('/api/history')) return response([]);
          return originalFetch(input, init);
        };

        const sessions = (await import('/static/js/sessions.js')).default;
        sessions.getCurrentSessionId = () => currentSession;
        sessions.getCurrentModel = () => 'model';
        sessions.getSessions = () => [{ id: 'current-chat', endpoint_url: '/chat', model: 'model', endpoint_id: 'endpoint' }];
        sessions.createDirectChat = () => { createDirectCalls += 1; currentSession = null; };
        sessions.materializePendingSession = async () => { currentSession = 'new-chat'; return true; };

        const library = await import('/static/js/documentLibrary.js?v=20260831richtexttools91&new-chat-test=1');
        library.initLibrary({
          apiBase: '', esc: String, getDocs: () => new Map(), isOpen: () => false,
          createDocument() {}, newDocument() {}, async loadDocument() {}, prepareDocumentOpen() {},
          switchToDoc() {}, openPanel() {}, addDocToTabs() {}, syncDocIndicator() {},
        });
        library.openLibrary();
        await new Promise(resolve => setTimeout(resolve, 120));
        document.querySelector('#doclib-grid .doclib-card').click();
        await new Promise(resolve => setTimeout(resolve, 180));
        document.querySelector('.doclib-expanded-open-btn').click();
        await new Promise(resolve => setTimeout(resolve, 30));
        Array.from(document.querySelectorAll('._lib-dd .dropdown-item-compact'))
          .find(item => item.textContent.includes('Open in new chat')).click();
        for (let i = 0; i < 20 && !postedBody; i += 1) await new Promise(resolve => setTimeout(resolve, 20));
        return { createDirectCalls, currentSession, postedBody };
      });
      console.log(JSON.stringify(state));
      await browser.close();
    """
    result = subprocess.run(
        ["node", "--input-type=module", "-e", script],
        cwd=ROOT,
        check=False,
        capture_output=True,
        text=True,
    )
    assert result.returncode == 0, result.stderr
    data = json.loads(result.stdout)
    assert data["createDirectCalls"] == 1
    assert data["currentSession"] == "new-chat"
    assert data["postedBody"]["session_id"] == "new-chat"