import os

import mozinfo
import mozinstall
import mozunit
import pytest


@pytest.mark.skipif(
    mozinfo.isWin,
    reason="Bug 1157352 - New firefox.exe needed for mozinstall 1.12 and higher.",
)
def test_get_binary(tmpdir, get_installer):
    """Test to retrieve binary from install path."""
    if mozinfo.isLinux:
        installdir = mozinstall.install(get_installer("tar.xz"), tmpdir.strpath)
        binary = os.path.join(installdir, "firefox")

        assert mozinstall.get_binary(installdir, "firefox") == binary

    elif mozinfo.isWin:
        installdir_exe = mozinstall.install(
            get_installer("exe"), tmpdir.join("exe").strpath
        )
        binary_exe = os.path.join(installdir_exe, "core", "firefox.exe")

        assert mozinstall.get_binary(installdir_exe, "firefox") == binary_exe

        installdir_zip = mozinstall.install(
            get_installer("zip"), tmpdir.join("zip").strpath
        )
        binary_zip = os.path.join(installdir_zip, "firefox.exe")

        assert mozinstall.get_binary(installdir_zip, "firefox") == binary_zip

    elif mozinfo.isMac:
        installdir = mozinstall.install(get_installer("dmg"), tmpdir.strpath)
        binary = os.path.join(installdir, "Contents", "MacOS", "firefox")

        assert mozinstall.get_binary(installdir, "firefox") == binary


def test_get_binary_error(tmpdir):
    """Test that an InvalidBinary error is raised."""
    with pytest.raises(mozinstall.InvalidBinary):
        mozinstall.get_binary(tmpdir.strpath, "firefox")


if __name__ == "__main__":
    mozunit.main()