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"}, {}, [])
