/*
 * 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.labeled.unicast;

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

import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import java.util.ArrayList;
import java.util.List;
import org.junit.Test;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev180329.Srgb;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev180329.originator.srgb.tlv.SrgbValue;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev180329.originator.srgb.tlv.SrgbValueBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev180329.update.attributes.bgp.prefix.sid.bgp.prefix.sid.tlvs.bgp.prefix.sid.tlv.LuLabelIndexTlv;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev180329.update.attributes.bgp.prefix.sid.bgp.prefix.sid.tlvs.bgp.prefix.sid.tlv.LuLabelIndexTlvBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev180329.update.attributes.bgp.prefix.sid.bgp.prefix.sid.tlvs.bgp.prefix.sid.tlv.LuOriginatorSrgbTlv;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev180329.update.attributes.bgp.prefix.sid.bgp.prefix.sid.tlvs.bgp.prefix.sid.tlv.LuOriginatorSrgbTlvBuilder;
import org.opendaylight.yangtools.yang.common.Uint32;

public final class BgpPrefixSidTlvsTest {

    @Test
    public void testLabelIndexParser() {
        final LabelIndexTlvParser parser = new LabelIndexTlvParser();
        final LuLabelIndexTlv tlv = new LuLabelIndexTlvBuilder().setLabelIndexTlv(Uint32.valueOf(333)).build();
        final ByteBuf serialized = Unpooled.buffer(7);
        final byte[] expected = new byte[] {0, 0, 0, 0, 0, 1, (byte)0x4d};
        parser.serializeBgpPrefixSidTlv(tlv, serialized);
        assertArrayEquals(expected, serialized.array());

        assertEquals(tlv.getLabelIndexTlv(), parser.parseBgpPrefixSidTlv(serialized).getLabelIndexTlv());

        assertEquals(1, parser.getType());
    }

    @Test
    public void testOriginatorParser() {
        final OriginatorSrgbTlvParser parser = new OriginatorSrgbTlvParser();
        final List<SrgbValue> list = new ArrayList<>();

        final Srgb srgb1 = new Srgb(Uint32.ONE);
        final Srgb srgb2 = new Srgb(Uint32.valueOf(2));
        list.add(new SrgbValueBuilder().setBase(srgb1).setRange(srgb2).build());
        list.add(new SrgbValueBuilder().setBase(srgb2).setRange(srgb1).build());

        final LuOriginatorSrgbTlv tlv = new LuOriginatorSrgbTlvBuilder().setSrgbValue(list).build();
        final ByteBuf serialized = Unpooled.buffer(14);
        parser.serializeBgpPrefixSidTlv(tlv, serialized);
        final byte[] expected = new byte[] {0, 0, 0, 0, 1, 0, 0, 2, 0, 0, 2, 0, 0, 1};
        assertArrayEquals(expected, serialized.array());

        final LuOriginatorSrgbTlv parsed = parser.parseBgpPrefixSidTlv(serialized);
        assertEquals(tlv.getSrgbValue().size(), parsed.getSrgbValue().size());
        assertEquals(tlv.getSrgbValue().get(0).getBase(), srgb1);
        assertEquals(tlv.getSrgbValue().get(0).getRange(), srgb2);
        assertEquals(tlv.getSrgbValue().get(1).getBase(), srgb2);
        assertEquals(tlv.getSrgbValue().get(1).getRange(), srgb1);

        assertEquals(3, parser.getType());
    }
}