/*
 * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */
package org.opendaylight.protocol.bgp.rib.impl;

import static org.junit.Assert.assertEquals;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.doReturn;
import static org.opendaylight.protocol.util.CheckUtil.readDataOperational;
import static org.opendaylight.protocol.util.CheckUtil.waitFutureSuccess;

import com.google.common.collect.Lists;
import io.netty.util.concurrent.Future;
import java.net.InetSocketAddress;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import org.junit.After;
import org.junit.Before;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.opendaylight.mdsal.singleton.api.ClusterSingletonService;
import org.opendaylight.mdsal.singleton.api.ClusterSingletonServiceProvider;
import org.opendaylight.protocol.bgp.inet.RIBActivator;
import org.opendaylight.protocol.bgp.openconfig.spi.BGPTableTypeRegistryConsumer;
import org.opendaylight.protocol.bgp.parser.BGPError;
import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
import org.opendaylight.protocol.bgp.parser.impl.BGPActivator;
import org.opendaylight.protocol.bgp.parser.spi.BGPExtensionProviderContext;
import org.opendaylight.protocol.bgp.parser.spi.pojo.SimpleBGPExtensionProviderContext;
import org.opendaylight.protocol.bgp.rib.impl.config.BgpPeer;
import org.opendaylight.protocol.bgp.rib.impl.spi.BGPPeerRegistry;
import org.opendaylight.protocol.bgp.rib.impl.spi.BGPSessionPreferences;
import org.opendaylight.protocol.bgp.rib.spi.RIBExtensionProviderContext;
import org.opendaylight.protocol.bgp.rib.spi.SimpleRIBExtensionProviderContext;
import org.opendaylight.protocol.util.InetSocketAddressUtil;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddressNoZone;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4AddressNoZone;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.bgp.rib.rib.loc.rib.tables.routes.Ipv4RoutesCase;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.prefixes.DestinationIpv4Builder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.prefixes.destination.ipv4.Ipv4PrefixesBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.routes.Ipv4Routes;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.routes.ipv4.routes.Ipv4Route;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.routes.ipv4.routes.Ipv4RouteKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationIpv4CaseBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.Notify;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.NotifyBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.Open;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.OpenBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.PathId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.Update;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.UpdateBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.open.message.BgpParameters;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.AttributesBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.attributes.AsPathBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.attributes.ClusterIdBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.attributes.LocalPrefBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.attributes.MultiExitDiscBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.attributes.OriginBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.attributes.OriginatorIdBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.attributes.as.path.SegmentsBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.update.message.WithdrawnRoutesBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.AttributesReachBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.BgpTableType;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.attributes.reach.MpReachNlriBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.attributes.reach.mp.reach.nlri.AdvertizedRoutesBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.BgpRib;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.PeerRole;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.BgpId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.BgpOrigin;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.ClusterIdentifier;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.Ipv4AddressFamily;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.Ipv6AddressFamily;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.UnicastSubsequentAddressFamily;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.next.hop.c.next.hop.Ipv4NextHopCaseBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.next.hop.c.next.hop.ipv4.next.hop._case.Ipv4NextHopBuilder;
import org.opendaylight.yangtools.concepts.Registration;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.opendaylight.yangtools.yang.common.Uint16;
import org.opendaylight.yangtools.yang.common.Uint32;

