def get_vrrp_subflow(self, prefix): sf_name = prefix + '-' + constants.GET_VRRP_SUBFLOW vrrp_subflow = linear_flow.Flow(sf_name) vrrp_subflow.add( network_tasks.GetAmphoraeNetworkConfigs( name=sf_name + '-' + constants.GET_AMP_NETWORK_CONFIG, requires=constants.LOADBALANCER, provides=constants.AMPHORAE_NETWORK_CONFIG)) vrrp_subflow.add( amphora_driver_tasks.AmphoraUpdateVRRPInterface( name=sf_name + '-' + constants.AMP_UPDATE_VRRP_INTF, requires=constants.LOADBALANCER, provides=constants.LOADBALANCER)) vrrp_subflow.add( database_tasks.CreateVRRPGroupForLB( name=sf_name + '-' + constants.CREATE_VRRP_GROUP_FOR_LB, requires=constants.LOADBALANCER, provides=constants.LOADBALANCER)) vrrp_subflow.add( amphora_driver_tasks.AmphoraVRRPUpdate( name=sf_name + '-' + constants.AMP_VRRP_UPDATE, requires=(constants.LOADBALANCER, constants.AMPHORAE_NETWORK_CONFIG))) vrrp_subflow.add( amphora_driver_tasks.AmphoraVRRPStart( name=sf_name + '-' + constants.AMP_VRRP_START, requires=constants.LOADBALANCER)) return vrrp_subflow
def test_amphora_vrrp_update(self, mock_driver, mock_generate_uuid, mock_log, mock_get_session, mock_listener_repo_get, mock_listener_repo_update, mock_amphora_repo_update): amphora_vrrp_update_obj = (amphora_driver_tasks.AmphoraVRRPUpdate()) amphora_vrrp_update_obj.execute(_LB_mock) mock_driver.update_vrrp_conf.assert_called_once_with(_LB_mock)