import pytest
from tests.support.classic.asserts import assert_error, assert_success


def get_cookies(session):
    return session.transport.send("GET", f"session/{session.session_id}/cookie")


# To minimize Firefox restarts, run tests requiring system access first,
# followed by those that don't; so only one restart is needed.


@pytest.mark.geckodriver(allow_system_access=True)
def test_parent_process_context_with_system_access(session):
    session.url = "about:about"

    response = get_cookies(session)
    assert_success(response)


def test_parent_process_context_without_system_access(parent_process_session):
    response = get_cookies(parent_process_session)
    assert_error(response, "unsupported operation")


def test_extension_context_without_system_access(session, install_new_tab_extension):
    handle, _ = install_new_tab_extension
    session.window_handle = handle
    assert session.url.startswith("moz-extension://")

    response = get_cookies(session)
    assert_error(response, "unsupported operation")