Ink Bench: fix a share-link restore race, patient start in the e2e test

lib/bench.js

@@ -171,7 +171,7 @@ function shareURL() {
 async function restoreShare(code) {
   const st = JSON.parse(unb64u(code));
   if (st.v !== 1) throw new Error('Unknown share link version.');
-  open('compare');
+  S.seg = null; await open('compare', st.s);
   await selectSegment(st.s);
   S.tRef = +st.t[0]; S.tPred = +st.t[1]; S.invert = !!st.i; S.target = st.g || 'ink'; S.model = st.m || { name: '' };
   S.ann = (st.an || []).map((a, i) => C.cleanAnnotation({ x: a[0], y: a[1], w: a[2], h: a[3], letter: a[4], confidence: a[5], note: a[6] }, i));
@@ -673,11 +673,13 @@ function renderBoard() {
 // ------------------------------------------------------------------ open, close, API
 
 let opener = null;
-async function open(tab = 'compare') {
+async function open(tab = 'compare', seg = null) {
   await loadIndex(); build();
+  if (seg && !S.seg) S.seg = seg;
   opener = document.activeElement;
   $('bench').hidden = false; document.body.classList.add('bench-open');
-  if (!S.seg) await selectSegment(Object.keys(S.index.segments).includes('w046') ? 'w046' : Object.keys(S.index.segments)[0]);
+  if (S.seg && !S.ref) await selectSegment(S.seg);
+  else if (!S.seg) await selectSegment(Object.keys(S.index.segments).includes('w046') ? 'w046' : Object.keys(S.index.segments)[0]);
   showTab(tab);
   $('bClose').focus();
 }

tests/e2e_bench.py

@@ -38,7 +38,7 @@ with sync_playwright() as pw:
     p.wait_for_selector('#btnBench', timeout=60000)
     p.wait_for_selector('#loader', state='detached', timeout=120000)
     p.wait_for_timeout(500)
-    p.click('#btnBench')
+    p.click('#btnBench', timeout=180000)  # software WebGL in headless runs can keep the main thread busy at start
     p.wait_for_selector('#bench:not([hidden])')
     txt = p.inner_text('#bench')
     check('Nothing is uploaded' in txt, 'UI says nothing is uploaded')
@@ -153,7 +153,7 @@ with sync_playwright() as pw:
     ct = p.inner_text('#bench')
     check('Apache License 2.0' in ct and 'CC BY-NC 4.0' in ct and 'Chase Hendrick' in ct, 'cite panel shows Apache-2.0, CC BY-NC 4.0 and attribution')
     p.screenshot(path=os.path.join(shots, 'bench_cite.png'))
-    p.click('#bClose'); p.wait_for_timeout(200); p.click('#btnInfo'); p.wait_for_timeout(400)
+    p.click('#bClose'); p.wait_for_timeout(200); p.click('#btnInfo', timeout=120000); p.wait_for_timeout(400)
     check('Why this matters for researchers' in p.inner_text('#about'), 'About panel has the researchers section')
     p.locator('#about .bench-why').scroll_into_view_if_needed(); p.screenshot(path=os.path.join(shots, 'bench_about.png'))
     R['api'] = p.evaluate("({ shim: typeof window.unrollBench, viaUnroll: !!(window.unroll && window.unroll.bench), hasUnroll: typeof window.unroll })")