diff options
| author | Dustin Falgout <dustin@falgout.us> | 2015-11-05 03:47:50 -0600 |
|---|---|---|
| committer | Dustin Falgout <dustin@falgout.us> | 2015-12-03 17:19:59 -0600 |
| commit | e0365fcbfbc3bb6db5f2ac65ef5a14c62ddaf572 (patch) | |
| tree | f08ea740d42d5a4cdda68ab9ce7e8cb1b0b7f2ea | |
| parent | 5f9f3a41458d1340af5d89de971fc947bdfdf7d6 (diff) | |
| download | docker-py-e0365fcbfbc3bb6db5f2ac65ef5a14c62ddaf572.tar.gz | |
Use new methods to raise exceptions in create_host_config.
Added tests for the new methods.
Closes #840
Signed-off-by: Dustin Falgout <dustin@falgout.us>
| -rw-r--r-- | docker/utils/utils.py | 49 | ||||
| -rw-r--r-- | tests/integration/container_test.py | 17 |
2 files changed, 30 insertions, 36 deletions
diff --git a/docker/utils/utils.py b/docker/utils/utils.py index fc3bfac..46e3eb9 100644 --- a/docker/utils/utils.py +++ b/docker/utils/utils.py @@ -513,14 +513,10 @@ def create_host_config(binds=None, port_bindings=None, lxc_conf=None, if mem_swappiness is not None: if version_lt(version, '1.20'): - raise errors.InvalidVersion( - 'mem_swappiness param not supported for API version < 1.20' - ) + raise host_config_version_error('mem_swappiness', '1.20') if not isinstance(mem_swappiness, int): - raise TypeError( - 'Invalid type for mem_swappiness param: expected int but' - ' found {0}'.format(type(mem_swappiness)) - ) + raise host_config_type_error(mem_swappiness, 'int') + host_config['MemorySwappiness'] = mem_swappiness if pid_mode not in (None, 'host'): @@ -571,9 +567,7 @@ def create_host_config(binds=None, port_bindings=None, lxc_conf=None, if group_add: if version_lt(version, '1.20'): - raise errors.InvalidVersion( - 'group_add param not supported for API version < 1.20' - ) + raise host_config_version_error('group_add', '1.20') host_config['GroupAdd'] = [six.text_type(grp) for grp in group_add] if dns is not None: @@ -581,10 +575,7 @@ def create_host_config(binds=None, port_bindings=None, lxc_conf=None, if security_opt is not None: if not isinstance(security_opt, list): - raise errors.DockerException( - 'Invalid type for security_opt param: expected list but found' - ' {0}'.format(type(security_opt)) - ) + raise host_config_type_error(security_opt, 'list') host_config['SecurityOpt'] = security_opt if volumes_from is not None: @@ -633,10 +624,7 @@ def create_host_config(binds=None, port_bindings=None, lxc_conf=None, if ulimits is not None: if not isinstance(ulimits, list): - raise errors.DockerException( - 'Invalid type for ulimits param: expected list but found' - ' {0}'.format(type(ulimits)) - ) + raise host_config_type_error(ulimits, 'list') host_config['Ulimits'] = [] for l in ulimits: if not isinstance(l, Ulimit): @@ -646,35 +634,24 @@ def create_host_config(binds=None, port_bindings=None, lxc_conf=None, if log_config is not None: if not isinstance(log_config, LogConfig): if not isinstance(log_config, dict): - raise errors.DockerException( - 'Invalid type for log_config param: expected LogConfig but' - ' found {0}'.format(type(log_config)) - ) + raise host_config_type_error(log_config, 'LogConfig') log_config = LogConfig(**log_config) host_config['LogConfig'] = log_config if cpu_quota: if not isinstance(cpu_quota, int): - raise TypeError( - 'Invalid type for cpu_quota param: expected int but' - ' found {0}'.format(type(cpu_quota)) - ) + raise host_config_type_error(cpu_quota, 'int') if version_lt(version, '1.19'): - raise errors.InvalidVersion( - 'cpu_quota param not supported for API version < 1.19' - ) + raise host_config_version_error('cpu_quota', '1.19') + host_config['CpuQuota'] = cpu_quota if cpu_period: if not isinstance(cpu_period, int): - raise TypeError( - 'Invalid type for cpu_period param: expected int but' - ' found {0}'.format(type(cpu_period)) - ) + raise host_config_type_error(cpu_period, 'int') if version_lt(version, '1.19'): - raise errors.InvalidVersion( - 'cpu_period param not supported for API version < 1.19' - ) + raise host_config_version_error('cpu_period', '1.19') + host_config['CpuPeriod'] = cpu_period return host_config diff --git a/tests/integration/container_test.py b/tests/integration/container_test.py index 0396514..76a4418 100644 --- a/tests/integration/container_test.py +++ b/tests/integration/container_test.py @@ -364,6 +364,23 @@ class CreateContainerTest(helpers.BaseTestCase): host_config = inspect['HostConfig'] self.assertIn('MemorySwappiness', host_config) + def test_create_host_config_exception_raising(self): + with self.assertRaises(TypeError) as exc: + ctnr1 = self.client.create_container( + BUSYBOX, 'true', + host_config=self.client.create_host_config(mem_swappiness='40') + ) + self.assertIn('Invalid type for', exc.exception.response.text) + self.client.remove_container(ctnr1.get('Id', ctnr1), force=True) + + with self.assertRaises(docker.errors.InvalidVersion) as exc: + ctnr2 = self.client.create_container( + BUSYBOX, 'true', + host_config=self.client.create_host_config(version='1.18', mem_swappiness=40) + ) + self.assertIn('param is not supported in', exc.exception.response.text) + self.client.remove_container(ctnr2.get('Id', ctnr2), force=True) + class VolumeBindTest(helpers.BaseTestCase): def setUp(self): |
