summaryrefslogtreecommitdiffstats
path: root/testing/mozbase/mozinstall/tests/test_binary.py
blob: 6454c78ef51e6fc4e7f6474d52f28ab3e0dd1098 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
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.bz2"), 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()