public abstract class AbstractAddPathTest extends DefaultRibPoliciesMockTest {
    private static final int RETRY_TIMER = 10;
    static final String RIB_ID = "127.0.0.1";
    static final BgpId BGP_ID = new BgpId(RIB_ID);
    static final Ipv4AddressNoZone PEER1 = new Ipv4AddressNoZone("127.0.0.2");
    static final Ipv4AddressNoZone PEER2 = new Ipv4AddressNoZone("127.0.0.3");
    static final Ipv4AddressNoZone PEER3 = new Ipv4AddressNoZone("127.0.0.4");
    static final Ipv4AddressNoZone PEER4 = new Ipv4AddressNoZone("127.0.0.5");
    static final Ipv4AddressNoZone PEER5 = new Ipv4AddressNoZone("127.0.0.6");
    static final Ipv4AddressNoZone PEER6 = new Ipv4AddressNoZone("127.0.0.7");
    static final AsNumber AS_NUMBER = new AsNumber(Uint32.valueOf(AS));
    static final Uint16 PORT = Uint16.valueOf(InetSocketAddressUtil.getRandomPort());
    static final Ipv4Prefix PREFIX1 = new Ipv4Prefix("1.1.1.1/32");
    private static final ClusterIdentifier CLUSTER_ID = new ClusterIdentifier(RIB_ID);
    static final int HOLDTIMER = 2180;
    private static final Ipv4AddressNoZone NH1 = new Ipv4AddressNoZone("2.2.2.2");
    static final Update UPD_100 = createSimpleUpdate(PREFIX1, new PathId(Uint32.ONE), CLUSTER_ID, 100);
    static final Update UPD_50 = createSimpleUpdate(PREFIX1, new PathId(Uint32.TWO), CLUSTER_ID, 50);
    static final Update UPD_200 = createSimpleUpdate(PREFIX1, new PathId(Uint32.valueOf(3)), CLUSTER_ID, 200);
    static final Update UPD_20 = createSimpleUpdate(PREFIX1, new PathId(Uint32.ONE), CLUSTER_ID, 20);
    static final Update UPD_NA_100 = createSimpleUpdate(PREFIX1, null, CLUSTER_ID, 100);
    static final Update UPD_NA_100_EBGP = createSimpleUpdateEbgp(PREFIX1);
    static final Update UPD_NA_200 = createSimpleUpdate(PREFIX1, null, CLUSTER_ID, 200);
    static final Update UPD_NA_200_EBGP = createSimpleUpdateEbgp(PREFIX1);
    static final TablesKey TABLES_KEY = new TablesKey(Ipv4AddressFamily.VALUE, UnicastSubsequentAddressFamily.VALUE);
    static final List<BgpTableType> TABLES_TYPE = List.of(new BgpTableTypeImpl(TABLES_KEY.getAfi(),
        TABLES_KEY.getSafi()));
    static final Set<TablesKey> AFI_SAFIS_ADVERTIZED = Set.of(TABLES_KEY);

    static final InstanceIdentifier<BgpRib> BGP_IID = InstanceIdentifier.create(BgpRib.class);
    static final int GRACEFUL_RESTART_TIME = 5;
    @Mock
    protected ClusterSingletonServiceProvider clusterSingletonServiceProvider;
    BGPDispatcherImpl serverDispatcher;
    final RIBExtensionProviderContext ribExtension = new SimpleRIBExtensionProviderContext();
    private final BGPExtensionProviderContext context = new SimpleBGPExtensionProviderContext();
    private final RIBActivator ribActivator = new RIBActivator();
    private BGPActivator bgpActivator;
    private BGPNettyGroups groups;
    private org.opendaylight.protocol.bgp.inet.BGPActivator inetActivator;
    protected StrictBGPPeerRegistry serverRegistry;
    protected ConstantCodecsRegistry codecsRegistry;

    private List<BGPDispatcherImpl> clientDispatchers;

    @Override
    @Before
    public void setUp() throws Exception {
        super.setUp();

        ribActivator.startRIBExtensionProvider(ribExtension, mappingService.currentSerializer());

        bgpActivator = new BGPActivator();
        inetActivator = new org.opendaylight.protocol.bgp.inet.BGPActivator();
        bgpActivator.start(context);
        inetActivator.start(context);
        groups = new BGPNettyGroups();
        serverRegistry = new StrictBGPPeerRegistry();
        serverDispatcher = new BGPDispatcherImpl(context, groups, serverRegistry);
        doReturn(Mockito.mock(Registration.class)).when(clusterSingletonServiceProvider)
            .registerClusterSingletonService(any(ClusterSingletonService.class));

        codecsRegistry = new ConstantCodecsRegistry(mappingService.currentSerializer());
        clientDispatchers = new ArrayList<>();
    }

