summaryrefslogtreecommitdiff
path: root/lib/spack/spack/test/cmd/extensions.py
blob: e3f82b9439c1389770f3f966d9cf468d1ca35c8f (plain) (blame)
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# Copyright 2013-2023 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)


import pytest

from spack.main import SpackCommand, SpackCommandError
from spack.spec import Spec

extensions = SpackCommand("extensions")


@pytest.fixture
def python_database(mock_packages, mutable_database):
    specs = [Spec(s).concretized() for s in ["python", "py-extension1", "py-extension2"]]

    for spec in specs:
        spec.package.do_install(fake=True, explicit=True)

    yield


@pytest.mark.not_on_windows("All Fetchers Failed")
@pytest.mark.db
def test_extensions(mock_packages, python_database, config, capsys):
    ext2 = Spec("py-extension2").concretized()

    def check_output(ni):
        with capsys.disabled():
            output = extensions("python")
            packages = extensions("-s", "packages", "python")
            installed = extensions("-s", "installed", "python")
        assert "==> python@2.7.11" in output
        assert "==> 2 extensions" in output
        assert "py-extension1" in output
        assert "py-extension2" in output

        assert "==> 2 extensions" in packages
        assert "py-extension1" in packages
        assert "py-extension2" in packages
        assert "installed" not in packages

        assert ("%s installed" % (ni if ni else "None")) in output
        assert ("%s installed" % (ni if ni else "None")) in installed

    check_output(2)
    ext2.package.do_uninstall(force=True)
    check_output(1)


def test_extensions_no_arguments(mock_packages):
    out = extensions()
    assert "python" in out


def test_extensions_raises_if_not_extendable(mock_packages):
    with pytest.raises(SpackCommandError):
        extensions("flake8")


def test_extensions_raises_if_multiple_specs(mock_packages):
    with pytest.raises(SpackCommandError):
        extensions("python", "flake8")