Implement a basic inventory retrieval plugin using Nova

Change-Id: I07c6947eb3b3eaaa4503de91a0667a4ac6e60d73
Signed-off-by: Julien Danjou <julien@danjou.info>

climate/inventory/__init__.py

@@ -0,0 +1,31 @@
+# -*- encoding: utf-8 -*-
+#
+# Copyright © 2013 Julien Danjou <julien@danjou.info>
+#
+# 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 abc
+
+
+class Plugin(object):
+    """Base class for inventory plugins."""
+
+    __metaclass__ = abc.ABCMeta
+
+    @abc.abstractmethod
+    def list_hosts(self):
+        """List hosts."""
+
+    @abc.abstractmethod
+    def get_host_details(self, host):
+        """Get host details."""

climate/inventory/nova.py

@@ -0,0 +1,59 @@
+# -*- encoding: utf-8 -*-
+#
+# Copyright © 2013 Julien Danjou <julien@danjou.info>
+#
+# 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 os
+from oslo.config import cfg
+
+from novaclient import client
+from climate import inventory
+
+
+CLI_OPTIONS = [
+    cfg.StrOpt('os-username',
+               default=os.environ.get('OS_USERNAME', 'climate'),
+               help='Username to use for OpenStack service access'),
+    cfg.StrOpt('os-password',
+               default=os.environ.get('OS_PASSWORD', 'admin'),
+               help='Password to use for OpenStack service access'),
+    cfg.StrOpt('os-tenant-id',
+               default=os.environ.get('OS_TENANT_ID', ''),
+               help='Tenant ID to use for OpenStack service access'),
+    cfg.StrOpt('os-tenant-name',
+               default=os.environ.get('OS_TENANT_NAME', 'admin'),
+               help='Tenant name to use for OpenStack service access'),
+    cfg.StrOpt('os-auth-url',
+               default=os.environ.get('OS_AUTH_URL',
+                                      'http://localhost:5000/v2.0'),
+               help='Auth URL to use for openstack service access'),
+]
+cfg.CONF.register_cli_opts(CLI_OPTIONS)
+
+
+class NovaInventory(inventory.Plugin):
+
+    def __init__(self):
+        self.novaclient = client.Client(
+            "2",
+            username=cfg.CONF.os_username,
+            api_key=cfg.CONF.os_password,
+            auth_url=cfg.CONF.os_auth_url,
+            project_id=cfg.CONF.os_tenant_name or cfg.CONF.os_tenant_id)
+
+    def list_hosts(self):
+        return self.novaclient.hypervisors.list()
+
+    def get_host_details(self, host):
+        return self.novaclient.hypervisors.get(host)

tests/inventory/test_nova.py

@@ -0,0 +1,59 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+#
+# Copyright © 2013 Julien Danjou <julien@danjou.info>
+#
+# 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
+import stubout
+import unittest
+
+from climate.inventory import nova
+
+
+class ServiceTestCase(unittest.TestCase):
+
+    @staticmethod
+    def fake_hypervisors_list():
+        a = mock.MagicMock()
+        a.id = 1
+        b = mock.MagicMock()
+        b.id = 2
+        return [a, b]
+
+    @staticmethod
+    def fake_hypervisors_get(h):
+        return {'stuff': 'foobar',
+                'cpu_info': {'arch': 'x86'}}
+
+    def setUp(self):
+        self.i = nova.NovaInventory()
+        self.stubs = stubout.StubOutForTesting()
+        self.stubs.Set(self.i.novaclient.hypervisors, "list",
+                       self.fake_hypervisors_list)
+        self.stubs.Set(self.i.novaclient.hypervisors, "get",
+                       self.fake_hypervisors_get)
+
+    def test_list_hosts(self):
+        hosts = self.i.list_hosts()
+        self.assertEqual(len(hosts), 2)
+
+    def test_get_host_detail(self):
+        hosts = self.i.list_hosts()
+        detail = self.i.get_host_details(hosts[0])
+        self.assertEqual(detail['cpu_info']['arch'], 'x86')
+
+    def tearDown(self):
+        self.stubs.UnsetAll()
+        self.stubs.SmartUnsetAll()

tools/pip-requires

@@ -1,3 +1,4 @@
 eventlet
 iso8601
-oslo.config>=1.1.0
+oslo.config>=1.1.0
+python-novaclient

tools/test-requires

@@ -1 +1,3 @@
-nose
+nose
+mock
+mox