Jelajahi Sumber

Test cgroup_parent option is being sent.

Signed-off-by: Mazz Mosley <[email protected]>
Mazz Mosley 10 tahun lalu
induk
melakukan
ca36628a0e
1 mengubah file dengan 12 tambahan dan 7 penghapusan
  1. 12 7
      tests/unit/service_test.py

+ 12 - 7
tests/unit/service_test.py

@@ -146,6 +146,18 @@ class ServiceTest(unittest.TestCase):
             2000000000
         )
 
+    def test_cgroup_parent(self):
+        self.mock_client.create_host_config.return_value = {}
+
+        service = Service(name='foo', image='foo', hostname='name', client=self.mock_client, cgroup_parent='test')
+        service._get_container_create_options({'some': 'overrides'}, 1)
+
+        self.assertTrue(self.mock_client.create_host_config.called)
+        self.assertEqual(
+            self.mock_client.create_host_config.call_args[1]['cgroup_parent'],
+            'test'
+        )
+
     def test_log_opt(self):
         self.mock_client.create_host_config.return_value = {}
 
@@ -353,13 +365,6 @@ class ServiceTest(unittest.TestCase):
         service.create_container(do_build=False)
         self.assertFalse(self.mock_client.build.called)
 
-    def test_create_container_no_build_cgroup_parent(self):
-        service = Service('foo', client=self.mock_client, build='.')
-        service.image = lambda: {'Id': 'abc123'}
-
-        service.create_container(do_build=False, cgroup_parent='test')
-        self.assertFalse(self.mock_client.build.called)
-
     def test_create_container_no_build_but_needs_build(self):
         service = Service('foo', client=self.mock_client, build='.')
         service.image = lambda *args, **kwargs: mock_get_image([])