From 6fe5d2b54351143ee4eab090ccd58c5067985078 Mon Sep 17 00:00:00 2001
From: George Lester <glester491@gmail.com>
Date: Tue, 5 Jul 2016 23:43:25 -0700
Subject: [PATCH] Implemented oom_score_adj

Signed-off-by: George Lester <glester491@gmail.com>
---
 compose/config/config.py               |  1 +
 compose/config/config_schema_v2.0.json |  1 +
 compose/service.py                     |  2 ++
 docs/compose-file.md                   |  2 +-
 tests/integration/service_test.py      |  5 +++++
 tests/unit/config/config_test.py       | 20 ++++++++++++++++++++
 6 files changed, 30 insertions(+), 1 deletion(-)

diff --git a/compose/config/config.py b/compose/config/config.py
index 7a2b3d366..d3ab1d4be 100644
--- a/compose/config/config.py
+++ b/compose/config/config.py
@@ -70,6 +70,7 @@ DOCKER_CONFIG_KEYS = [
     'mem_limit',
     'memswap_limit',
     'net',
+    'oom_score_adj'
     'pid',
     'ports',
     'privileged',
diff --git a/compose/config/config_schema_v2.0.json b/compose/config/config_schema_v2.0.json
index e84d13179..c08fa4d7a 100644
--- a/compose/config/config_schema_v2.0.json
+++ b/compose/config/config_schema_v2.0.json
@@ -166,6 +166,7 @@
             }
           ]
         },
+        "oom_score_adj": {"type": "integer", "minimum": -1000, "maximum": 1000},
         "pid": {"type": ["string", "null"]},
 
         "ports": {
diff --git a/compose/service.py b/compose/service.py
index af572e5b5..733814666 100644
--- a/compose/service.py
+++ b/compose/service.py
@@ -53,6 +53,7 @@ DOCKER_START_KEYS = [
     'log_opt',
     'mem_limit',
     'memswap_limit',
+    'oom_score_adj',
     'pid',
     'privileged',
     'restart',
@@ -695,6 +696,7 @@ class Service(object):
             cpu_quota=options.get('cpu_quota'),
             shm_size=options.get('shm_size'),
             tmpfs=options.get('tmpfs'),
+            oom_score_adj=options.get('oom_score_adj')
         )
 
     def build(self, no_cache=False, pull=False, force_rm=False):
diff --git a/docs/compose-file.md b/docs/compose-file.md
index 464cf2716..f7b5a931c 100644
--- a/docs/compose-file.md
+++ b/docs/compose-file.md
@@ -715,7 +715,7 @@ then read-write will be used.
 >     - container_name
 >     - container_name:rw
 
-### cpu\_shares, cpu\_quota, cpuset, domainname, hostname, ipc, mac\_address, mem\_limit, memswap\_limit, privileged, read\_only, restart, shm\_size, stdin\_open, tty, user, working\_dir
+### cpu\_shares, cpu\_quota, cpuset, domainname, hostname, ipc, mac\_address, mem\_limit, memswap\_limit, oom_score_adj, privileged, read\_only, restart, shm\_size, stdin\_open, tty, user, working\_dir
 
 Each of these is a single value, analogous to its
 [docker run](https://docs.docker.com/engine/reference/run/) counterpart.
diff --git a/tests/integration/service_test.py b/tests/integration/service_test.py
index 1801f5bfc..02f9cc0ba 100644
--- a/tests/integration/service_test.py
+++ b/tests/integration/service_test.py
@@ -854,6 +854,11 @@ class ServiceTest(DockerClientTestCase):
         container = create_and_start_container(service)
         self.assertEqual(container.get('HostConfig.RestartPolicy.Name'), 'always')
 
+    def test_oom_score_adj_value(self):
+        service = self.create_service('web', oom_score_adj=500)
+        container = create_and_start_container(service)
+        self.assertEqual(container.get('HostConfig.OomScoreAdj'), 500)
+
     def test_restart_on_failure_value(self):
         service = self.create_service('web', restart={
             'Name': 'on-failure',
diff --git a/tests/unit/config/config_test.py b/tests/unit/config/config_test.py
index 2dad224ba..1be8aefa2 100644
--- a/tests/unit/config/config_test.py
+++ b/tests/unit/config/config_test.py
@@ -1243,6 +1243,26 @@ class ConfigTest(unittest.TestCase):
             }
         ]
 
+    def test_oom_score_adj_option(self):
+
+        actual = config.load(build_config_details({
+            'version': '2',
+            'services': {
+                'web': {
+                    'image': 'alpine',
+                    'oom_score_adj': 500
+                }
+            }
+        }))
+
+        assert actual.services == [
+            {
+                'name': 'web',
+                'image': 'alpine',
+                'oom_score_adj': 500
+            }
+        ]
+
     def test_merge_service_dicts_from_files_with_extends_in_base(self):
         base = {
             'volumes': ['.:/app'],