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
|
from unittest.mock import patch
from pytest import raises
import sys
import unittest.mock as mock
import kiwi
from ..test_helper import argv_kiwi_tests
from kiwi.defaults import Defaults
from kiwi.builder.archive import ArchiveBuilder
from kiwi.exceptions import KiwiArchiveSetupError
class TestArchiveBuilder:
def setup(self):
Defaults.set_platform_name('x86_64')
self.xml_state = mock.Mock()
self.xml_state.profiles = None
self.xml_state.get_image_version = mock.Mock(
return_value='1.2.3'
)
self.xml_state.get_build_type_name = mock.Mock(
return_value='tbz'
)
self.xml_state.xml_data.get_name = mock.Mock(
return_value='myimage'
)
self.setup = mock.Mock()
kiwi.builder.archive.SystemSetup = mock.Mock(
return_value=self.setup
)
self.archive = ArchiveBuilder(
self.xml_state, 'target_dir', 'root_dir'
)
def setup_method(self, cls):
self.setup()
def test_create_unknown_archive_type(self):
xml_state = mock.Mock()
xml_state.get_build_type_name = mock.Mock(
return_value='bogus'
)
xml_state.get_image_version = mock.Mock(
return_value='1.2.3'
)
xml_state.xml_data.get_name = mock.Mock(
return_value='myimage'
)
archive = ArchiveBuilder(
xml_state, 'target_dir', 'root_dir'
)
with raises(KiwiArchiveSetupError):
archive.create()
@patch('kiwi.builder.archive.ArchiveTar')
def test_create_tar(self, mock_tar):
Defaults.set_platform_name('x86_64')
self.archive.requested_archive_type = 'tbz'
archive = mock.Mock()
mock_tar.return_value = archive
self.archive.create()
mock_tar.assert_called_once_with(
'target_dir/myimage.x86_64-1.2.3.tar'
)
archive.create_xz_compressed.assert_called_once_with(
'root_dir', exclude=[
'image', '.kconfig', 'run/*', 'tmp/*',
'.buildenv', 'var/cache/kiwi'
], xz_options=None
)
self.setup.export_package_verification.assert_called_once_with(
'target_dir'
)
self.setup.export_package_list.assert_called_once_with(
'target_dir'
)
@patch('kiwi.builder.archive.ArchiveCpio')
def test_create_cpio(self, mock_cpio):
Defaults.set_platform_name('x86_64')
self.archive.requested_archive_type = 'cpio'
archive = mock.Mock()
mock_cpio.return_value = archive
self.archive.create()
mock_cpio.assert_called_once_with(
'target_dir/myimage.x86_64-1.2.3.cpio'
)
archive.create.assert_called_once_with(
'root_dir', exclude=[
'image', '.kconfig', 'run/*', 'tmp/*',
'.buildenv', 'var/cache/kiwi'
]
)
self.setup.export_package_verification.assert_called_once_with(
'target_dir'
)
self.setup.export_package_list.assert_called_once_with(
'target_dir'
)
def teardown(self):
sys.argv = argv_kiwi_tests
def teardown_method(self, cls):
self.teardown()
|
