From fdb88eaddfc05befc7e0fd3b9ac8165dfb30f5f1 Mon Sep 17 00:00:00 2001
From: Petr Blaho <pblaho@redhat.com>
Date: Thu, 11 Jul 2013 15:32:44 +0200
Subject: [PATCH] Adds ResourceClass and ResourceClassManager (list, get)

ResourceClass just inherit from Resource now.
ResourceClassManager implement list and get methods.
---
 .../tests/v1_0/test_resource_class.py         | 35 +++++++++++++++++++
 tuskarclient/v1_0/resource_classes.py         | 32 +++++++++++++++++
 2 files changed, 67 insertions(+)
 create mode 100644 tuskarclient/tests/v1_0/test_resource_class.py
 create mode 100644 tuskarclient/v1_0/resource_classes.py

diff --git a/tuskarclient/tests/v1_0/test_resource_class.py b/tuskarclient/tests/v1_0/test_resource_class.py
new file mode 100644
index 0000000..0c5e189
--- /dev/null
+++ b/tuskarclient/tests/v1_0/test_resource_class.py
@@ -0,0 +1,35 @@
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+import mock
+
+from tuskarclient.v1_0 import resource_classes
+import tuskarclient.tests.utils as tutils
+
+
+class ResourceClassManagerTest(tutils.TestCase):
+    def setUp(self):
+        super(ResourceClassManagerTest, self).setUp()
+        self.api = mock.Mock()
+        self.rcm = resource_classes.ResourceClassManager(self.api)
+
+    def test_get(self):
+        self.rcm._get = mock.Mock(return_value='fake_resource_class')
+
+        self.assertEqual(self.rcm.get(42), 'fake_resource_class')
+        self.rcm._get.assert_called_with('/v1/resource_classes/42')
+
+    def test_list(self):
+        self.rcm._list = mock.Mock(return_value=['fake_resource_class'])
+
+        self.assertEqual(self.rcm.list(), ['fake_resource_class'])
+        self.rcm._list.assert_called_with('/v1/resource_classes')
diff --git a/tuskarclient/v1_0/resource_classes.py b/tuskarclient/v1_0/resource_classes.py
new file mode 100644
index 0000000..bb59c97
--- /dev/null
+++ b/tuskarclient/v1_0/resource_classes.py
@@ -0,0 +1,32 @@
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+from tuskarclient.common import base
+
+
+class ResourceClass(base.Resource):
+    def __repr__(self):
+        return "<ResourceClass {0}>".format(self._info)
+
+
+class ResourceClassManager(base.Manager):
+    resource_class = ResourceClass
+
+    @staticmethod
+    def _path(id=None):
+        return '/v1/resource_classes/%s' % id if id else '/v1/resource_classes'
+
+    def list(self):
+        return self._list(self._path())
+
+    def get(self, id):
+        return self._get(self._path(id))