summaryrefslogtreecommitdiff
path: root/lib/spack/spack/test/cmd/build_env.py
blob: 379de521b25457b4ad2f4fa3213c4c48d4a516a0 (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
# Copyright 2013-2019 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)

from six.moves import cPickle
import pytest

from spack.main import SpackCommand, SpackCommandError

info = SpackCommand('build-env')


@pytest.mark.parametrize('pkg', [
    ('zlib',),
    ('zlib', '--')
])
@pytest.mark.usefixtures('config')
def test_it_just_runs(pkg):
    info(*pkg)


@pytest.mark.parametrize('pkg,error_cls', [
    ('zlib libszip', SpackCommandError),
    ('', IndexError)
])
@pytest.mark.usefixtures('config')
def test_it_just_fails(pkg, error_cls):
    with pytest.raises(error_cls):
        info(pkg)


_out_file = 'env.out'


@pytest.mark.usefixtures('config')
def test_dump(tmpdir):
    with tmpdir.as_cwd():
        info('--dump', _out_file, 'zlib')
        with open(_out_file) as f:
            assert(any(line.startswith('PATH=') for line in f.readlines()))


@pytest.mark.usefixtures('config')
def test_pickle(tmpdir):
    with tmpdir.as_cwd():
        info('--pickle', _out_file, 'zlib')
        environment = cPickle.load(open(_out_file, 'rb'))
        assert(type(environment) == dict)
        assert('PATH' in environment)