org.opendaylight.ocpjava.protocol.impl.serialization.factories.ModifyStateInputFactoryTest.java Source code

Introduction

Here is the source code for org.opendaylight.ocpjava.protocol.impl.serialization.factories.ModifyStateInputFactoryTest.java

Source

/*
 * Copyright (c) 2016 Foxconn Corporation 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.ocpjava.protocol.impl.serialization.factories;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.UnpooledByteBufAllocator;
import java.lang.reflect.Method;
import java.util.List;
import java.util.ArrayList;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import org.opendaylight.ocpjava.protocol.api.extensibility.OCPSerializer;
import org.opendaylight.ocpjava.protocol.api.extensibility.SerializerRegistry;
import org.opendaylight.ocpjava.protocol.api.keys.MessageTypeKey;
import org.opendaylight.ocpjava.protocol.impl.serialization.SerializerRegistryImpl;
import org.opendaylight.ocpjava.protocol.impl.util.BufferHelper;
import org.opendaylight.ocpjava.protocol.api.util.EncodeConstants;
import org.opendaylight.yang.gen.v1.urn.opendaylight.ocp.protocol.rev150811.ModifyStateInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.ocp.protocol.rev150811.ModifyStateInputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.ocp.common.types.rev150811.ObjId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.ocp.common.types.rev150811.OcpMsgType;
import org.opendaylight.yang.gen.v1.urn.opendaylight.ocp.common.types.rev150811.StateType;
import org.opendaylight.yang.gen.v1.urn.opendaylight.ocp.common.types.rev150811.StateVal;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author Marko Lai <marko.ch.lai@foxconn.com>
 *
 */
public class ModifyStateInputFactoryTest {
    private static final Logger LOG = LoggerFactory.getLogger(ModifyStateInputFactoryTest.class);

    private SerializerRegistry registry;
    private OCPSerializer<ModifyStateInput> modifyStateInputFactory;

    /**
     * Initializes serializer registry and stores correct factory in field
     */
    @Before
    public void startUp() {
        registry = new SerializerRegistryImpl();
        registry.init();
        modifyStateInputFactory = registry
                .getSerializer(new MessageTypeKey<>(EncodeConstants.OCP_VERSION_ID, ModifyStateInput.class));
    }

    /**
     * Testing of {@link HealthCheckInputMessageFactory} for correct translation from POJO
     * @throws Exception
     */
    @Test
    public void testElementsSet() throws Exception {
        ModifyStateInputBuilder hib = new ModifyStateInputBuilder();
        BufferHelper.setupHeader(hib, OcpMsgType.valueOf("MODIFYSTATEREQ"));
        boolean testEvent = true;
        String testObjId = "ALL";
        String testType = "AST";
        String testVal = "UNLOCKED";

        //set xid
        Method m2_t = hib.getClass().getMethod("setXid", Long.class);
        m2_t.invoke(hib, new Long(0));

        //set ObjId
        Method m3_t = hib.getClass().getMethod("setObjId", ObjId.class);
        m3_t.invoke(hib, new ObjId(testObjId));

        //set stateType
        Method m4_t = hib.getClass().getMethod("setStateType", StateType.class);
        m4_t.invoke(hib, StateType.valueOf(testType));

        //set stateValue
        Method m5_t = hib.getClass().getMethod("setStateValue", StateVal.class);
        m5_t.invoke(hib, StateVal.valueOf(testVal));

        ModifyStateInput hi = hib.build();
        LOG.debug("ModifyStateInputFactoryTest - hi objId value = {}", hi.getObjId());

        ByteBuf out = UnpooledByteBufAllocator.DEFAULT.buffer();
        modifyStateInputFactory.serialize(hi, out);

        //Verify and check the bytebuf info
        LOG.debug("ModifyStateInputFactoryTest - out = {}", out.readableBytes());
        byte[] bytes = new byte[out.readableBytes()];
        int readerIndex = out.readerIndex();
        out.getBytes(readerIndex, bytes);

        String buf = new String(bytes, "UTF-8");
        StringBuilder seq = new StringBuilder("");
        seq.append("<obj objID=\"");
        seq.append(testObjId);
        seq.append("\">");
        boolean checkVal = buf.contains(seq);

        //Check and compare elements
        Assert.assertEquals("Wrong length", true, checkVal);
    }
}