def build_widgets(self):
instructions = Text("Remove services from {}".format(
self.machine.hostname))
self.machine_widget = MachineWidget(self.machine,
self.controller,
show_hardware=True)
def show_remove_p(cc):
md = self.controller.get_assignments(cc)
for atype, ms in md.items():
hostnames = [m.hostname for m in ms]
if self.machine.hostname in hostnames:
return True
return False
actions = [(show_remove_p, 'Remove', self.do_remove)]
self.services_list = ServicesList(self.controller,
actions,
actions,
machine=self.machine)
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.machine_widget,
Divider(), self.services_list
])
return LineBox(p, title="Remove Services")
def test_dont_show_assignments(self):
"""Widget with show_assignments set to FALSE should NOT show
assignments"""
self.pc.assign(self.mock_machine, CharmNovaCompute, AssignmentType.LXC)
w = MachineWidget(self.mock_machine, self.pc, show_assignments=False)
self.assertFalse(search_in_widget("LXC.*Compute", w))
def test_hardware_not_shown(self):
"""show_hardware=False should NOT show hardware details"""
w = MachineWidget(self.mock_machine, self.pc, show_hardware=False)
self.assertFalse(search_in_widget("arch", w))
self.assertFalse(search_in_widget("cores", w))
self.assertFalse(search_in_widget("mem", w))
self.assertFalse(search_in_widget("storage", w))
def build_widgets(self):
instructions = Text("Remove services from {}".format(
self.machine.hostname))
self.machine_widget = MachineWidget(self.machine,
self.controller,
show_hardware=True)
def show_remove_p(cc):
md = self.controller.get_assignments(cc)
for atype, ms in md.items():
hostnames = [m.hostname for m in ms]
if self.machine.hostname in hostnames:
return True
return False
actions = [(show_remove_p, 'Remove', self.do_remove)]
self.services_list = ServicesList(self.controller,
actions, actions,
machine=self.machine)
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.machine_widget,
Divider(), self.services_list])
return LineBox(p, title="Remove Services")
def add_machine_widget(self, machine):
mw = MachineWidget(machine, self.controller, self.actions,
self.show_hardware, self.show_assignments)
self.machine_widgets.append(mw)
options = self.machine_pile.options()
self.machine_pile.contents.append((mw, options))
self.machine_pile.contents.append(
(AttrMap(Padding(Divider('\u23bc'), left=2, right=2),
'label'), options))
return mw
def test_actions_use_pred(self):
"""Action predicates control whether a button appears (disabled)"""
# NOTE: this test assumes that disabled buttons are just the
# button label with parentheses.
fake_action_func = MagicMock()
fake_pred = MagicMock()
fake_pred.return_value = False
actions = [(fake_pred, "fake-action", fake_action_func)]
w = MachineWidget(self.mock_machine, self.pc, actions=actions)
self.assertTrue(search_in_widget("\(.*fake-action.*\)", w))
fake_pred.assert_called_with(self.mock_machine)
fake_pred.return_value = True
fake_pred.reset_mock()
w.update()
self.assertTrue(search_in_widget("<.*fake-action.*>", w))
fake_pred.assert_called_with(self.mock_machine)
def test_actions_use_pred(self):
"""Action predicates control whether a button appears (disabled)"""
# NOTE: this test assumes that disabled buttons are just the
# button label with parentheses.
fake_action_func = MagicMock()
fake_pred = MagicMock()
fake_pred.return_value = False
actions = [(fake_pred, "fake-action", fake_action_func)]
w = MachineWidget(self.mock_machine, self.pc, actions=actions)
self.assertTrue(search_in_widget("\(.*fake-action.*\)", w))
fake_pred.assert_called_with(self.mock_machine)
fake_pred.return_value = True
fake_pred.reset_mock()
w.update()
self.assertTrue(search_in_widget("<.*fake-action.*>", w))
fake_pred.assert_called_with(self.mock_machine)
def test_show_actions(self):
"""Actions passed as 2-tuples should always be shown as buttons"""
fake_action_func = MagicMock()
actions = [("fake-action", fake_action_func)]
w = MachineWidget(self.mock_machine, self.pc, actions=actions)
self.assertTrue(search_in_widget("fake-action", w))
def test_show_assignments(self):
"""Widget with show_assignments set should show assignments"""
self.pc.assign(self.mock_machine, CharmNovaCompute, AssignmentType.LXC)
w = MachineWidget(self.mock_machine, self.pc, show_assignments=True)
self.assertTrue(search_in_widget("LXC.*Compute", w))
class ServiceChooser(WidgetWrap):
"""Presents list of services to put on a machine.
Supports multiple selection, implying separate containers using
--to.
"""
def __init__(self, controller, machine, parent_widget):
"""
controller is a PlacementController
machine is the machine to show Services for
parent_widget should support 'remove_overlay()'
"""
self.controller = controller
self.machine = machine
self.parent_widget = parent_widget
w = self.build_widgets()
super().__init__(w)
def build_widgets(self):
instructions = Text("Remove services from {}".format(
self.machine.hostname))
self.machine_widget = MachineWidget(self.machine,
self.controller,
show_hardware=True)
def show_remove_p(cc):
md = self.controller.get_assignments(cc)
for atype, ms in md.items():
hostnames = [m.hostname for m in ms]
if self.machine.hostname in hostnames:
return True
return False
actions = [(show_remove_p, 'Remove', self.do_remove)]
self.services_list = ServicesList(self.controller,
actions, actions,
machine=self.machine)
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.machine_widget,
Divider(), self.services_list])
return LineBox(p, title="Remove Services")
def update(self):
self.machine_widget.update()
self.services_list.update()
def do_add(self, sender, charm_class, atype):
self.controller.assign(self.machine, charm_class, atype)
self.update()
def do_remove(self, sender, charm_class):
self.controller.remove_one_assignment(self.machine,
charm_class)
self.update()
def close_pressed(self, sender):
self.parent_widget.remove_overlay(self)
class ServiceChooser(WidgetWrap):
"""Presents list of services to put on a machine.
Supports multiple selection, implying separate containers using
--to.
"""
def __init__(self, controller, machine, parent_widget):
"""
controller is a PlacementController
machine is the machine to show Services for
parent_widget should support 'remove_overlay()'
"""
self.controller = controller
self.machine = machine
self.parent_widget = parent_widget
w = self.build_widgets()
super().__init__(w)
def build_widgets(self):
instructions = Text("Remove services from {}".format(
self.machine.hostname))
self.machine_widget = MachineWidget(self.machine,
self.controller,
show_hardware=True)
def show_remove_p(cc):
md = self.controller.get_assignments(cc)
for atype, ms in md.items():
hostnames = [m.hostname for m in ms]
if self.machine.hostname in hostnames:
return True
return False
actions = [(show_remove_p, 'Remove', self.do_remove)]
self.services_list = ServicesList(self.controller,
actions,
actions,
machine=self.machine)
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.machine_widget,
Divider(), self.services_list
])
return LineBox(p, title="Remove Services")
def update(self):
self.machine_widget.update()
self.services_list.update()
def do_add(self, sender, charm_class, atype):
self.controller.assign(self.machine, charm_class, atype)
self.update()
def do_remove(self, sender, charm_class):
self.controller.remove_one_assignment(self.machine, charm_class)
self.update()
def close_pressed(self, sender):
self.parent_widget.remove_overlay(self)
