def test_check_hypervisor_stats(self): mock_compute = mock.Mock() mock_stats = mock.Mock() return_values = [ { 'count': 0, 'memory_mb': 0, 'vcpus': 0 }, { 'count': 1, 'memory_mb': 1, 'vcpus': 1 }, ] mock_stats.to_dict.side_effect = return_values mock_compute.hypervisors.statistics.return_value = mock_stats stats = utils.check_hypervisor_stats(mock_compute, nodes=1, memory=1, vcpu=1) self.assertEqual(stats, None) self.assertEqual(mock_stats.to_dict.call_count, 1) stats = utils.check_hypervisor_stats(mock_compute, nodes=1, memory=1, vcpu=1) self.assertEqual(stats, return_values[-1]) self.assertEqual(mock_stats.to_dict.call_count, 2)
def _pre_heat_deploy(self): """Setup before the Heat stack create or update has been done.""" clients = self.app.client_manager compute_client = clients.compute self.log.debug("Checking hypervisor stats") if utils.check_hypervisor_stats(compute_client) is None: print("Expected hypervisor stats not met", file=sys.stderr) return
def test_check_hypervisor_stats(self): mock_compute = mock.Mock() mock_stats = mock.Mock() return_values = [ {'count': 0, 'memory_mb': 0, 'vcpus': 0}, {'count': 1, 'memory_mb': 1, 'vcpus': 1}, ] mock_stats.to_dict.side_effect = return_values mock_compute.hypervisors.statistics.return_value = mock_stats stats = utils.check_hypervisor_stats( mock_compute, nodes=1, memory=1, vcpu=1) self.assertEqual(stats, None) self.assertEqual(mock_stats.to_dict.call_count, 1) stats = utils.check_hypervisor_stats( mock_compute, nodes=1, memory=1, vcpu=1) self.assertEqual(stats, return_values[-1]) self.assertEqual(mock_stats.to_dict.call_count, 2)