/*
 * Copyright (c) 2015 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.parser.impl.message.update;

import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import com.google.common.primitives.Bytes;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import java.util.ArrayList;
import java.util.List;
import org.junit.Before;
import org.junit.Test;
import org.opendaylight.protocol.bgp.parser.BGPDocumentedException;
import org.opendaylight.protocol.bgp.parser.BGPParsingException;
import org.opendaylight.protocol.bgp.parser.spi.extended.community.ExtendedCommunityRegistry;
import org.opendaylight.protocol.bgp.parser.spi.pojo.ServiceLoaderBGPExtensionProviderContext;
import org.opendaylight.protocol.util.ByteArray;
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.ExtendedCommunities;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.attributes.ExtendedCommunitiesBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.ShortAsNumber;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.extended.community.extended.community.RouteOriginExtendedCommunityCase;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.extended.community.extended.community.RouteOriginExtendedCommunityCaseBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.extended.community.extended.community.route.origin.extended.community._case.RouteOriginExtendedCommunityBuilder;
import org.opendaylight.yangtools.yang.common.Uint32;

public class ExtendedCommunitiesAttributeParserTest {

    private static final byte[] INPUT = {
        0x40, 0x03, 0x00, 54, 0, 0, 1, 76
    };

    private static final byte[] UNKOWN = {
        0x40, 0x05, 0x00, 54, 0, 0, 1, 76
    };

    private ExtendedCommunitiesAttributeParser handler;

    @Before
    public void setUp() {
        final ExtendedCommunityRegistry exReg = ServiceLoaderBGPExtensionProviderContext.getSingletonInstance()
                .getExtendedCommunityRegistry();
        this.handler = new ExtendedCommunitiesAttributeParser(exReg);
    }

    @Test
    public void testExtendedCommunityAttributeParser() throws BGPDocumentedException, BGPParsingException {
        final RouteOriginExtendedCommunityCase routeOrigin = new RouteOriginExtendedCommunityCaseBuilder()
                .setRouteOriginExtendedCommunity(new RouteOriginExtendedCommunityBuilder()
                    .setGlobalAdministrator(new ShortAsNumber(Uint32.valueOf(54)))
                    .setLocalAdministrator(new byte[] { 0, 0, 1, 76 }).build())
                .build();
        final ExtendedCommunities expected = new ExtendedCommunitiesBuilder().setTransitive(false)
                .setExtendedCommunity(routeOrigin).build();
        final AttributesBuilder attBuilder = new AttributesBuilder();

        this.handler.parseAttribute(Unpooled.copiedBuffer(INPUT), attBuilder, null);
        final ExtendedCommunities parsed = attBuilder.getExtendedCommunities().get(0);
        assertEquals(expected, parsed);

        final ByteBuf output = Unpooled.buffer(INPUT.length);
        this.handler.serializeAttribute(attBuilder.build(), output);
        assertArrayEquals(Bytes.concat(new byte[] {(byte)192, 16, 8}, INPUT), ByteArray.readAllBytes(output));
    }

    @Test
    public void testEmptyListExtendedCommunityAttributeParser() throws BGPDocumentedException, BGPParsingException {
        final List<ExtendedCommunities> extendedCommunitiesList = new ArrayList<>();
        final AttributesBuilder attBuilder = new AttributesBuilder().setExtendedCommunities(extendedCommunitiesList);
        final ByteBuf output = Unpooled.buffer();

        this.handler.serializeAttribute(attBuilder.build(), output);
        assertEquals(output, output);
    }

    @Test
    public void testEmptyExtendedCommunityAttributeParser() throws BGPDocumentedException, BGPParsingException {
        final ByteBuf output = Unpooled.buffer();
        this.handler.serializeAttribute(new AttributesBuilder().build(), output);
        assertEquals(Unpooled.buffer(), output);
    }

    @Test
    public void testExtendedCommunityAttributeParserUnknown() throws BGPDocumentedException, BGPParsingException {
        final AttributesBuilder attBuilder = new AttributesBuilder();
        this.handler.parseAttribute(Unpooled.copiedBuffer(UNKOWN), attBuilder, null);
        assertTrue(attBuilder.getExtendedCommunities().isEmpty());
    }
}