import json
import subprocess
from pathlib import Path


MODULE = Path(__file__).resolve().parents[1] / "frontend" / "shared-image-capture.js"


def test_shared_image_is_consumed_once_and_restored_into_capture():
    script = f"""
const shared = require({json.dumps(str(MODULE))});
const calls=[];
const store={{
  get:async id=>({{filename:'bug.png',contentType:'image/png',blob:{{size:6,type:'image/png'}}}}),
  delete:async id=>calls.push(['delete',id]),
}};
shared.consume({{
  marker:'image', store,
  restore:value=>calls.push(['restore',value.filename]),
  status:message=>calls.push(['status',message]),
}}).then(first=>shared.consume({{marker:'',store,restore:()=>calls.push(['restore-again'])}}).then(second=>{{
  process.stdout.write(JSON.stringify({{first,second,calls}}));
}}));
"""
    completed = subprocess.run(["node", "-e", script], capture_output=True, text=True)

    assert completed.returncode == 0, completed.stderr
    assert json.loads(completed.stdout) == {
        "first": True,
        "second": False,
        "calls": [
            ["restore", "bug.png"],
            ["delete", "shared-image"],
            ["status", "Shared screenshot ready to file with this issue."],
        ],
    }


def test_invalid_shared_image_reports_error_without_touching_open_capture():
    script = f"""
const shared = require({json.dumps(str(MODULE))});
const calls=[];
shared.consume({{
  marker:'unsupported', store:{{get:async()=>null,delete:async()=>calls.push('delete')}},
  restore:()=>calls.push('restore'), status:message=>calls.push(message),
}}).then(result=>process.stdout.write(JSON.stringify({{result,calls}})));
"""
    completed = subprocess.run(["node", "-e", script], capture_output=True, text=True)

    assert completed.returncode == 0, completed.stderr
    assert json.loads(completed.stdout) == {
        "result": False,
        "calls": ["Share one PNG, JPEG, or WebP screenshot."],
    }