/
test_runlike.py
205 lines (153 loc) · 6.61 KB
/
test_runlike.py
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
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
import unittest
import os
import pipes
from click.testing import CliRunner
from subprocess import check_output
from runlike.runlike import cli
from typing import List
def setUpModule():
check_output("./fixtures.sh")
class BaseTest(unittest.TestCase):
@classmethod
def start_runlike(cls, args: List[str]):
runner = CliRunner()
cls.outputs = [""] * 6
for i in range(1, 6):
result = runner.invoke(cli, args + [f"runlike_fixture{i}"])
assert result.exit_code == 0, "runlike did not finish successfully"
cls.outputs[i] = result.output
def starts_with(self, prefix, fixture_index=1):
hay = self.outputs[fixture_index]
assert hay.startswith(prefix), f"{hay}\ndoes not start with {prefix}"
def expect_substr(self, substr, fixture_index=1):
hay = self.outputs[fixture_index]
self.assertIn(substr, hay)
def dont_expect_substr(self, substr, fixture_index=1):
hay = self.outputs[fixture_index]
self.assertNotIn(substr, hay)
class TestRunlike(BaseTest):
@classmethod
def setUpClass(cls):
cls.start_runlike(["--pretty"])
def test_tcp_port(self):
self.expect_substr("-p 300 \\")
def test_tcp_port_with_host_port(self):
self.expect_substr("-p 400:400 \\")
def test_expose(self):
self.expect_substr("--expose=1000 \\")
def test_udp(self):
self.expect_substr("-p 301/udp \\")
def test_dns(self):
self.expect_substr("--dns=8.8.8.8 \\")
self.expect_substr("--dns=8.8.4.4 \\")
def test_udp_with_host_port(self):
self.expect_substr("-p 503:502/udp \\")
def test_udp_with_host_port_and_ip(self):
self.expect_substr("-p 127.0.0.1:601:600/udp \\")
def test_host_volumes(self):
cur_dir = os.path.dirname(os.path.realpath(__file__))
self.expect_substr("--volume=%s:/workdir" % pipes.quote(cur_dir))
def test_no_host_volume(self):
self.expect_substr('--volume=/random_volume')
def test_tty(self):
self.expect_substr('-t \\')
self.dont_expect_substr('-t \\', 2)
def test_autoremove(self):
self.expect_substr('--rm \\', 5)
def test_restart_always(self):
self.expect_substr('--restart=always \\')
def test_restart_on_failure(self):
self.expect_substr('--restart=on-failure \\', 2)
def test_restart_with_max(self):
self.expect_substr('--restart=on-failure:3 \\', 3)
def test_restart_not_present(self):
self.dont_expect_substr('--restart', 4)
def test_hostname(self):
self.expect_substr('--hostname=Essos \\')
def test_hostname_not_present(self):
self.dont_expect_substr('--hostname \\', 2)
def test_network_mode(self):
self.dont_expect_substr('--network', 1)
self.expect_substr('--network=host', 2)
self.expect_substr('--network=runlike_fixture_bridge', 3)
def test_privileged_mode(self):
self.expect_substr('--privileged \\')
def test_privileged_not_present(self):
self.dont_expect_substr('--privileged \\', 2)
def test_multi_labels(self):
self.expect_substr("--label='com.example.environment=test' \\", 1)
self.expect_substr(
"--label='com.example.notescaped=$KEEP_DOLLAR' \\", 1)
self.dont_expect_substr("--label='ImageLabel", 1)
def test_one_label(self):
self.expect_substr("--label='com.example.version=1' \\", 2)
self.dont_expect_substr("--label='ImageLabel", 2)
def test_labels_not_present(self):
self.dont_expect_substr('--label', 3)
def test_extra_hosts(self):
self.expect_substr('--add-host hostname2:127.0.0.2 \\', 1)
self.expect_substr('--add-host hostname3:127.0.0.3 \\', 1)
def test_extra_hosts_not_present(self):
self.dont_expect_substr('--add-host', 2)
def test_log_driver_default_no_opts(self):
self.dont_expect_substr('--log-driver', 2)
self.dont_expect_substr('--log-opt', 2)
def test_log_driver_default_with_opts(self):
self.dont_expect_substr('--log-driver', 3)
self.expect_substr('--log-opt mode=non-blocking \\', 3)
self.expect_substr('--log-opt max-buffer-size=4m \\', 3)
def test_log_driver_present(self):
self.expect_substr('--log-driver=fluentd \\')
def test_log_driver_options_present(self):
self.expect_substr('--log-opt fluentd-async-connect=true \\')
self.expect_substr('--log-opt tag=docker.runlike \\')
def test_links(self):
self.expect_substr('--link runlike_fixture4:alias_of4 \\', 5)
self.expect_substr('--link runlike_fixture1 \\', 5)
def test_command(self):
self.dont_expect_substr('/bin/bash', 1)
self.expect_substr('/bin/bash sleep.sh', 2)
self.expect_substr("bash -c 'bash sleep.sh'", 3)
self.expect_substr(r"bash -c 'bash \'sleep.sh\'", 4)
def test_user(self):
self.expect_substr('--user=daemon')
self.dont_expect_substr('--user', 2)
def test_mac_address(self):
self.expect_substr('--mac-address=6a:00:01:ad:d9:e0', 4)
self.dont_expect_substr('--mac-address', 2)
def test_env(self):
val = '''FOO=thing="quoted value with 'spaces' and 'single quotes'"'''
self.expect_substr("""--env=%s""" % pipes.quote(val))
self.expect_substr("--env=SET_WITHOUT_VALUE")
self.dont_expect_substr("--env=IMAGE_ENV")
self.expect_substr("--env='UTF_8=ユーザー別サイト'")
def test_cap_add(self):
self.expect_substr("--cap-add=CHOWN")
def test_devices(self):
self.expect_substr("--device /dev/null:/dev/null:r")
self.expect_substr("--device /dev/null:/dev/null", 2)
def test_workdir(self):
self.expect_substr("--workdir=/workdir")
self.dont_expect_substr('--workdir', 2)
def test_runtime(self):
self.expect_substr('--runtime=runc')
def test_pid_mode(self):
self.expect_substr('--pid host', 2)
self.dont_expect_substr('--pid')
def test_memory(self):
self.expect_substr('--memory="2147483648"')
def test_memory_reservation(self):
self.expect_substr('--memory-reservation="1610612736"')
def test_cpuset(self):
self.expect_substr('--cpuset-cpus=0', 3)
self.expect_substr('--cpuset-mems=0', 3)
def test_starts_with_docker_run(self):
self.starts_with('docker run ')
def test_name(self):
self.expect_substr('--name=runlike_fixture1')
class TestRunlikeNoName(BaseTest):
@classmethod
def setUpClass(cls):
cls.start_runlike(["--no-name"])
def test_no_name(self):
self.dont_expect_substr('--name')