def test_next_view(self):
view_est_mod = ViewEstablishmentModule(0, self.resolver, 2, 0)
pred_module = PredicatesAndAction(view_est_mod, 0, self.resolver, 2, 0)
# Should update to view 1 in next
pred_module.views = [{CURRENT: 0, NEXT: 0}]
pred_module.next_view()
self.assertEqual(pred_module.views, [{CURRENT: 0, NEXT: 1}])
# number of byzantine nodes = 2 so "next" should flip to view 0
pred_module.views = [{CURRENT: 1, NEXT: 1}]
pred_module.next_view()
self.assertEqual(pred_module.views, [{CURRENT: 1, NEXT: 0}])
def test_automaton_phase_0_actions(self):
view_est_mod = ViewEstablishmentModule(0, self.resolver, 2, 0)
pred_module = PredicatesAndAction(view_est_mod, 0, self.resolver, 2, 0)
# Case 0 should call adopt and next_view in view Establishment module
view_est_mod.next_phs = Mock()
pred_module.adopt = Mock()
pred_module.reset_v_change = Mock()
pred_module.view_pair_to_adopt = pred_module.RST_PAIR
self.assertEqual(
pred_module.automation(ViewEstablishmentEnums.ACTION, 0, 0),
ViewEstablishmentEnums.NO_RETURN_VALUE)
view_est_mod.next_phs.assert_any_call()
#pred_module.reset_v_change.assert_any_call()
pred_module.adopt.assert_called_once_with(pred_module.RST_PAIR)
# Case 1a should call next_view and next_phs in view Establishment module
pred_module.next_view = Mock()
view_est_mod.next_phs = Mock()
pred_module.vChange = [True for i in range(2)]
self.assertEqual(
pred_module.automation(ViewEstablishmentEnums.ACTION, 0, 1),
ViewEstablishmentEnums.NO_RETURN_VALUE)
pred_module.next_view.assert_any_call()
view_est_mod.next_phs.assert_any_call()
# Case 1b should call next_phs in View Establishment module
pred_module.vChange = [False for i in range(2)]
view_est_mod.next_phs = Mock()
pred_module.views[pred_module.id] = pred_module.RST_PAIR
self.assertEqual(
pred_module.automation(ViewEstablishmentEnums.ACTION, 0, 1),
ViewEstablishmentEnums.NO_RETURN_VALUE)
view_est_mod.next_phs.assert_any_call()
# Case 2 should return "No action" and call reset_v_change
pred_module.reset_v_change = Mock()
self.assertEqual(
pred_module.automation(ViewEstablishmentEnums.ACTION, 0, 2),
ViewEstablishmentEnums.NO_ACTION)
# pred_module.reset_v_change.assert_any_call()
# Case 3 should return "Reset"
pred_module.reset_all = MagicMock(
return_value=ViewEstablishmentEnums.RESET)
#pred_module.reset_v_change = Mock()
self.assertEqual(
pred_module.automation(ViewEstablishmentEnums.ACTION, 0, 3),
ViewEstablishmentEnums.RESET)