mirror of https://github.com/docker/docker-py.git
139 lines
5.0 KiB
Python
139 lines
5.0 KiB
Python
from docker.utils.socket import next_frame_size
|
|
from docker.utils.socket import read_exactly
|
|
|
|
from .base import BaseAPIIntegrationTest, BUSYBOX
|
|
from ..helpers import requires_api_version
|
|
|
|
|
|
class ExecTest(BaseAPIIntegrationTest):
|
|
def test_execute_command(self):
|
|
container = self.client.create_container(BUSYBOX, 'cat',
|
|
detach=True, stdin_open=True)
|
|
id = container['Id']
|
|
self.client.start(id)
|
|
self.tmp_containers.append(id)
|
|
|
|
res = self.client.exec_create(id, ['echo', 'hello'])
|
|
self.assertIn('Id', res)
|
|
|
|
exec_log = self.client.exec_start(res)
|
|
self.assertEqual(exec_log, b'hello\n')
|
|
|
|
def test_exec_command_string(self):
|
|
container = self.client.create_container(BUSYBOX, 'cat',
|
|
detach=True, stdin_open=True)
|
|
id = container['Id']
|
|
self.client.start(id)
|
|
self.tmp_containers.append(id)
|
|
|
|
res = self.client.exec_create(id, 'echo hello world')
|
|
self.assertIn('Id', res)
|
|
|
|
exec_log = self.client.exec_start(res)
|
|
self.assertEqual(exec_log, b'hello world\n')
|
|
|
|
def test_exec_command_as_user(self):
|
|
container = self.client.create_container(BUSYBOX, 'cat',
|
|
detach=True, stdin_open=True)
|
|
id = container['Id']
|
|
self.client.start(id)
|
|
self.tmp_containers.append(id)
|
|
|
|
res = self.client.exec_create(id, 'whoami', user='default')
|
|
self.assertIn('Id', res)
|
|
|
|
exec_log = self.client.exec_start(res)
|
|
self.assertEqual(exec_log, b'default\n')
|
|
|
|
def test_exec_command_as_root(self):
|
|
container = self.client.create_container(BUSYBOX, 'cat',
|
|
detach=True, stdin_open=True)
|
|
id = container['Id']
|
|
self.client.start(id)
|
|
self.tmp_containers.append(id)
|
|
|
|
res = self.client.exec_create(id, 'whoami')
|
|
self.assertIn('Id', res)
|
|
|
|
exec_log = self.client.exec_start(res)
|
|
self.assertEqual(exec_log, b'root\n')
|
|
|
|
def test_exec_command_streaming(self):
|
|
container = self.client.create_container(BUSYBOX, 'cat',
|
|
detach=True, stdin_open=True)
|
|
id = container['Id']
|
|
self.tmp_containers.append(id)
|
|
self.client.start(id)
|
|
|
|
exec_id = self.client.exec_create(id, ['echo', 'hello\nworld'])
|
|
self.assertIn('Id', exec_id)
|
|
|
|
res = b''
|
|
for chunk in self.client.exec_start(exec_id, stream=True):
|
|
res += chunk
|
|
self.assertEqual(res, b'hello\nworld\n')
|
|
|
|
def test_exec_start_socket(self):
|
|
container = self.client.create_container(BUSYBOX, 'cat',
|
|
detach=True, stdin_open=True)
|
|
container_id = container['Id']
|
|
self.client.start(container_id)
|
|
self.tmp_containers.append(container_id)
|
|
|
|
line = 'yay, interactive exec!'
|
|
# `echo` appends CRLF, `printf` doesn't
|
|
exec_id = self.client.exec_create(
|
|
container_id, ['printf', line], tty=True)
|
|
self.assertIn('Id', exec_id)
|
|
|
|
socket = self.client.exec_start(exec_id, socket=True)
|
|
self.addCleanup(socket.close)
|
|
|
|
next_size = next_frame_size(socket)
|
|
self.assertEqual(next_size, len(line))
|
|
data = read_exactly(socket, next_size)
|
|
self.assertEqual(data.decode('utf-8'), line)
|
|
|
|
def test_exec_start_detached(self):
|
|
container = self.client.create_container(BUSYBOX, 'cat',
|
|
detach=True, stdin_open=True)
|
|
container_id = container['Id']
|
|
self.client.start(container_id)
|
|
self.tmp_containers.append(container_id)
|
|
|
|
exec_id = self.client.exec_create(
|
|
container_id, ['printf', "asdqwe"])
|
|
self.assertIn('Id', exec_id)
|
|
|
|
response = self.client.exec_start(exec_id, detach=True)
|
|
|
|
self.assertEqual(response, "")
|
|
|
|
def test_exec_inspect(self):
|
|
container = self.client.create_container(BUSYBOX, 'cat',
|
|
detach=True, stdin_open=True)
|
|
id = container['Id']
|
|
self.client.start(id)
|
|
self.tmp_containers.append(id)
|
|
|
|
exec_id = self.client.exec_create(id, ['mkdir', '/does/not/exist'])
|
|
self.assertIn('Id', exec_id)
|
|
self.client.exec_start(exec_id)
|
|
exec_info = self.client.exec_inspect(exec_id)
|
|
self.assertIn('ExitCode', exec_info)
|
|
self.assertNotEqual(exec_info['ExitCode'], 0)
|
|
|
|
@requires_api_version('1.25')
|
|
def test_exec_command_with_env(self):
|
|
container = self.client.create_container(BUSYBOX, 'cat',
|
|
detach=True, stdin_open=True)
|
|
id = container['Id']
|
|
self.client.start(id)
|
|
self.tmp_containers.append(id)
|
|
|
|
res = self.client.exec_create(id, 'env', environment=["X=Y"])
|
|
self.assertIn('Id', res)
|
|
|
|
exec_log = self.client.exec_start(res)
|
|
self.assertIn(b'X=Y\n', exec_log)
|