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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
#
# Copyright (C) 2018 Codethink Limited
# Copyright (C) 2019 Bloomberg Finance LP
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library. If not, see <http://www.gnu.org/licenses/>.
#
# Pylint doesn't play well with fixtures and dependency injection from pytest
# pylint: disable=redefined-outer-name
import os
import pytest
from buildstream import _yaml
from .._utils.site import HAVE_SANDBOX
from .. import create_repo, ALL_REPO_KINDS
from .. import cli # pylint: disable=unused-import
# Project directory
TOP_DIR = os.path.dirname(os.path.realpath(__file__))
DATA_DIR = os.path.join(TOP_DIR, 'project')
def create_test_file(*path, mode=0o644, content='content\n'):
path = os.path.join(*path)
os.makedirs(os.path.dirname(path), exist_ok=True)
with open(path, 'w') as f:
f.write(content)
os.fchmod(f.fileno(), mode)
def create_test_directory(*path, mode=0o644):
create_test_file(*path, '.keep', content='')
path = os.path.join(*path)
os.chmod(path, mode)
@pytest.mark.integration
@pytest.mark.datafiles(DATA_DIR)
@pytest.mark.parametrize("kind", ['local', *ALL_REPO_KINDS])
@pytest.mark.skipif(not HAVE_SANDBOX, reason='Only available with a functioning sandbox')
def test_deterministic_source_umask(cli, tmpdir, datafiles, kind):
project = str(datafiles)
element_name = 'list.bst'
element_path = os.path.join(project, 'elements', element_name)
repodir = os.path.join(str(tmpdir), 'repo')
sourcedir = os.path.join(project, 'source')
create_test_file(sourcedir, 'a.txt', mode=0o700)
create_test_file(sourcedir, 'b.txt', mode=0o755)
create_test_file(sourcedir, 'c.txt', mode=0o600)
create_test_file(sourcedir, 'd.txt', mode=0o400)
create_test_file(sourcedir, 'e.txt', mode=0o644)
create_test_file(sourcedir, 'f.txt', mode=0o4755)
create_test_file(sourcedir, 'g.txt', mode=0o2755)
create_test_file(sourcedir, 'h.txt', mode=0o1755)
create_test_directory(sourcedir, 'dir-a', mode=0o0700)
create_test_directory(sourcedir, 'dir-c', mode=0o0755)
create_test_directory(sourcedir, 'dir-d', mode=0o4755)
create_test_directory(sourcedir, 'dir-e', mode=0o2755)
create_test_directory(sourcedir, 'dir-f', mode=0o1755)
if kind == 'local':
source = {'kind': 'local',
'path': 'source'}
else:
repo = create_repo(kind, repodir)
ref = repo.create(sourcedir)
source = repo.source_config(ref=ref)
element = {
'kind': 'manual',
'depends': [
{
'filename': 'base.bst',
'type': 'build'
}
],
'sources': [
source
],
'config': {
'install-commands': [
'ls -l >"%{install-root}/ls-l"'
]
}
}
_yaml.dump(element, element_path)
def get_value_for_umask(umask):
checkoutdir = os.path.join(str(tmpdir), 'checkout-{}'.format(umask))
old_umask = os.umask(umask)
try:
result = cli.run(project=project, args=['build', element_name])
result.assert_success()
result = cli.run(project=project, args=['artifact', 'checkout', element_name, '--directory', checkoutdir])
result.assert_success()
with open(os.path.join(checkoutdir, 'ls-l'), 'r') as f:
return f.read()
finally:
os.umask(old_umask)
cli.remove_artifact_from_cache(project, element_name)
assert get_value_for_umask(0o022) == get_value_for_umask(0o077)
|