File size: 17,455 Bytes
0388e8d 00861b2 0388e8d 4754d8a 00861b2 4754d8a 2c1d035 4754d8a 2c1d035 00861b2 4754d8a 00861b2 4754d8a 00861b2 0388e8d 2c1d035 0388e8d 5084a8e 5492cc8 5084a8e 2787e18 5c1bf9a 2787e18 2c1d035 2787e18 84aa36a 2787e18 84aa36a 5708dbe 84aa36a 5708dbe 84aa36a 2787e18 5084a8e 2a74af3 631e579 0388e8d 2c1d035 0388e8d 2c1d035 0388e8d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 | """Search-host navigation: source checks, session handoff and focus ownership."""
import unittest
import subprocess
from pathlib import Path
from tests import test_paging_recovery as paging
class ReaderNavigationHostTests(unittest.TestCase):
@classmethod
def setUpClass(cls):
paging.PagingRecoveryTests.setUpClass()
@classmethod
def tearDownClass(cls):
paging.PagingRecoveryTests.tearDownClass()
def setUp(self):
self.fixture = paging.PagingRecoveryTests()
self.fixture.setUp()
self.page = self.fixture.page
self.page.route('**/static/reader.html?**', lambda route: route.fulfill(
content_type='text/html', body='<body>reader</body>'))
def tearDown(self):
self.assertEqual(self.fixture.errors, [])
self.fixture.tearDown()
def test_bundled_shell_starts_without_separate_dependency_scripts(self):
root = Path(__file__).resolve().parents[1]
shell = subprocess.check_output(['node', '--input-type=module', '-e', '''
import {readFileSync} from 'node:fs';
import {stripBundledScripts} from './scripts/build/compose_app.mjs';
process.stdout.write(stripBundledScripts(readFileSync('static/index.html','utf8')));
'''], cwd=root)
bundle = subprocess.check_output(['node', 'scripts/build/compose_app.mjs'], cwd=root)
with self.fixture.browser.new_context(service_workers='block') as context:
page = context.new_page()
page.on('pageerror', lambda error: self.fixture.errors.append(str(error)))
page.add_init_script(paging.FETCH_SCRIPT)
page.route('**/api/**', lambda route: route.fulfill(content_type='application/json', body='[]'))
page.route('https://vomebook-hitokoto.hf.space/**', lambda route: route.fulfill(body='{}'))
page.route('**/static/app.js', lambda route: route.fulfill(content_type='text/javascript', body=bundle))
page.route(self.fixture.origin + '/?q=paging', lambda route: route.fulfill(content_type='text/html', body=shell))
page.goto(self.fixture.origin + '/?q=paging', wait_until='domcontentloaded')
page.wait_for_function('STATE.results.length === 100 && !STATE.isLoading')
self.assertEqual(page.locator('script[src]').count(), 1)
self.assertTrue(page.evaluate('typeof VoiceOfMLSearchSession.createSearchSession === "function"'))
self.assertTrue(page.evaluate('typeof VoiceOfMLDownloadController.createDownloadController === "function"'))
self.assertTrue(page.evaluate('navigateToReader("/static/reader.html?ext=txt")'))
self.assertTrue(page.evaluate('closeReaderOverlay()'))
def test_replace_preserves_return_focus_session_and_background_attributes(self):
result = self.page.evaluate('''() => {
const trigger=document.createElement('button');document.body.append(trigger);trigger.focus();
const existing=document.createElement('div');existing.setAttribute('inert','');
existing.setAttribute('aria-hidden','false');document.body.append(existing);
const readerPath=new URL(document.querySelector('script[src*="search/reader-navigation.js"]').src).pathname.replace('search/reader-navigation.js','reader.html');
const target=readerPath+'?ext=txt&url=https://huggingface.co/datasets/VoiceOfML/Test/resolve/main/a.txt';
const returnUrl=location.href;
navigateToReader(target);const first=readerOverlay;
const token=new URL(first.src).searchParams.get('nav');
const hidden=trigger.hasAttribute('inert') && trigger.getAttribute('aria-hidden')==='true';
handleReaderMessage({origin:location.origin,source:first.contentWindow,data:{type:'voice-reader-open',url:target+'&title=next'}});
const next=readerOverlay, saved=JSON.parse(sessionStorage.getItem('reader-navigation-current'));
const state=history.state;
const nextUrl=new URL(next.src);
const inherited=nextUrl.searchParams.get('nav')===token && nextUrl.searchParams.get('return')===returnUrl;
const session=saved.readerUrl===next.src && state.readerUrl===next.src && saved.returnScroll.viewKey;
const share=!new URL(location.href).searchParams.has('nav') && !new URL(location.href).searchParams.has('return');
const removed=!first.isConnected;
history.replaceState(null,'',returnUrl);closeReaderOverlay();
const restored=document.activeElement===trigger && !trigger.hasAttribute('inert') && !trigger.hasAttribute('aria-hidden') &&
existing.hasAttribute('inert') && existing.getAttribute('aria-hidden')==='false';
const closed=!next.isConnected && closeReaderOverlay()===false;
trigger.remove();existing.remove();return {hidden,inherited,session:!!session,share,removed,restored,closed};
}''')
self.assertTrue(all(result.values()), result)
def test_click_replenishes_evicted_reader_metadata(self):
result = self.page.evaluate('''() => {
const button=DOM.resultsList.querySelector('[data-action="read"]');
const id=new URL(button.dataset.readerUrl,location.origin).searchParams.get('id');
getReaderLink(STATE.results[Number(button.closest('.result-item').dataset.index)]);
const expected=JSON.parse(sessionStorage.getItem('reader-source:'+id));
for(let i=0;i<200;i++) cacheReaderMetadata('other-'+i,{url:'other'},true);
const evicted=!sessionStorage.getItem('reader-source:'+id);
let captured;
navigateToReader=()=>{captured=JSON.parse(sessionStorage.getItem('reader-source:'+id));};
button.click();
return {evicted,expected,captured};
}''')
self.assertTrue(result['evicted'])
self.assertIsNotNone(result['expected'])
self.assertEqual(result['captured'], result['expected'])
def test_close_disposes_child_synchronously_before_detachment(self):
self.page.evaluate('navigateToReader("/static/reader.html?ext=txt")')
self.page.frame_locator('.reader-overlay').locator('body').wait_for()
result = self.page.evaluate('''() => {
const child=readerOverlay.contentWindow, frame=readerOverlay;
let disposed=false, attached=false;
child.addEventListener('voice-reader-dispose', () => {
disposed=true; attached=frame.isConnected;
});
closeReaderOverlay();
return {disposed, attached, removed:!frame.isConnected, blank:frame.src==='about:blank'};
}''')
self.assertTrue(all(result.values()), result)
def test_pending_shell_has_immediate_feedback_and_rejects_unowned_reveal(self):
held = []
self.context = self.fixture.context
self.context.route('**/static/pending-reader.html?**', lambda route: held.append(route))
self.page.evaluate('''() => {
window.__pendingNavigation = VoiceOfMLReaderNavigation.createNavigation('/static/pending-reader.html');
__pendingNavigation.mount(new URL('/static/pending-reader.html?ext=txt', location.origin));
}''')
self.page.locator('.reader-overlay-opening').wait_for()
self.assertTrue(self.page.locator('.reader-overlay-pending').count())
self.page.evaluate('window.postMessage({type:"voice-reader-shell"},location.origin)')
self.page.wait_for_timeout(30)
self.assertTrue(self.page.locator('.reader-overlay-opening').is_visible())
self.page.evaluate('__pendingNavigation.unmount()')
self.assertEqual(self.page.locator('.reader-overlay-opening').count(), 0)
self.assertEqual(self.page.locator('.reader-overlay').count(), 0)
for route in held:
route.fulfill(content_type='text/html', body='<body>cancelled</body>')
def test_shell_is_warmed_without_pointer_intent_or_reader_execution(self):
self.page.wait_for_function('warmedReaderAssets.has("/static/reader.html")')
self.assertTrue(self.page.locator('link[rel="modulepreload"][href="/static/reader.js"]').count())
self.assertEqual(self.page.locator('.reader-overlay').count(), 0)
self.assertFalse(self.page.evaluate('!!window.VoiceOfMLReaderPdfDiagnostics'))
def test_random_reader_body_timeout_after_close_remains_retryable(self):
result = self.page.evaluate('''async () => {
const originalFetch=fetch, originalTimer=setTimeout, previousUrl=location.href;
const localFallback=typeof ensureLocalDataLoaded==='function'?ensureLocalDataLoaded:null;
let attempts=0, aborted=false;
STATE.dataLoaded=false;
if (localFallback) ensureLocalDataLoaded=async()=>false;
window.setTimeout=(fn,ms,...args)=>originalTimer(fn,ms===10000?50:ms,...args);
window.fetch=(input,options)=>{
if (!String(input).includes('/api/random-reader')) return originalFetch(input,options);
if (++attempts===1) return Promise.resolve(new Response(new ReadableStream({start(controller){
options.signal.addEventListener('abort',()=>{aborted=true;controller.error(new DOMException('timed out','AbortError'));},{once:true});
}})));
return Promise.resolve(new Response(JSON.stringify({Repo:'VoiceOfML/Test',File:'next',Extension:'txt',
Link:'https://huggingface.co/datasets/VoiceOfML/Test/resolve/main/next.txt',Folder:[]})));
};
try {
const path=new URL(document.querySelector('script[src*="search/reader-navigation.js"]').src).pathname.replace('search/reader-navigation.js','reader.html');
navigateToReader(path+'?ext=txt');closeReaderOverlay();history.replaceState(null,'',previousUrl);
await randomTxt();await new Promise(resolve=>originalTimer(resolve,100));
const recovered=aborted&&!!readerOverlay;
closeReaderOverlay();history.replaceState(null,'',previousUrl);
await randomTxt();await new Promise(resolve=>originalTimer(resolve,100));
return {recovered,attempts,opened:!!readerOverlay};
} finally {
window.fetch=originalFetch;window.setTimeout=originalTimer;
if(localFallback)ensureLocalDataLoaded=localFallback;
}
}''')
self.assertEqual(result, {'recovered':True,'attempts':3,'opened':True})
def test_random_reader_retries_transient_http_but_not_permanent_or_stale_failure(self):
result = self.page.evaluate('''async () => {
const originalFetch=fetch, savedUrl=location.href;
const success={Repo:'VoiceOfML/Test',File:'next',Extension:'txt',
Link:'https://huggingface.co/datasets/VoiceOfML/Test/resolve/main/next.txt',Folder:[]};
const rows=[];
try {
for(const code of [503,429,404]) {
let requests=0;
window.fetch=(input,options)=>String(input).includes('/api/random-reader')
? Promise.resolve(++requests===1
? new Response('{}',{status:code}) : new Response(JSON.stringify(success)))
: originalFetch(input,options);
const id=++randomReaderRequestId;
try { await fetchRandomReaderRecord('/api/random-reader',id,savedUrl); } catch (_) {}
rows.push(requests);
}
let requests=0;
window.fetch=(input,options)=>{
if (!String(input).includes('/api/random-reader')) return originalFetch(input,options);
requests++;randomReaderRequestId++;return Promise.reject(new TypeError('network'));
};
await fetchRandomReaderRecord('/api/random-reader',++randomReaderRequestId,savedUrl);
rows.push(requests);
return rows;
} finally {window.fetch=originalFetch;}
}''')
self.assertEqual(result, [2,2,1,1])
def test_reader_folder_url_uses_current_filters(self):
result = self.page.evaluate('''() => {
STATE.filterMinSize=2048; STATE.filterMaxSize=8192;
STATE.filterExtensions=['pdf','txt']; STATE.searchFolders=false; STATE.exact=false;
DOM.sortSelect.value='size'; STATE.recordHistory=false;
const url=new URL(getReaderFolderUrl({Repo:'VoiceOfML/Test',Folder:['a.b','书']}));
return {path:url.pathname,params:Object.fromEntries(url.searchParams)};
}''')
self.assertEqual(result['path'], '/Test')
params = result['params']
self.assertEqual(params['sort'], 'size')
self.assertEqual(params['ext'], 'pdf,txt')
self.assertEqual(params['folder_self'], 'a.b/书')
self.assertEqual(params['min_size'], '2KB')
self.assertEqual(params['max_size'], '8KB')
self.assertEqual(params['search_folders'], 'false')
self.assertEqual(params['exact'], '0')
self.assertEqual(params['history'], '0')
self.assertNotIn('q', params)
def test_reader_return_captures_cleared_query_before_search_finishes(self):
result = self.page.evaluate('''() => {
STATE.query = '旧查询'; DOM.searchInput.value = STATE.query; syncStateToURL(true);
const originalSearch = doSearch;
doSearch = () => new Promise(() => {});
try {
submitSearchQuery('', {clearResults: true, updateInput: true});
const current = location.href;
navigateToReader('/static/reader.html?ext=txt');
const saved = JSON.parse(sessionStorage.getItem('reader-navigation-current'));
history.back();
return {current, route: saved.returnScroll.route, reader: readerOverlay !== null};
} finally { doSearch = originalSearch; }
}''')
self.assertNotIn('q=%E6%97%A7%E6%9F%A5%E8%AF%A2', result['current'])
self.assertNotIn('q=%E6%97%A7%E6%9F%A5%E8%AF%A2', result['route'])
self.assertTrue(result['reader'])
def test_reader_folder_return_merges_all_current_search_settings(self):
result = self.page.evaluate('''() => {
STATE.query='手机';
STATE.filterExtensions=['pdf','txt'];
STATE.filterMinSize=2048; STATE.filterMaxSize=8192;
STATE.searchFolders=false; STATE.exact=false; STATE.recordHistory=false;
STATE.useMirrorLinks=false; DOM.sortSelect.value='size';
let target=new URL('/Test?folder_self=docs', location.origin);
target=readerNavigation.mergeSearchParams(target, buildSearchURLParams(), {
keys: READER_RETURN_SEARCH_KEYS
});
return Object.fromEntries(target.searchParams);
}''')
self.assertEqual(result['q'], '手机')
self.assertEqual(result['ext'], 'pdf,txt')
self.assertEqual(result['min_size'], '2KB')
self.assertEqual(result['max_size'], '8KB')
self.assertEqual(result['sort'], 'size')
self.assertEqual(result['search_folders'], 'false')
self.assertEqual(result['exact'], '0')
self.assertEqual(result['history'], '0')
self.assertEqual(result['mirror'], '0')
self.assertEqual(result['folder_self'], 'docs')
def test_untrusted_and_stale_messages_cannot_change_active_reader(self):
result = self.page.evaluate('''() => {
const readerPath=new URL(document.querySelector('script[src*="search/reader-navigation.js"]').src).pathname.replace('search/reader-navigation.js','reader.html');
const target=readerPath+'?ext=txt';navigateToReader(target);
const first=readerOverlay, oldSource=first.contentWindow, theme=STATE.isDark;
const message={type:'voice-reader-open',url:target+'&title=bad'};
handleReaderMessage({origin:'https://invalid.example',source:oldSource,data:message});
handleReaderMessage({origin:location.origin,source:window,data:message});
handleReaderMessage({origin:location.origin,source:oldSource,data:{...message,url:'/wrong-path'}});
const untouched=readerOverlay===first;
handleReaderMessage({origin:location.origin,source:oldSource,data:{...message,url:target+'&title=valid'}});
const second=readerOverlay;
handleReaderMessage({origin:location.origin,source:oldSource,data:{type:'voice-reader-theme',theme:theme?'light':'dark'}});
return {untouched,replaced:second!==first,staleIgnored:STATE.isDark===theme};
}''')
self.assertTrue(all(result.values()), result)
def test_unavailable_session_storage_does_not_block_replacement(self):
result = self.page.evaluate('''() => {
const readerPath=new URL(document.querySelector('script[src*="search/reader-navigation.js"]').src).pathname.replace('search/reader-navigation.js','reader.html');
const target=readerPath+'?ext=txt';navigateToReader(target);const first=readerOverlay;
const original=Storage.prototype.setItem;
try {
Storage.prototype.setItem=()=>{throw new DOMException('Full','QuotaExceededError');};
handleReaderMessage({origin:location.origin,source:first.contentWindow,data:{type:'voice-reader-open',url:target+'&title=next'}});
return readerOverlay!==first && !first.isConnected && history.state.readerUrl===readerOverlay.src;
} finally {Storage.prototype.setItem=original;}
}''')
self.assertTrue(result)
|