Source code
Revision control
Copy as Markdown
Other Tools
import pytest
ADD_PROMPT_CSS = ".ql-editor.textarea.new-input-ui"
SEND_CSS = "[data-mat-icon-name=send]"
SIGNED_OUT_TEXT = "been signed out"
RESPONSE_CSS = "message-content"
EDIT_OLD_PROMPT_CSS = "[data-mat-icon-name=edit]"
OLD_PROMPT_CSS = "textarea[id^=mat-input-]"
CANCEL_CSS = "button.cancel-button"
async def check_paste_works(client):
client.set_clipboard("")
await client.navigate(URL)
client.await_css(ADD_PROMPT_CSS, is_displayed=True).send_keys("hello")
client.await_css(SEND_CSS, is_displayed=True).click()
signed_out, _ = client.await_first_element_of(
[client.text(SIGNED_OUT_TEXT), client.css(RESPONSE_CSS)], is_displayed=True
)
if signed_out:
pytest.skip("Blocked from accessing site. Please try testing manually.")
return
client.soft_click(client.await_css(EDIT_OLD_PROMPT_CSS))
client.await_css(OLD_PROMPT_CSS, is_displayed=True).click()
client.execute_script("document.execCommand('selectAll')")
client.execute_script("document.execCommand('copy')")
client.await_css(CANCEL_CSS, is_displayed=True).click()
prompt = client.await_css(ADD_PROMPT_CSS, is_displayed=True)
await client.apz_click(element=prompt, offset=[40, 20])
client.do_paste()
return client.execute_script(
"""
return arguments[0].innerText.trim() === "hello"
""",
prompt,
)
async def check_addons_work(client):
# This is a regression test for the issue described in
client.install_addon(
{
"manifest.json": """{
"manifest_version": 2,
"name": "Pasting Test",
"version": "1.0",
"content_scripts": [
{
"matches": ["*://gemini.google.com/*"],
"js": ["test.js"]
}
]
}
""",
"test.js": """
document.addEventListener("paste", e => {
const clipboardInit = cloneInto({}, window);
clipboardInit.clipboardData = new DataTransfer();
const newEvent = new wrappedJSObject.ClipboardEvent(e.type, clipboardInit);
try {
newEvent.clipboardData.setData("text/plain", e.clipboardData.getData("text/plain"));
console.error("Test result: success");
} catch (_) {
console.error("Test result: failed");
}
});
""",
}
)
await client.navigate(URL)
client.set_clipboard("test")
client.await_css(ADD_PROMPT_CSS, is_displayed=True).click()
promise = await client.promise_console_message_listener("Test result:")
client.do_paste()
msg = (await promise)["args"][0]["value"]
assert msg == "Test result: success"
@pytest.mark.skip_platforms("android")
@pytest.mark.asyncio
@pytest.mark.with_interventions
async def test_enabled(client):
assert await check_paste_works(client)
@pytest.mark.skip_platforms("android")
@pytest.mark.asyncio
@pytest.mark.with_interventions
async def test_addons_works(client):
await check_addons_work(client)
@pytest.mark.skip_platforms("android")
@pytest.mark.asyncio
@pytest.mark.without_interventions
async def test_disabled(client):
assert not await check_paste_works(client)