Spaces:
Running
Running
| """Browser smoke checks. Run with `python test_browser.py` and installed Chromium + playwright. | |
| Loads the current source files directly into Chromium; no HTTP server is needed. | |
| Uses system Chromium or an installed Playwright Chromium. | |
| """ | |
| from playwright.sync_api import sync_playwright | |
| import os,json,re,shutil | |
| from pathlib import Path | |
| CHROME=os.getenv('CHROMIUM_BINARY') or shutil.which('chromium') or shutil.which('google-chrome') | |
| checks=[] | |
| def assertion(cond): | |
| assert cond,'assertion failed' | |
| def check(name, fn): | |
| try:fn();checks.append((name,True)) | |
| except Exception as e:checks.append((name,False,str(e)));print('FAIL',name, str(e)[:220]);raise | |
| with sync_playwright() as p: | |
| browser=p.chromium.launch(**({'executable_path':CHROME} if CHROME else {}),headless=True,args=['--no-sandbox','--disable-dev-shm-usage']) | |
| context=browser.new_context(viewport={'width':1480,'height':1100},accept_downloads=True) | |
| page=context.new_page();errors=[] | |
| page.on('pageerror',lambda err:errors.append(str(err))) | |
| base=Path(__file__).resolve().parent | |
| html=(base/'index.html').read_text() | |
| html=re.sub(r'<link rel="stylesheet" href="styles.css"/>', '<style>'+ (base/'styles.css').read_text() + '</style>',html) | |
| html=re.sub(r'<script src="[^"]+"></script>', '', html) | |
| page.set_content(html,wait_until='load') | |
| for script in ('engine.js','demo-data.js','app.js'):page.add_script_tag(content=(base/script).read_text()) | |
| check('Title and layout',lambda: (assertion(page.title().startswith('ImpactLab')),assertion(page.locator('#indicator option').count()==8))) | |
| check('Synthetic source visible',lambda:assertion('SYNTHETIC DEMO' in page.locator('#sourceBanner').inner_text())) | |
| check('Overview metrics rendered',lambda:assertion('DiD contrast' in page.locator('#view').inner_text())) | |
| check('Default DiD is calculated',lambda:assertion(page.evaluate('window.__impactState.did.ok') is True)) | |
| check('Synthetic control has convex weights',lambda:assertion(page.evaluate('window.__impactState.synth.ok') is True)) | |
| check('Demo 3000 rows',lambda:assertion(page.evaluate('window.__impactState.rows.length')==3000)) | |
| page.locator('[data-tab="trends"]').click() | |
| check('Trends tab renders chart',lambda:assertion(page.locator('#view svg').count()>=1)) | |
| page.locator('[data-tab="did"]').click() | |
| check('DiD tab renders interval',lambda:assertion('Year-resampling range' in page.locator('#view').inner_text())) | |
| page.locator('[data-tab="synthetic"]').click() | |
| check('Synthetic tab visualizes donors',lambda:assertion('Donor weights' in page.locator('#view').inner_text())) | |
| page.locator('[data-tab="placebos"]').click() | |
| check('Placebo analyses show diagnostics',lambda:assertion('unit placebos' in page.locator('#view').inner_text())) | |
| page.locator('[data-tab="quality"]').click() | |
| check('Coverage audit reports countries',lambda:assertion('Panel completeness' in page.locator('#view').inner_text())) | |
| page.locator('[data-tab="protocol"]').click() | |
| page.locator('#policyName').fill('Hypothetical education reform (not verified)') | |
| page.locator('#policyUrl').fill('https://example.org/report') | |
| page.locator('#researchNotes').fill('This is an exploratory research note. No causal claim.') | |
| page.locator('[data-tab="export"]').click() | |
| check('Research report contains notes and provenance',lambda:assertion('This is an exploratory research note.' in page.locator('#reportPreview').inner_text() and 'SYNTHETIC DEMO' in page.locator('#reportPreview').inner_text())) | |
| with page.expect_download() as dl:page.locator('#exportCsv').click() | |
| check('CSV downloads',lambda:assertion(dl.value.suggested_filename.endswith('.csv'))) | |
| with page.expect_download() as dl2:page.locator('#exportMd').click() | |
| check('Markdown downloads',lambda:assertion(dl2.value.suggested_filename.endswith('.md'))) | |
| with page.expect_download() as dl3:page.locator('#exportJson').click() | |
| sess=json.loads(dl3.value.path().read_text()) | |
| check('JSON session saves exact provenance',lambda:assertion(sess['source']=='demo' and sess['notes'].startswith('This is an exploratory') and len(sess['rows'])==3000)) | |
| page.locator('#startYear').fill('2018');page.locator('#runBtn').click() | |
| check('Invalid short design yields visible warning',lambda:assertion(not page.locator('#errorPanel').is_hidden())) | |
| page.locator('#startYear').fill('2003');page.locator('#runBtn').click() | |
| check('Warning clears after valid design',lambda:assertion(page.locator('#errorPanel').is_hidden())) | |
| # Fake network errors must preserve earlier results, not mislabel as live. | |
| page.route('https://api.worldbank.org/**',lambda route:route.fulfill(status=503,body='unavailable')) | |
| page.locator('#liveBtn').click();page.wait_for_timeout(500) | |
| check('Live fetch failure preserves synthetic data',lambda:assertion(page.evaluate('window.__impactState.source')=='demo' and not page.locator('#errorPanel').is_hidden())) | |
| page.unroute('https://api.worldbank.org/**') | |
| def fake_response(route): | |
| u=route.request.url | |
| data=[{'countryiso3code':code,'date':str(y),'value':70+(i%4)+.1*(y-2000)} for i,code in enumerate(['CHL','ARG','PER','URY']) for y in range(2000,2025)] | |
| route.fulfill(status=200,headers={'Access-Control-Allow-Origin':'*','Content-Type':'application/json'},body=json.dumps([{'page':1,'pages':1,'total':len(data)},data])) | |
| page.route('https://api.worldbank.org/**',fake_response) | |
| page.locator('#liveBtn').click();page.wait_for_timeout(750) | |
| check('World Bank response adapter switches source to LIVE',lambda:assertion(page.evaluate('window.__impactState.source')=='worldbank')) | |
| check('Loaded record count matches mock response',lambda:assertion(page.evaluate('window.__impactState.rows.length')==100)) | |
| check('UI labels live data',lambda:assertion('WORLD BANK LIVE' in page.locator('#sourceBanner').inner_text())) | |
| page.locator('#demoBtn').click() | |
| check('Demo restores previous defaults',lambda:assertion(page.evaluate('window.__impactState.source')=='demo' and page.evaluate('window.__impactState.rows.length')==3000)) | |
| # Import JSON must be flagged imported; do not automatically trust source. | |
| f=page.locator('[data-tab="export"]');f.click();page.locator('#importJson').set_input_files({'name':'session.json','mimeType':'application/json','buffer':json.dumps(sess).encode()});page.wait_for_timeout(150) | |
| check('Session restores as UNVERIFIED IMPORTED',lambda:assertion(page.evaluate('window.__impactState.source')=='imported')) | |
| # malicious URL should never become link | |
| page.locator('#policyUrl').fill('javascript:alert(1)');page.locator('[data-tab="protocol"]').click() | |
| check('Unsafe URL excluded',lambda:assertion('javascript:alert' not in page.locator('#view').inner_html())) | |
| page.locator('#demoBtn').click() | |
| # Desktop and mobile screenshots | |
| page.screenshot(path=str(base/'ImpactLab_LATAM_DESKTOP_PREVIEW.png'),full_page=True) | |
| mobile=browser.new_context(viewport={'width':390,'height':844},device_scale_factor=1,accept_downloads=True) | |
| mob=mobile.new_page();mob.set_content(html,wait_until='load');[mob.add_script_tag(content=(base/script).read_text()) for script in ('engine.js','demo-data.js','app.js')] | |
| check('Mobile interface renders',lambda:assertion(mob.locator('#view .metric').count()>=4)) | |
| check('Mobile layout has no horizontal overflow',lambda:assertion(mob.evaluate('document.documentElement.scrollWidth <= innerWidth+2'))) | |
| mob.screenshot(path=str(base/'ImpactLab_LATAM_MOBILE_PREVIEW.png'),full_page=True) | |
| check('Desktop runtime errors zero',lambda:assertion(len(errors)==0)) | |
| browser.close() | |
| for item in checks:print('PASS' if item[1] else 'FAIL',item[0]) | |
| print('TOTAL',len(checks),'PASS',sum(1 for x in checks if x[1]),'FAIL',sum(1 for x in checks if not x[1])) | |