def test_establish(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 current
pred_module.views = [{CURRENT: 0, NEXT: 1}]
self.resolver.on_view_established = Mock()
pred_module.establish()
self.assertEqual(pred_module.views, [{CURRENT: 1, NEXT: 1}])
def test_automaton_phase_1_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
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, 1, 0),
ViewEstablishmentEnums.NO_RETURN_VALUE)
pred_module.adopt.assert_called_once_with(pred_module.RST_PAIR)
pred_module.reset_v_change.assert_any_call()
# Case 1 should call next_view and next_phs in view Establishment module
view_est_mod.next_phs = Mock()
pred_module.reset_v_change = Mock()
pred_module.establish = Mock()
self.resolver.execute = Mock()
pred_module.views[pred_module.id] = pred_module.RST_PAIR
self.assertEqual(
pred_module.automation(ViewEstablishmentEnums.ACTION, 1, 1),
ViewEstablishmentEnums.NO_RETURN_VALUE)
view_est_mod.next_phs.assert_any_call()
pred_module.reset_v_change.assert_any_call()
pred_module.establish.assert_any_call()
# self.resolver.execute.assert_called_once_with(module=Module.REPLICATION_MODULE,
# func=Function.REPLICA_FLUSH)
# Case 2 should return "No action" and call reset_v_change
pred_module.reset_v_change = Mock()
self.assertEqual(
pred_module.automation(ViewEstablishmentEnums.ACTION, 1, 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)
self.assertEqual(
pred_module.automation(ViewEstablishmentEnums.ACTION, 1, 3),
ViewEstablishmentEnums.RESET)