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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
import io
import json
import unittest
try:
# Python >= 3.10
from collections.abc import Iterable
except ImportError:
# Python < 3.10
from collections.abc import Iterable
from unittest.mock import patch
import requests_mock
from podman import PodmanClient, api, tests
from podman.domain.images import Image
from podman.errors import BuildError, DockerException
class TestBuildCase(unittest.TestCase):
"""Test ImagesManager build().
Note:
Mock responses need to be coded for libpod returns. The python bindings are responsible
for mapping to compatible output.
"""
def setUp(self) -> None:
super().setUp()
self.client = PodmanClient(base_url=tests.BASE_SOCK)
def tearDown(self) -> None:
super().tearDown()
self.client.close()
@patch.object(api, "create_tar")
@patch.object(api, "prepare_containerfile")
def test_build(self, mock_prepare_containerfile, mock_create_tar):
mock_prepare_containerfile.return_value = "Containerfile"
mock_create_tar.return_value = b"This is a mocked tarball."
stream = [
{"stream": " ---\u003e a9eb17255234"},
{"stream": "Step 1 : VOLUME /data"},
{"stream": " ---\u003e Running in abdc1e6896c6"},
{"stream": " ---\u003e 713bca62012e"},
{"stream": "Removing intermediate container abdc1e6896c6"},
{"stream": "Step 2 : CMD [\"/bin/sh\"]"},
{"stream": " ---\u003e Running in dba30f2a1a7e"},
{"stream": " ---\u003e 032b8b2855fc"},
{"stream": "Removing intermediate container dba30f2a1a7e"},
{"stream": "032b8b2855fc\n"},
]
buffer = io.StringIO()
for entry in stream:
buffer.write(json.JSONEncoder().encode(entry))
buffer.write("\n")
with requests_mock.Mocker() as mock:
mock.post(
tests.LIBPOD_URL + "/build"
"?t=latest"
"&buildargs=%7B%22BUILD_DATE%22%3A+%22January+1%2C+1970%22%7D"
"&cpuperiod=10"
"&extrahosts=%7B%22database%22%3A+%22127.0.0.1%22%7D"
"&labels=%7B%22Unittest%22%3A+%22true%22%7D",
text=buffer.getvalue(),
)
mock.get(
tests.LIBPOD_URL + "/images/032b8b2855fc/json",
json={
"Id": "032b8b2855fc",
"ParentId": "",
"RepoTags": ["fedora:latest", "fedora:33", "<none>:<none>"],
"RepoDigests": [
"fedora@sha256:9598a10fa72b402db876ccd4b3d240a4061c7d1e442745f1896ba37e1bf38664"
],
"Created": 1614033320,
"Size": 23855104,
"VirtualSize": 23855104,
"SharedSize": 0,
"Labels": {},
"Containers": 2,
},
)
image, logs = self.client.images.build(
path="/tmp/context_dir",
tag="latest",
buildargs={
"BUILD_DATE": "January 1, 1970",
},
container_limits={
"cpuperiod": 10,
},
extra_hosts={"database": "127.0.0.1"},
labels={"Unittest": "true"},
)
self.assertIsInstance(image, Image)
self.assertEqual(image.id, "032b8b2855fc")
self.assertIsInstance(logs, Iterable)
@patch.object(api, "create_tar")
@patch.object(api, "prepare_containerfile")
def test_build_logged_error(self, mock_prepare_containerfile, mock_create_tar):
mock_prepare_containerfile.return_value = "Containerfile"
mock_create_tar.return_value = b"This is a mocked tarball."
stream = [
{"error": "We do not need any stinking badges."},
]
buffer = io.StringIO()
for entry in stream:
buffer.write(json.JSONEncoder().encode(entry))
buffer.write("\n")
with requests_mock.Mocker() as mock:
mock.post(
tests.LIBPOD_URL + "/build",
text=buffer.getvalue(),
)
with self.assertRaises(BuildError) as e:
self.client.images.build(path="/tmp/context_dir")
self.assertEqual(e.exception.msg, "We do not need any stinking badges.")
@requests_mock.Mocker()
def test_build_no_context(self, mock):
mock.post(tests.LIBPOD_URL + "/images/build")
with self.assertRaises(TypeError):
self.client.images.build()
@requests_mock.Mocker()
def test_build_encoding(self, mock):
mock.post(tests.LIBPOD_URL + "/images/build")
with self.assertRaises(DockerException):
self.client.images.build(path="/root", gzip=True, encoding="utf-8")
if __name__ == '__main__':
unittest.main()
|
