Fix #573

Author: shin-

docker/utils/decorators.py

@@ -3,11 +3,14 @@
 
 def check_resource(f):
     def wrapped(self, resource_id=None, *args, **kwargs):
-        if resource_id is None and (
-            kwargs.get('container') is None and kwargs.get('image') is None
-        ):
-            raise errors.NullResource(
-                'image or container param is None'
-            )
+        if resource_id is None:
+            if kwargs.get('container'):
+                resource_id = kwargs.pop('container')
+            elif kwargs.get('image'):
+                resource_id = kwargs.pop('image')
+            else:
+                raise errors.NullResource(
+                    'image or container param is undefined'
+                )
         return f(self, resource_id, *args, **kwargs)
     return wrapped

tests/test.py

@@ -675,17 +675,23 @@ def test_start_container_none(self):
         try:
             self.client.start(container=None)
         except ValueError as e:
-            self.assertEqual(str(e), 'image or container param is None')
+            self.assertEqual(str(e), 'image or container param is undefined')
         else:
             self.fail('Command should raise ValueError')
 
         try:
             self.client.start(None)
         except ValueError as e:
-            self.assertEqual(str(e), 'image or container param is None')
+            self.assertEqual(str(e), 'image or container param is undefined')
         else:
             self.fail('Command should raise ValueError')
 
+    def test_start_container_regression_573(self):
+        try:
+            self.client.start(**{'container': fake_api.FAKE_CONTAINER_ID})
+        except Exception as e:
+            self.fail('Command should not raise exception: {0}'.format(e))
+
     def test_create_container_with_lxc_conf(self):
         try:
             self.client.create_container(