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
|
import os
import pytest
from unittest import mock
from buildstream import _yaml
from buildstream._artifactcache import CACHE_SIZE_FILE
from buildstream._exceptions import ErrorDomain
from tests.testutils import cli, create_element_size
# XXX: Currently lacking:
# * A way to check whether it's faster to read cache size on
# successive invocations.
# * A way to check whether the cache size file has been read.
def create_project(project_dir):
project_file = os.path.join(project_dir, "project.conf")
project_conf = {
"name": "test"
}
_yaml.dump(project_conf, project_file)
element_name = "test.bst"
create_element_size(element_name, project_dir, ".", [], 1024)
def test_cache_size_roundtrip(cli, tmpdir):
# Builds (to put files in the cache), then invokes buildstream again
# to check nothing breaks
# Create project
project_dir = str(tmpdir)
create_project(project_dir)
# Build, to populate the cache
res = cli.run(project=project_dir, args=["build", "test.bst"])
res.assert_success()
# Show, to check that nothing breaks while reading cache size
res = cli.run(project=project_dir, args=["show", "test.bst"])
res.assert_success()
def test_cache_size_write(cli, tmpdir):
# Builds (to put files in the cache), then checks a number is
# written to the cache size file.
project_dir = str(tmpdir)
create_project(project_dir)
# Artifact cache must be in a known place
artifactdir = os.path.join(project_dir, "artifacts")
cli.configure({"artifactdir": artifactdir})
# Build, to populate the cache
res = cli.run(project=project_dir, args=["build", "test.bst"])
res.assert_success()
# Inspect the artifact cache
sizefile = os.path.join(artifactdir, CACHE_SIZE_FILE)
assert os.path.isfile(sizefile)
with open(sizefile, "r") as f:
size_data = f.read()
size = int(size_data)
def test_quota_over_1024T(cli, tmpdir):
KiB = 1024
MiB = (KiB * 1024)
GiB = (MiB * 1024)
TiB = (GiB * 1024)
cli.configure({
'cache': {
'quota': 2048 * TiB
}
})
project = tmpdir.join("main")
os.makedirs(str(project))
_yaml.dump({'name': 'main'}, str(project.join("project.conf")))
volume_space_patch = mock.patch(
"buildstream._artifactcache.ArtifactCache._get_cache_volume_size",
autospec=True,
return_value=(1025 * TiB, 1025 * TiB)
)
with volume_space_patch:
result = cli.run(project, args=["build", "file.bst"])
result.assert_main_error(ErrorDomain.ARTIFACT, 'insufficient-storage-for-quota')
|
