diff --git a/blazarclient/client.py b/blazarclient/client.py index 54b1303..b35dc22 100644 --- a/blazarclient/client.py +++ b/blazarclient/client.py @@ -34,4 +34,6 @@ def Client(version=1, service_type='reservation', *args, **kwargs): }) raise exception.UnsupportedVersion(msg) - return importutils.import_object(client_path, *args, **kwargs) + return importutils.import_object(client_path, + service_type=service_type, + *args, **kwargs) diff --git a/blazarclient/tests/test_client.py b/blazarclient/tests/test_client.py index 2c8cc4e..388db6d 100644 --- a/blazarclient/tests/test_client.py +++ b/blazarclient/tests/test_client.py @@ -32,12 +32,14 @@ class BaseClientTestCase(tests.TestCase): def test_with_v1(self): self.client.Client() self.import_obj.assert_called_once_with( - 'blazarclient.v1.client.Client') + 'blazarclient.v1.client.Client', + service_type='reservation') def test_with_v1a0(self): self.client.Client(version='1a0') self.import_obj.assert_called_once_with( - 'blazarclient.v1.client.Client') + 'blazarclient.v1.client.Client', + service_type='reservation') def test_with_wrong_vers(self): self.assertRaises(exception.UnsupportedVersion,