def test_get_vcpu_used(self):
self.mox.StubOutWithMock(ovz_utils, 'get_cpuunits_capability')
ovz_utils.get_cpuunits_capability().AndReturn(fakes.CPUUNITSCAPA)
self.mox.StubOutWithMock(ovz_utils, 'get_vcpu_total')
ovz_utils.get_vcpu_total().AndReturn(fakes.HOSTSTATS['vcpus'])
self.mox.ReplayAll()
used = int(fakes.HOSTSTATS['vcpus'] *
(float(fakes.CPUUNITSCAPA['subscribed']) /
fakes.CPUUNITSCAPA['total']))
result = ovz_utils.get_vcpu_used()
self.assertEqual(result, used)
def test_get_vcpu_total_failure(self):
self.mox.StubOutWithMock(
ovz_utils.multiprocessing, 'cpu_count')
ovz_utils.multiprocessing.cpu_count().AndRaise(NotImplementedError)
self.mox.ReplayAll()
result = ovz_utils.get_vcpu_total()
self.assertEqual(result, 0)
def test_get_vcpu_total_success(self):
self.mox.StubOutWithMock(
ovz_utils.multiprocessing, 'cpu_count')
ovz_utils.multiprocessing.cpu_count().AndReturn(
fakes.HOSTSTATS['vcpus'])
self.mox.ReplayAll()
result = ovz_utils.get_vcpu_total()
self.assertEqual(result, fakes.HOSTSTATS['vcpus'])