    @Override
    @After
    public void tearDown() throws Exception {
        groups.close();
        super.tearDown();
    }

    void sendRouteAndCheckIsOnLocRib(final BGPSessionImpl session, final Ipv4Prefix prefix, final long localPreference,
        final int expectedRoutesOnDS) throws Exception {
        waitFutureSuccess(session.writeAndFlush(createSimpleUpdate(prefix, null, null, localPreference)));
        checkLocRib(expectedRoutesOnDS);
    }

    void sendWithdrawalRouteAndCheckIsOnLocRib(final BGPSessionImpl session, final Ipv4Prefix prefix,
        final long localPreference, final int expectedRoutesOnDS) throws Exception {
        waitFutureSuccess(session.writeAndFlush(createSimpleWithdrawalUpdate(prefix, localPreference)));
        checkLocRib(expectedRoutesOnDS);
    }

    void sendNotification(final BGPSessionImpl session) {
        final Notify notMsg = new NotifyBuilder().setErrorCode(BGPError.OPT_PARAM_NOT_SUPPORTED.getCode())
            .setErrorSubcode(BGPError.OPT_PARAM_NOT_SUPPORTED.getSubcode()).setData(new byte[] { 4, 9 }).build();
        waitFutureSuccess(session.writeAndFlush(notMsg));
    }

    void causeBGPError(final BGPSessionImpl session) {
        final Open openObj = new OpenBuilder().setBgpIdentifier(new Ipv4AddressNoZone("1.1.1.1"))
            .setHoldTimer(Uint16.valueOf(50)).setMyAsNumber(Uint16.valueOf(72)).build();
        waitFutureSuccess(session.writeAndFlush(openObj));
    }

    private void checkLocRib(final int expectedRoutesOnDS) throws Exception {
        // FIXME: remove this sleep
        Thread.sleep(100);
        readDataOperational(getDataBroker(), BGP_IID, bgpRib -> {
            final Ipv4RoutesCase routes = (Ipv4RoutesCase) bgpRib.getRib().values().iterator().next().getLocRib()
                    .nonnullTables().values().iterator().next().getRoutes();
            final int size;
            if (routes != null) {
                final Ipv4Routes routesCase = routes.getIpv4Routes();
                if (routesCase != null) {
                    final Map<Ipv4RouteKey, Ipv4Route> routeList = routesCase.getIpv4Route();
                    size = routeList == null ? 0 : routeList.size();
                } else {
                    size = 0;
                }
            } else {
                size = 0;
            }

            assertEquals(expectedRoutesOnDS, size);
            return bgpRib;
        });
    }

    void checkPeersPresentOnDataStore(final int numberOfPeers) throws Exception {
        readDataOperational(getDataBroker(), BGP_IID, bgpRib -> {
            assertEquals(numberOfPeers, bgpRib.getRib().values().iterator().next().nonnullPeer().size());
            return bgpRib;
        });
    }

    BGPSessionImpl createPeerSession(final Ipv4AddressNoZone peer, final BgpParameters bgpParameters,
        final SimpleSessionListener sessionListener) throws InterruptedException {
        return createPeerSession(peer, bgpParameters, sessionListener, AS_NUMBER);
    }

    BGPSessionImpl createPeerSession(final Ipv4AddressNoZone peer, final BgpParameters bgpParameters,
                                     final SimpleSessionListener sessionListener,
                                     final AsNumber remoteAsNumber) throws InterruptedException {
        final StrictBGPPeerRegistry clientRegistry = new StrictBGPPeerRegistry();
        final BGPDispatcherImpl clientDispatcher = new BGPDispatcherImpl(context, groups, clientRegistry);

        clientDispatchers.add(clientDispatcher);
        clientRegistry.addPeer(new IpAddressNoZone(new Ipv4AddressNoZone(RIB_ID)), sessionListener,
                new BGPSessionPreferences(remoteAsNumber, HOLDTIMER, new BgpId(peer),
                        AS_NUMBER, Lists.newArrayList(bgpParameters)));

        return connectPeer(peer, clientDispatcher);
    }

