This commit is contained in:
Joffrey F 2015-04-28 10:44:32 -07:00
parent aceb7a69b5
commit f6b816c264
2 changed files with 17 additions and 8 deletions

View File

@ -3,11 +3,14 @@ from .. import errors
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

View File

@ -675,17 +675,23 @@ class DockerClientTest(Cleanup, base.BaseTestCase):
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(