Add support for storage_opt in host_config

Signed-off-by: Joffrey F <[email protected]>

Author: shin-

docker/api/container.py

@@ -549,6 +549,8 @@ def create_host_config(self, *args, **kwargs):
             security_opt (:py:class:`list`): A list of string values to
                 customize labels for MLS systems, such as SELinux.
             shm_size (str or int): Size of /dev/shm (e.g. ``1G``).
+            storage_opt (dict): Storage driver options per container as a
+                key-value mapping.
             sysctls (dict): Kernel parameters to set in the container.
             tmpfs (dict): Temporary filesystems to mount, as a dictionary
                 mapping a path inside the container to options for that path.

docker/models/containers.py

@@ -586,6 +586,8 @@ def run(self, image, command=None, stdout=True, stderr=False,
                 Default: ``False``.
             stop_signal (str): The stop signal to use to stop the container
                 (e.g. ``SIGINT``).
+            storage_opt (dict): Storage driver options per container as a
+                key-value mapping.
             sysctls (dict): Kernel parameters to set in the container.
             tmpfs (dict): Temporary filesystems to mount, as a dictionary
                 mapping a path inside the container to options for that path.
@@ -833,6 +835,7 @@ def prune(self, filters=None):
     'restart_policy',
     'security_opt',
     'shm_size',
+    'storage_opt',
     'sysctls',
     'tmpfs',
     'ulimits',

docker/types/containers.py

@@ -117,7 +117,7 @@ def __init__(self, version, binds=None, port_bindings=None,
                  oom_kill_disable=False, shm_size=None, sysctls=None,
                  tmpfs=None, oom_score_adj=None, dns_opt=None, cpu_shares=None,
                  cpuset_cpus=None, userns_mode=None, pids_limit=None,
-                 isolation=None, auto_remove=False):
+                 isolation=None, auto_remove=False, storage_opt=None):
 
         if mem_limit is not None:
             self['Memory'] = parse_bytes(mem_limit)
@@ -412,6 +412,11 @@ def __init__(self, version, binds=None, port_bindings=None,
                 raise host_config_version_error('auto_remove', '1.25')
             self['AutoRemove'] = auto_remove
 
+        if storage_opt is not None:
+            if version_lt(version, '1.24'):
+                raise host_config_version_error('storage_opt', '1.24')
+            self['StorageOpt'] = storage_opt
+
 
 def host_config_type_error(param, param_value, expected):
     error_msg = 'Invalid type for {0} param: expected {1} but found {2}'

tests/integration/api_container_test.py

@@ -422,6 +422,22 @@ def test_create_with_stop_timeout(self):
         config = self.client.inspect_container(container)
         assert config['Config']['StopTimeout'] == 25
 
+    @requires_api_version('1.24')
+    def test_create_with_storage_opt(self):
+        if self.client.info()['Driver'] == 'aufs':
+            return pytest.skip('Not supported on AUFS')
+        host_config = self.client.create_host_config(
+            storage_opt={'size': '120G'}
+        )
+        container = self.client.create_container(
+            BUSYBOX, ['echo', 'test'], host_config=host_config
+        )
+        self.tmp_containers.append(container)
+        config = self.client.inspect_container(container)
+        assert config['HostConfig']['StorageOpt'] == {
+            'size': '120G'
+        }
+
 
 class VolumeBindTest(BaseAPIIntegrationTest):
     def setUp(self):

tests/integration/base.py

@@ -75,6 +75,10 @@ def setUp(self):
             version=TEST_API_VERSION, timeout=60, **kwargs_from_env()
         )
 
+    def tearDown(self):
+        super(BaseAPIIntegrationTest, self).tearDown()
+        self.client.close()
+
     def run_container(self, *args, **kwargs):
         container = self.client.create_container(*args, **kwargs)
         self.tmp_containers.append(container)