import pytest
from support.addons import (
    get_ids_for_installed_addons,
    is_addon_private_browsing_allowed,
    is_addon_temporary_installed,
)
from support.helpers import get_pref, set_pref
from tests.support.classic.asserts import assert_error, assert_success
from tests.support.helpers import get_extension_path

from . import ADDON_ID, install_addon, uninstall_addon


def test_install_invalid_addon(session):
    response = install_addon(session, "path", get_extension_path("firefox/invalid.xpi"))
    assert_error(response, "unknown error")


def test_install_nonexistent_addon(session):
    response = install_addon(session, "path", get_extension_path("does-not-exist.xpi"))
    assert_error(response, "unknown error")


@pytest.mark.geckodriver(allow_system_access=True)
@pytest.mark.parametrize("value", [True, False], ids=["required", "not required"])
def test_install_unsigned_addon_with_signature(session, value):
    # Even though "xpinstall.signatures.required" preference is enabled in Firefox by default,
    # it's disabled for wpt tests, so test both values here.
    original_value = get_pref(session, "xpinstall.signatures.required")
    set_pref(session, "xpinstall.signatures.required", value)

    try:
        response = install_addon(
            session,
            "path",
            get_extension_path("firefox/unsigned.xpi"),
            False,
        )

        if value is True:
            assert_error(response, "unknown error")
        else:
            addon_id = assert_success(response)

            installed_addon_ids = get_ids_for_installed_addons(session)

            try:
                assert addon_id in installed_addon_ids
                assert addon_id == ADDON_ID
                assert is_addon_temporary_installed(session, addon_id) is False
            finally:
                # Clean up the addon.
                uninstall_addon(session, addon_id)
    finally:
        set_pref(session, "xpinstall.signatures.required", original_value)


@pytest.mark.geckodriver(allow_system_access=True)
def test_temporary_install_unsigned_addon(session):
    response = install_addon(
        session, "path", get_extension_path("firefox/unsigned.xpi"), True
    )
    addon_id = assert_success(response)

    installed_addon_ids = get_ids_for_installed_addons(session)

    try:
        assert addon_id in installed_addon_ids
        assert addon_id == ADDON_ID
        assert is_addon_temporary_installed(session, addon_id) is True
    finally:
        # Clean up the addon.
        uninstall_addon(session, addon_id)


@pytest.mark.geckodriver(allow_system_access=True)
@pytest.mark.parametrize("temporary", [True, False])
def test_install_signed_addon(session, temporary):
    response = install_addon(
        session, "path", get_extension_path("firefox/signed.xpi"), temporary
    )
    addon_id = assert_success(response)

    installed_addon_ids = get_ids_for_installed_addons(session)

    try:
        assert addon_id in installed_addon_ids
        assert addon_id == ADDON_ID
        assert is_addon_temporary_installed(session, addon_id) is temporary
    finally:
        # Clean up the addon.
        uninstall_addon(session, addon_id)


@pytest.mark.geckodriver(allow_system_access=True)
def test_install_mixed_separator_windows(session):
    os = session.capabilities["platformName"]

    # Only makes sense to test on Windows.
    if os == "windows":
        # Ensure the base path has only \
        addon_path = get_extension_path("firefox").replace("/", "\\")
        addon_path += "/signed.xpi"

        response = install_addon(session, "path", addon_path, False)
        addon_id = assert_success(response)

        installed_addon_ids = get_ids_for_installed_addons(session)

        try:
            assert addon_id in installed_addon_ids
            assert addon_id == ADDON_ID
            assert is_addon_temporary_installed(session, addon_id) is False
        finally:
            # Clean up the addon.
            uninstall_addon(session, addon_id)


@pytest.mark.geckodriver(allow_system_access=True)
@pytest.mark.parametrize("allow_private_browsing", [True, False])
def test_install_addon_with_private_browsing(session, allow_private_browsing):
    response = install_addon(
        session,
        "path",
        get_extension_path("firefox/signed.xpi"),
        False,
        allow_private_browsing,
    )
    addon_id = assert_success(response)

    installed_addon_ids = get_ids_for_installed_addons(session)

    try:
        assert addon_id in installed_addon_ids
        assert addon_id == ADDON_ID
        assert (
            is_addon_private_browsing_allowed(session, addon_id)
            is allow_private_browsing
        )
    finally:
        # Clean up the addon.
        uninstall_addon(session, addon_id)