iframe-proxy | Sunbelt Computer Software

21 lines (17 loc) · 617 Bytes

def test_eval_mode(wdriver):

assert wdriver.execute_script("return window.rp.pyEval('1+1')") == 2

def test_exec_mode(wdriver):

assert wdriver.execute_script("return window.rp.pyExec('1+1')") is None

def test_exec_single_mode(wdriver):

assert wdriver.execute_script("return window.rp.pyExecSingle('1+1')") == 2

stdout = wdriver.execute_script(

"""

let output = "";

save_output = function(text) {{

output += text

}};

window.rp.pyExecSingle('1+1\\n2+2',{stdout: save_output});

return output;

"""

)

assert stdout == "2\n4\n"

Sunbelt Computer Software