def test_show_matching_filter(self, mock_machinewidget):
self.mock_maas_state.machines.return_value = [
self.mock_machine, self.mock_machine2, self.mock_machine3
]
# a little extra work to ensure that calls to
# MockWidget.__init__() return mocks with the intended machine
# attribute set:
mw1 = MagicMock(name="mw1")
mw1.machine = self.mock_machine
mw2 = MagicMock(name="mw2")
mw2.machine = self.mock_machine2
mw3 = MagicMock(name="mw3")
mw3.machine = self.mock_machine3
# the rest are the placeholders:
mw4 = MagicMock(name="mock_placeholder_widget1")
mw4.machine = self.pc.sub_placeholder
mw5 = MagicMock(name="mock_placeholder_widget2")
mw5.machine = self.pc.def_placeholder
mock_machinewidget.side_effect = [mw1, mw2, mw3, mw4, mw5]
ml = MachinesList(self.pc, self.actions)
self.assertEqual(5, len(ml.machine_widgets))
ml.filter_string = "machine1-filter_label"
ml.update()
print("ml.machinewidgets is {}".format(ml.machine_widgets))
self.assertEqual(1, len(ml.machine_widgets))
def test_widgets_config(self, mock_machinewidget):
for show_hardware in [False, True]:
for show_assignments in [False, True]:
MachinesList(self.pc,
self.actions,
show_hardware=show_hardware,
show_assignments=show_assignments)
mock_machinewidget.assert_any_call(self.mock_machine, self.pc,
self.actions, show_hardware,
show_assignments)
mock_machinewidget.reset_mock()
def build_widgets(self):
def has_services_p(m):
pc = self.placement_controller
n = sum(
[len(al) for at, al in pc.assignments_for_machine(m).items()])
return n > 0
clear_machine_func = self.placement_view.do_clear_machine
show_chooser_func = self.placement_view.do_show_service_chooser
self.open_maas_button = AttrMap(
Button("Open in Browser", on_press=self.browse_maas),
'button_secondary', 'button_secondary focus')
bc = self.placement_view.config.juju_env['bootstrap-config']
maasname = "'{}' <{}>".format(bc['name'], bc['maas-server'])
maastitle = "Connected to MAAS {} l:root p:{}".format(
maasname, self.placement_view.config.getopt('openstack_password'))
tw = Columns([
Text(maastitle),
Padding(self.open_maas_button,
align='right',
width=BUTTON_SIZE,
right=2)
])
self.machines_list = MachinesList(
self.placement_controller,
[(has_services_p, 'Clear All Services', clear_machine_func),
(has_services_p, 'Remove Some Services', show_chooser_func)],
show_hardware=True,
title_widgets=tw)
self.machines_list.update()
self.machines_list_pile = Pile([self.machines_list, Divider()])
# placeholders replaced in update() with absolute indexes, so
# if you change this list, check update().
pl = [
Text(('subheading', "Machines"), align='center'),
Divider(),
Pile([]), # machines_list
Divider()
]
self.main_pile = Pile(pl)
return self.main_pile
def build_widgets(self):
if self.charm_class.allow_multi_units:
machine_string = "machines"
plural_string = "s"
else:
machine_string = "a machine"
plural_string = ""
instructions = Text("Select {} to host {}".format(
machine_string, self.charm_class.display_name))
self.service_widget = ServiceWidget(self.charm_class,
self.controller,
show_constraints=True,
show_placements=True)
all_actions = [(AssignmentType.BareMetal, 'Add as Bare Metal',
self.do_select_baremetal),
(AssignmentType.LXC, 'Add as LXC', self.do_select_lxc),
(AssignmentType.KVM, 'Add as KVM', self.do_select_kvm)]
actions = [(label, func) for atype, label, func in all_actions
if atype in self.charm_class.allowed_assignment_types]
constraints = self.charm_class.constraints
# NOTE: show_assignments=False is a WORKAROUND for #194
self.machines_list = MachinesList(self.controller,
actions,
constraints=constraints,
show_hardware=True,
show_assignments=False)
self.machines_list.update()
close_button = AttrMap(Button('X', on_press=self.close_pressed),
'button_secondary', 'button_secondary focus')
p = Pile([
GridFlow([close_button], 5, 1, 0, 'right'), instructions,
Divider(), self.service_widget,
Divider(), self.machines_list
])
return LineBox(p, title="Select Machine{}".format(plural_string))
def test_hide_non_matching_constraints(self, mock_machinewidget):
ml = MachinesList(self.pc, self.actions, {'cpu_cores': 16384})
self.assertEqual(0, len(ml.machine_widgets))
def test_show_matching_constraints(self, mock_machinewidget):
ml = MachinesList(self.pc, self.actions, {'cpu_cores': 2})
self.assertEqual(1, len(ml.machine_widgets))
