diff --git a/tuskarclient/tests/v1/test_racks.py b/tuskarclient/tests/v1/test_racks.py index abbde09..1b47da0 100644 --- a/tuskarclient/tests/v1/test_racks.py +++ b/tuskarclient/tests/v1/test_racks.py @@ -35,6 +35,28 @@ class RackManagerTest(tutils.TestCase): self.assertEqual(self.rm.list(), ['fake_rack']) self.rm._list.assert_called_with('/v1/racks') + def test_create(self): + self.rm._create = mock.Mock(return_value=['fake_rack']) + + self.assertEqual( + self.rm.create(dummy='dummy rack data'), + ['fake_rack']) + + self.rm._create.assert_called_with( + '/v1/racks', + {'dummy': 'dummy rack data'}) + + def test_update(self): + self.rm._update = mock.Mock(return_value=['fake_rack']) + + self.assertEqual( + self.rm.update(42, dummy='dummy rack data'), + ['fake_rack']) + + self.rm._update.assert_called_with( + '/v1/racks/42', + {'dummy': 'dummy rack data'}) + def test_delete(self): self.rm._delete = mock.Mock(return_value=None) diff --git a/tuskarclient/v1/racks.py b/tuskarclient/v1/racks.py index e1125cb..41fbcda 100644 --- a/tuskarclient/v1/racks.py +++ b/tuskarclient/v1/racks.py @@ -33,5 +33,11 @@ class RackManager(base.Manager): def list(self): return self._list(self._path()) + def create(self, **kwargs): + return self._create(self._path(), kwargs) + + def update(self, rack_id, **kwargs): + return self._update(self._single_path(rack_id), kwargs) + def delete(self, rack_id): return self._delete(self._single_path(rack_id))