def setUp(self): self.mock_maas_state = MagicMock() with NamedTemporaryFile(mode='w+', encoding='utf-8') as tempf: utils.spew(tempf.name, yaml.dump(dict())) self.conf = Config(tempf.name, {}, save_backups=False) temp_bundle_f = NamedTemporaryFile(mode='r') self.conf.setopt('bundle_filename', temp_bundle_f.name) self.service_1 = create_service("nova-compute", { "num_units": 1, "charm": "cs:trusty/nova-compute-100"}, {}, []) self.service_2 = create_service("keystone", { "num_units": 1, "charm": "cs:trusty/keystone-100"}, {}, []) self.bundle_patcher = patch("bundleplacer.controller.Bundle") self.mock_bundle = self.bundle_patcher.start() self.mock_bundle_i = self.mock_bundle.return_value pm = PropertyMock(return_value=[self.service_1, self.service_2]) type(self.mock_bundle_i).services = pm self.pc = PlacementController(self.mock_maas_state, self.conf) self.mock_machine = MagicMock(name='machine1') pmid = PropertyMock(return_value='fake-instance-id-1') type(self.mock_machine).instance_id = pmid self.mock_machine_2 = MagicMock(name='machine2') pmid2 = PropertyMock(return_value='fake-instance-id-2') type(self.mock_machine_2).instance_id = pmid2 self.mock_machines = [self.mock_machine, self.mock_machine_2] self.mock_maas_state.machines.return_value = self.mock_machines
def setUp(self): self.service_1 = create_service("nova-compute", { "num_units": 1, "charm": "cs:trusty/nova-compute"}, {}, [])