    static BGPPeer configurePeer(final BGPTableTypeRegistryConsumer tableRegistry, final Ipv4AddressNoZone peerAddress,
            final RIBImpl ribImpl, final BgpParameters bgpParameters, final PeerRole peerRole,
            final BGPPeerRegistry bgpPeerRegistry) {
        return configurePeer(tableRegistry, peerAddress, ribImpl, bgpParameters, peerRole, bgpPeerRegistry,
                AFI_SAFIS_ADVERTIZED, Set.of());
    }

    static BGPPeer configurePeer(final BGPTableTypeRegistryConsumer tableRegistry,
            final Ipv4AddressNoZone peerAddress, final RIBImpl ribImpl, final BgpParameters bgpParameters,
            final PeerRole peerRole, final BGPPeerRegistry bgpPeerRegistry, final Set<TablesKey> afiSafiAdvertised,
            final Set<TablesKey> gracefulAfiSafiAdvertised) {
        final BgpPeer bgpPeer = Mockito.mock(BgpPeer.class);
        doReturn(Optional.empty()).when(bgpPeer).getErrorHandling();
        return configurePeer(tableRegistry, peerAddress, ribImpl, bgpParameters, peerRole, bgpPeerRegistry,
                afiSafiAdvertised, gracefulAfiSafiAdvertised, Map.of(), bgpPeer);
    }

    static BGPPeer configurePeer(final BGPTableTypeRegistryConsumer tableRegistry, final Ipv4AddressNoZone peerAddress,
            final RIBImpl ribImpl, final BgpParameters bgpParameters, final PeerRole peerRole,
            final BGPPeerRegistry bgpPeerRegistry, final Set<TablesKey> afiSafiAdvertised,
            final Set<TablesKey> gracefulAfiSafiAdvertised, final Map<TablesKey, Integer> llGracefulTimersAdvertised,
            final BgpPeer peer) {
        final IpAddressNoZone ipAddress = new IpAddressNoZone(peerAddress);

        final BGPPeer bgpPeer = new BGPPeer(tableRegistry, new IpAddressNoZone(peerAddress), null, ribImpl, peerRole,
                null, null, null, afiSafiAdvertised, gracefulAfiSafiAdvertised, llGracefulTimersAdvertised, peer);
        final List<BgpParameters> tlvs = Lists.newArrayList(bgpParameters);
        bgpPeerRegistry.addPeer(ipAddress, bgpPeer,
                new BGPSessionPreferences(AS_NUMBER, HOLDTIMER, new BgpId(RIB_ID), AS_NUMBER, tlvs));
        bgpPeer.instantiateServiceInstance();
        return bgpPeer;
    }

    private static BGPSessionImpl connectPeer(final Ipv4Address localAddress, final BGPDispatcherImpl dispatcherImpl)
            throws InterruptedException {
        final Future<BGPSessionImpl> future = dispatcherImpl
                .createClient(new InetSocketAddress(localAddress.getValue(), PORT.toJava()),
                        new InetSocketAddress(RIB_ID, PORT.toJava()), RETRY_TIMER, true);
        Thread.sleep(200);
        waitFutureSuccess(future);
        Thread.sleep(100);
        return future.getNow();
    }

    static BgpParameters createParameter(final boolean addPath) {
        return createParameter(addPath, false, null);
    }

    static BgpParameters createParameter(final boolean addPath,
                                         final boolean addIpv6,
                                         final Map<TablesKey, Boolean> gracefulTables) {
        final TablesKey ipv4Key = new TablesKey(Ipv4AddressFamily.VALUE, UnicastSubsequentAddressFamily.VALUE);
        final List<TablesKey> advertisedTables = Lists.newArrayList(ipv4Key);
        if (addIpv6) {
            final TablesKey ipv6Key = new TablesKey(Ipv6AddressFamily.VALUE, UnicastSubsequentAddressFamily.VALUE);
            advertisedTables.add(ipv6Key);
        }
        final List<TablesKey> addPathTables = new ArrayList<>();
        if (addPath) {
            addPathTables.add(ipv4Key);
        }
        return PeerUtil.createBgpParameters(advertisedTables, addPathTables, gracefulTables, GRACEFUL_RESTART_TIME);
    }

