From ca27bda1016a911e2ccc6a3297030970f0ee7827 Mon Sep 17 00:00:00 2001 From: Petr Blaho Date: Tue, 16 Jul 2013 15:26:19 +0200 Subject: [PATCH] Adds update method to ResourceClassManager with tests --- tuskarclient/tests/v1/test_resource_class.py | 11 +++++++++++ tuskarclient/v1/resource_classes.py | 3 +++ 2 files changed, 14 insertions(+) diff --git a/tuskarclient/tests/v1/test_resource_class.py b/tuskarclient/tests/v1/test_resource_class.py index d76f4b3..be128a5 100644 --- a/tuskarclient/tests/v1/test_resource_class.py +++ b/tuskarclient/tests/v1/test_resource_class.py @@ -45,6 +45,17 @@ class ResourceClassManagerTest(tutils.TestCase): '/v1/resource_classes', {'dummy': 'dummy resource class data'}) + def test_update(self): + self.rcm._update = mock.Mock(return_value=['fake_resource_class']) + + self.assertEqual( + self.rcm.update(42, dummy='dummy resource class data'), + ['fake_resource_class']) + + self.rcm._update.assert_called_with( + '/v1/resource_classes/42', + {'dummy': 'dummy resource class data'}) + def test_delete(self): self.rcm._delete = mock.Mock(return_value=None) diff --git a/tuskarclient/v1/resource_classes.py b/tuskarclient/v1/resource_classes.py index b11f1a5..fbc94da 100644 --- a/tuskarclient/v1/resource_classes.py +++ b/tuskarclient/v1/resource_classes.py @@ -34,5 +34,8 @@ class ResourceClassManager(base.Manager): def create(self, **kwargs): return self._create(self._path(), kwargs) + def update(self, id, **kwargs): + return self._update(self._single_path(id), kwargs) + def delete(self, id): return self._delete(self._single_path(id))