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
|
# Copyright 2021 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Tests for zmake packers."""
import pathlib
import tempfile
import hypothesis # pylint:disable=import-error
import hypothesis.strategies as st # pylint:disable=import-error
import pytest # pylint:disable=import-error
import zmake.output_packers as packers
# Strategies for use with hypothesis
absolute_path = st.from_regex(regex=r"\A/[\w/]*\Z")
class FakePacker(packers.BasePacker):
"""Fake packer to expose protected methods."""
def __init__(self, max_size):
super().__init__(project=None)
self.max_size = max_size
def check_packed_file_size(self, file, dir_map):
"""Expose the _check_packed_file_size method."""
return self._check_packed_file_size(file, dir_map)
def _get_max_image_bytes(self, dir_map):
return self.max_size
def pack_firmware(self, work_dir, jobclient, dir_map, version_string=""):
del version_string
assert False
@hypothesis.given(st.binary(min_size=101, max_size=200))
@hypothesis.settings(deadline=60000)
def test_file_size_unbounded(data):
"""Test with file size unbounded."""
packer = FakePacker(None)
with tempfile.TemporaryDirectory() as temp_dir_name:
file = pathlib.Path(temp_dir_name) / "zephyr.elf"
with open(file, "wb") as outfile:
outfile.write(data)
assert packer.check_packed_file_size(file=file, dir_map={}) == file
@hypothesis.given(st.binary(min_size=5, max_size=100))
@hypothesis.settings(deadline=60000)
def test_file_size_in_bounds(data):
"""Test with file size limited."""
packer = FakePacker(100)
with tempfile.TemporaryDirectory() as temp_dir_name:
file = pathlib.Path(temp_dir_name) / "ec.bin"
with open(file, "wb") as outfile:
outfile.write(data)
assert packer.check_packed_file_size(file=file, dir_map={}) == file
@hypothesis.given(st.binary(min_size=101, max_size=200))
@hypothesis.settings(deadline=60000)
def test_file_size_out_of_bounds(data):
"""Test with file size limited, and file exceeds limit."""
packer = FakePacker(100)
with tempfile.TemporaryDirectory() as temp_dir_name:
file = pathlib.Path(temp_dir_name) / "ec.bin"
with open(file, "wb") as outfile:
outfile.write(data)
with pytest.raises(RuntimeError):
packer.check_packed_file_size(file=file, dir_map={})
|