    private static Update createSimpleUpdate(final Ipv4Prefix prefix, final PathId pathId,
            final ClusterIdentifier clusterId, final long localPreference) {
        final AttributesBuilder attBuilder = new AttributesBuilder();
        attBuilder.setLocalPref(new LocalPrefBuilder().setPref(Uint32.valueOf(localPreference)).build());
        attBuilder.setOrigin(new OriginBuilder().setValue(BgpOrigin.Igp).build());
        attBuilder.setAsPath(new AsPathBuilder().setSegments(List.of()).build());
        attBuilder.setMultiExitDisc(new MultiExitDiscBuilder().setMed(Uint32.ZERO).build());
        if (clusterId != null) {
            attBuilder.setClusterId(new ClusterIdBuilder().setCluster(List.of(clusterId)).build());
            attBuilder.setOriginatorId(new OriginatorIdBuilder()
                .setOriginator(new Ipv4AddressNoZone(clusterId))
                .build());
        }
        addAttributeAugmentation(attBuilder, prefix, pathId);
        return new UpdateBuilder().setAttributes(attBuilder.build()).build();
    }

    private static Update createSimpleUpdateEbgp(final Ipv4Prefix prefix) {
        final AttributesBuilder attBuilder = new AttributesBuilder()
            .setOrigin(new OriginBuilder().setValue(BgpOrigin.Igp).build())
            .setAsPath(new AsPathBuilder()
                .setSegments(List.of(new SegmentsBuilder().setAsSequence(List.of(AS_NUMBER)).build()))
                .build());
        addAttributeAugmentation(attBuilder, prefix, null);

        return new UpdateBuilder().setAttributes(attBuilder.build()).build();
    }

    private static void addAttributeAugmentation(final AttributesBuilder attBuilder, final Ipv4Prefix prefix,
            final PathId pathId) {
        attBuilder.setUnrecognizedAttributes(Map.of());
        attBuilder.addAugmentation(new AttributesReachBuilder()
            .setMpReachNlri(new MpReachNlriBuilder()
                .setCNextHop(new Ipv4NextHopCaseBuilder()
                    .setIpv4NextHop(new Ipv4NextHopBuilder().setGlobal(NH1).build())
                    .build())
                .setAfi(Ipv4AddressFamily.VALUE)
                .setSafi(UnicastSubsequentAddressFamily.VALUE)
                .setAdvertizedRoutes(new AdvertizedRoutesBuilder()
                    .setDestinationType(new DestinationIpv4CaseBuilder()
                        .setDestinationIpv4(new DestinationIpv4Builder()
                            .setIpv4Prefixes(List.of(new Ipv4PrefixesBuilder()
                                .setPathId(pathId)
                                .setPrefix(new Ipv4Prefix(prefix))
                                .build()))
                            .build())
                        .build())
                    .build())
                .build())
            .build());
    }

    private static Update createSimpleWithdrawalUpdate(final Ipv4Prefix prefix, final long localPreference) {
        // FIXME: seems to be unused
        final AttributesBuilder attBuilder = new AttributesBuilder()
            .setLocalPref(new LocalPrefBuilder().setPref(Uint32.valueOf(localPreference)).build())
            .setOrigin(new OriginBuilder().setValue(BgpOrigin.Igp).build())
            .setAsPath(new AsPathBuilder().setSegments(List.of()).build());
        attBuilder.setUnrecognizedAttributes(Map.of());

        return new UpdateBuilder()
                .setWithdrawnRoutes(List.of(new WithdrawnRoutesBuilder().setPrefix(prefix).build()))
                .build();
    }
}