diff --git a/tuskarclient/tests/v1/test_resource_class.py b/tuskarclient/tests/v1/test_resource_class.py index 1a7f2d8..07d7006 100644 --- a/tuskarclient/tests/v1/test_resource_class.py +++ b/tuskarclient/tests/v1/test_resource_class.py @@ -33,3 +33,14 @@ class ResourceClassManagerTest(tutils.TestCase): self.assertEqual(self.rcm.list(), ['fake_resource_class']) self.rcm._list.assert_called_with('/v1/resource_classes') + + def test_create(self): + self.rcm._create = mock.Mock(return_value=['fake_resource_class']) + + self.assertEqual( + self.rcm.create(dummy='dummy resource class data'), + ['fake_resource_class']) + + self.rcm._create.assert_called_with( + '/v1/resource_classes', + {'dummy': 'dummy resource class data'}) diff --git a/tuskarclient/v1/resource_classes.py b/tuskarclient/v1/resource_classes.py index bb59c97..e7e1ae9 100644 --- a/tuskarclient/v1/resource_classes.py +++ b/tuskarclient/v1/resource_classes.py @@ -30,3 +30,6 @@ class ResourceClassManager(base.Manager): def get(self, id): return self._get(self._path(id)) + + def create(self, **kwargs): + return self._create(self._path(), kwargs)