133 lines
4.2 KiB
Python
133 lines
4.2 KiB
Python
#!/usr/bin/env python
|
|
# -*- encoding: utf-8 -*-
|
|
#
|
|
# Copyright © 2013 eNovance <licensing@enovance.com>
|
|
#
|
|
# Author: 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
|
|
|
|
from ceilometer.tests import base
|
|
from ceilometer import nova_client
|
|
|
|
|
|
class TestNovaClient(base.TestCase):
|
|
|
|
def setUp(self):
|
|
super(TestNovaClient, self).setUp()
|
|
self.nv = nova_client.Client()
|
|
|
|
@staticmethod
|
|
def fake_flavors_get(*args, **kwargs):
|
|
a = mock.MagicMock()
|
|
a.id = args[0]
|
|
if a.id == 1:
|
|
a.name = 'm1.tiny'
|
|
elif a.id == 2:
|
|
a.name = 'm1.large'
|
|
else:
|
|
return None
|
|
return a
|
|
|
|
@staticmethod
|
|
def fake_images_get(*args, **kwargs):
|
|
a = mock.MagicMock()
|
|
a.id = args[0]
|
|
if a.id == 1:
|
|
a.name = 'ubuntu-12.04-x86'
|
|
a.metadata = {'kernel_id': 11,
|
|
'ramdisk_id': 21}
|
|
elif a.id == 2:
|
|
a.name = 'centos-5.4-x64'
|
|
a.metadata = {'kernel_id': 12,
|
|
'ramdisk_id': 22}
|
|
else:
|
|
return None
|
|
|
|
return a
|
|
|
|
@staticmethod
|
|
def fake_flavors_list():
|
|
a = mock.MagicMock()
|
|
a.id = 1
|
|
a.name = 'm1.tiny'
|
|
b = mock.MagicMock()
|
|
b.id = 2
|
|
b.name = 'm1.large'
|
|
return [a, b]
|
|
|
|
@staticmethod
|
|
def fake_servers_list(*args, **kwargs):
|
|
a = mock.MagicMock()
|
|
a.id = 42
|
|
a.flavor = {'id': 1}
|
|
a.image = {'id': 1}
|
|
return [a]
|
|
|
|
def test_instance_get_all_by_host(self):
|
|
self.stubs.Set(self.nv.nova_client.flavors, 'get',
|
|
self.fake_flavors_get)
|
|
self.stubs.Set(self.nv.nova_client.servers, 'list',
|
|
self.fake_servers_list)
|
|
self.stubs.Set(self.nv.nova_client.images, 'get',
|
|
self.fake_images_get)
|
|
|
|
instances = self.nv.instance_get_all_by_host('foobar')
|
|
self.assertEqual(len(instances), 1)
|
|
self.assertEqual(instances[0].flavor['name'], 'm1.tiny')
|
|
self.assertEqual(instances[0].image['name'], 'ubuntu-12.04-x86')
|
|
self.assertEqual(instances[0].kernel_id, 11)
|
|
self.assertEqual(instances[0].ramdisk_id, 21)
|
|
|
|
@staticmethod
|
|
def fake_servers_list_unknown_flavor(*args, **kwargs):
|
|
a = mock.MagicMock()
|
|
a.id = 42
|
|
a.flavor = {'id': 666}
|
|
a.image = {'id': 1}
|
|
return [a]
|
|
|
|
def test_instance_get_all_by_host_unknown_flavor(self):
|
|
self.stubs.Set(self.nv.nova_client.flavors, 'get',
|
|
self.fake_flavors_get)
|
|
self.stubs.Set(self.nv.nova_client.servers, 'list',
|
|
self.fake_servers_list_unknown_flavor)
|
|
self.stubs.Set(self.nv.nova_client.images, 'get',
|
|
self.fake_images_get)
|
|
|
|
instances = self.nv.instance_get_all_by_host('foobar')
|
|
self.assertEqual(len(instances), 1)
|
|
self.assertEqual(instances[0].flavor['name'], 'unknown-id-666')
|
|
|
|
@staticmethod
|
|
def fake_servers_list_unknown_image(*args, **kwargs):
|
|
a = mock.MagicMock()
|
|
a.id = 42
|
|
a.flavor = {'id': 1}
|
|
a.image = {'id': 666}
|
|
return [a]
|
|
|
|
def test_instance_get_all_by_host_unknown_image(self):
|
|
self.stubs.Set(self.nv.nova_client.flavors, 'get',
|
|
self.fake_flavors_get)
|
|
self.stubs.Set(self.nv.nova_client.servers, 'list',
|
|
self.fake_servers_list_unknown_image)
|
|
self.stubs.Set(self.nv.nova_client.images, 'get',
|
|
self.fake_images_get)
|
|
|
|
instances = self.nv.instance_get_all_by_host('foobar')
|
|
self.assertEqual(len(instances), 1)
|
|
self.assertEqual(instances[0].image['name'], 'unknown-id-666')
|