io.datty.spring.converter.embedded.IntMapEntityTest.java Source code

Introduction

Here is the source code for io.datty.spring.converter.embedded.IntMapEntityTest.java

Source

/*
 * Copyright (C) 2017 Datty.io Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 */
package io.datty.spring.converter.embedded;

import java.util.Collections;
import java.util.Map;

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

import io.datty.api.DattyRecord;
import io.datty.msgpack.MessageIO;
import io.datty.msgpack.core.reader.LongReader;
import io.datty.spring.support.DattyConverterUtil;
import io.netty.buffer.ByteBuf;

/**
 * IntMapEntityTest
 * 
 * @author Alex Shvid
 *
 */

public class IntMapEntityTest {

    @Test
    public void testNull() {

        IntMapEntity entity = new IntMapEntity();
        entity.setId(123L);

        DattyRecord rec = new DattyRecord();

        DattyConverterUtil.write(entity, rec);

        ByteBuf bb = rec.get("id").asByteBuf();
        Assert.assertNotNull(bb);

        Long id = LongReader.INSTANCE.read(bb.duplicate(), true);
        Assert.assertNotNull(id);
        Assert.assertEquals(123L, id.longValue());

        Assert.assertNull(rec.get("embedded"));

        IntMapEntity actual = DattyConverterUtil.read(IntMapEntity.class, rec);
        Assert.assertEquals(entity.getId(), actual.getId());
        Assert.assertNull(actual.getEmbedded());

    }

    @Test
    public void testEmbeddedEmpty() {

        IntMapEntity entity = new IntMapEntity();
        entity.setId(123L);
        entity.setEmbedded(Collections.<Integer, EmbeddedEntity>emptyMap());

        DattyRecord rec = new DattyRecord();

        DattyConverterUtil.write(entity, rec);

        ByteBuf bb = rec.get("id").asByteBuf();
        Assert.assertNotNull(bb);

        Long id = LongReader.INSTANCE.read(bb.duplicate(), true);
        Assert.assertNotNull(id);
        Assert.assertEquals(123L, id.longValue());

        bb = rec.get("embedded").asByteBuf();
        Assert.assertNotNull(bb);

        Object value = MessageIO.readValue(bb.duplicate(), true);
        Assert.assertNotNull(value);
        Assert.assertTrue(value instanceof Map);

        Map<Integer, Map> map = (Map<Integer, Map>) value;
        Assert.assertEquals(0, map.size());

        IntMapEntity actual = DattyConverterUtil.read(IntMapEntity.class, rec);
        Assert.assertEquals(entity.getId(), actual.getId());
        Assert.assertNotNull(actual.getEmbedded());
        Assert.assertEquals(0, actual.getEmbedded().size());

    }

    @Test
    public void testEmbeddedNull() {

        IntMapEntity entity = new IntMapEntity();
        entity.setId(123L);

        EmbeddedEntity embedded = new EmbeddedEntity();
        entity.setEmbedded(Collections.singletonMap(555, embedded));

        DattyRecord rec = new DattyRecord();

        DattyConverterUtil.write(entity, rec);

        ByteBuf bb = rec.get("id").asByteBuf();
        Assert.assertNotNull(bb);

        Long id = LongReader.INSTANCE.read(bb.duplicate(), true);
        Assert.assertNotNull(id);
        Assert.assertEquals(123L, id.longValue());

        bb = rec.get("embedded").asByteBuf();
        Assert.assertNotNull(bb);

        //System.out.println(StringMaps.toString(ByteBufUtil.getBytes(bb)));

        Object value = MessageIO.readValue(bb.duplicate(), true);
        Assert.assertNotNull(value);
        Assert.assertTrue(value instanceof Map);

        Map<Integer, Map> map = (Map<Integer, Map>) value;
        Assert.assertEquals(1, map.size());
        Assert.assertNotNull(map.get(555));
        Assert.assertNull(map.get(555).get("innerField"));

        IntMapEntity actual = DattyConverterUtil.read(IntMapEntity.class, rec);
        Assert.assertEquals(entity.getId(), actual.getId());
        Assert.assertNotNull(actual.getEmbedded());
        Assert.assertEquals(1, actual.getEmbedded().size());
        Assert.assertNull(actual.getEmbedded().get(555).getInnerField());

    }

    @Test
    public void testEmbedded() {

        IntMapEntity entity = new IntMapEntity();
        entity.setId(123L);

        EmbeddedEntity embedded = new EmbeddedEntity();
        embedded.setInnerField("inner");
        entity.setEmbedded(Collections.singletonMap(555, embedded));

        DattyRecord rec = new DattyRecord();

        DattyConverterUtil.write(entity, rec);

        ByteBuf bb = rec.get("id").asByteBuf();
        Assert.assertNotNull(bb);

        Long id = LongReader.INSTANCE.read(bb.duplicate(), true);
        Assert.assertNotNull(id);
        Assert.assertEquals(123L, id.longValue());

        bb = rec.get("embedded").asByteBuf();
        Assert.assertNotNull(bb);

        //System.out.println(StringMaps.toString(ByteBufUtil.getBytes(bb)));

        Object value = MessageIO.readValue(bb.duplicate(), true);
        Assert.assertNotNull(value);
        Assert.assertTrue(value instanceof Map);

        Map<Integer, Map> map = (Map<Integer, Map>) value;
        Assert.assertEquals(1, map.size());
        Assert.assertNotNull(map.get(555));
        Assert.assertEquals("inner", map.get(555).get("innerField"));

        IntMapEntity actual = DattyConverterUtil.read(IntMapEntity.class, rec);
        Assert.assertEquals(entity.getId(), actual.getId());
        Assert.assertNotNull(actual.getEmbedded());
        Assert.assertEquals(1, actual.getEmbedded().size());
        Assert.assertEquals("inner", actual.getEmbedded().get(555).getInnerField());

    }

}