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_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))
