nova/nova/tests/unit/objects/test_diagnostics.py

147 lines
6.5 KiB
Python

# Copyright (c) 2014 VMware, Inc.
#
# 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 nova.objects import diagnostics
from nova import test
class DiagnosticsComparisonMixin(object):
def assertDiagnosticsEqual(self, expected, actual):
expected.obj_reset_changes(recursive=True)
actual.obj_reset_changes(recursive=True)
# NOTE(snikitin): Fields 'cpu_details', 'disk_details' and
# 'nic_details' are list objects. They wouldn't be marked as 'changed'
# if new item will be added by 'append()' method
# (like # Diagnostics.add_*** methods do). So we have to reset these
# objects manually.
for obj in [expected, actual]:
for field in ['cpu_details', 'disk_details', 'nic_details']:
for item in getattr(obj, field):
item.obj_reset_changes()
self.assertEqual(expected.obj_to_primitive(),
actual.obj_to_primitive())
class DiagnosticsTests(test.NoDBTestCase):
def test_cpu_diagnostics(self):
cpu = diagnostics.CpuDiagnostics(id=1, time=7, utilisation=15)
self.assertEqual(1, cpu.id)
self.assertEqual(7, cpu.time)
self.assertEqual(15, cpu.utilisation)
def test_nic_diagnostics(self):
nic = diagnostics.NicDiagnostics(
mac_address='00:00:ca:fe:00:00',
rx_octets=1, rx_errors=2, rx_drop=3, rx_packets=4, rx_rate=5,
tx_octets=6, tx_errors=7, tx_drop=8, tx_packets=9, tx_rate=10)
self.assertEqual('00:00:ca:fe:00:00', nic.mac_address)
self.assertEqual(1, nic.rx_octets)
self.assertEqual(2, nic.rx_errors)
self.assertEqual(3, nic.rx_drop)
self.assertEqual(4, nic.rx_packets)
self.assertEqual(5, nic.rx_rate)
self.assertEqual(6, nic.tx_octets)
self.assertEqual(7, nic.tx_errors)
self.assertEqual(8, nic.tx_drop)
self.assertEqual(9, nic.tx_packets)
self.assertEqual(10, nic.tx_rate)
def test_disk_diagnostics(self):
disk = diagnostics.DiskDiagnostics(
read_bytes=1, read_requests=2,
write_bytes=3, write_requests=4,
errors_count=5)
self.assertEqual(1, disk.read_bytes)
self.assertEqual(2, disk.read_requests)
self.assertEqual(3, disk.write_bytes)
self.assertEqual(4, disk.write_requests)
self.assertEqual(5, disk.errors_count)
def test_memory_diagnostics(self):
memory = diagnostics.MemoryDiagnostics(maximum=1, used=2)
self.assertEqual(1, memory.maximum)
self.assertEqual(2, memory.used)
def test_diagnostics(self):
cpu_details = [diagnostics.CpuDiagnostics()]
nic_details = [diagnostics.NicDiagnostics()]
disk_details = [diagnostics.DiskDiagnostics()]
memory_details = diagnostics.MemoryDiagnostics(maximum=1, used=1)
diags = diagnostics.Diagnostics(
state='running', driver='libvirt', hypervisor_os='fake-os',
uptime=1, cpu_details=cpu_details, nic_details=nic_details,
disk_details=disk_details, config_drive=True,
memory_details=memory_details)
self.assertEqual('running', diags.state)
self.assertEqual('libvirt', diags.driver)
self.assertEqual('fake-os', diags.hypervisor_os)
self.assertEqual(1, diags.uptime)
self.assertTrue(diags.config_drive)
self.assertEqual(1, len(diags.cpu_details))
self.assertEqual(1, len(diags.nic_details))
self.assertEqual(1, len(diags.disk_details))
self.assertEqual(1, diags.num_cpus)
self.assertEqual(1, diags.num_disks)
self.assertEqual(1, diags.num_nics)
self.assertEqual(1, diags.memory_details.maximum)
self.assertEqual(1, diags.memory_details.used)
self.assertEqual('1.0', diags.VERSION)
def test_add_cpu(self):
diags = diagnostics.Diagnostics()
self.assertEqual([], diags.cpu_details)
diags.add_cpu(id=1, time=7, utilisation=15)
self.assertEqual(1, len(diags.cpu_details))
self.assertEqual(7, diags.cpu_details[0].time)
self.assertEqual(1, diags.cpu_details[0].id)
self.assertEqual(15, diags.cpu_details[0].utilisation)
self.assertEqual(1, diags.num_cpus)
def test_add_nic(self):
diags = diagnostics.Diagnostics()
self.assertEqual([], diags.nic_details)
diags.add_nic(mac_address='00:00:ca:fe:00:00',
rx_octets=1, rx_errors=2, rx_drop=3, rx_packets=4, rx_rate=5,
tx_octets=6, tx_errors=7, tx_drop=8, tx_packets=9, tx_rate=10)
self.assertEqual(1, len(diags.nic_details))
self.assertEqual('00:00:ca:fe:00:00', diags.nic_details[0].mac_address)
self.assertEqual(1, diags.nic_details[0].rx_octets)
self.assertEqual(2, diags.nic_details[0].rx_errors)
self.assertEqual(3, diags.nic_details[0].rx_drop)
self.assertEqual(4, diags.nic_details[0].rx_packets)
self.assertEqual(5, diags.nic_details[0].rx_rate)
self.assertEqual(6, diags.nic_details[0].tx_octets)
self.assertEqual(7, diags.nic_details[0].tx_errors)
self.assertEqual(8, diags.nic_details[0].tx_drop)
self.assertEqual(9, diags.nic_details[0].tx_packets)
self.assertEqual(10, diags.nic_details[0].tx_rate)
self.assertEqual(1, diags.num_nics)
def test_add_disk(self):
diags = diagnostics.Diagnostics()
self.assertEqual([], diags.disk_details)
diags.add_disk(read_bytes=1, read_requests=2,
write_bytes=3, write_requests=4,
errors_count=5)
self.assertEqual(1, len(diags.disk_details))
self.assertEqual(1, diags.disk_details[0].read_bytes)
self.assertEqual(2, diags.disk_details[0].read_requests)
self.assertEqual(3, diags.disk_details[0].write_bytes)
self.assertEqual(4, diags.disk_details[0].write_requests)
self.assertEqual(5, diags.disk_details[0].errors_count)
self.assertEqual(1, diags.num_disks)