Hessian2SerializationTest.java

/*
 *  Copyright 2009-2016 Weibo, Inc.
 *
 *    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 com.weibo.api.motan.serialize;

import java.io.IOException;
import java.io.Serializable;
import java.sql.Timestamp;
import java.util.HashMap;
import java.util.Map;

import junit.framework.TestCase;

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

import com.weibo.api.motan.serialize.SubModel.SerializationObject1;

/**
 * @author maijunsheng
 * @version 创建时间：2013-6-3
 * 
 */
public class Hessian2SerializationTest extends TestCase {
    Hessian2Serialization hessian2Serialization = new Hessian2Serialization();

    @Test
    public void testCompatibility() throws Exception {
        SerializationObject1 test1 = new SerializationObject1();

        SerializationObject1 result = hessian2Serialization.deserialize(hessian2Serialization.serialize(test1), SerializationObject1.class);
        assertEquals(test1.a, result.a);
        assertEquals(test1.b, result.b);
    }

    @Test
    public void testSerialization() throws IOException {
        Model model = new Model();

        model.add("world1", new SubModel("hello1", 1));
        model.add("world2", new SubModel("hello2", 2));
        model.add("world3", new SubModel("hello3", 2));

        Model result = hessian2Serialization.deserialize(hessian2Serialization.serialize(model), Model.class);

        Assert.assertNotNull(result);
        Assert.assertEquals(model.toString(), result.toString());

        UserAttentions user = new UserAttentions();
        user.setAddTimes(null);
        UserAttentions userResult = hessian2Serialization.deserialize(hessian2Serialization.serialize(user), UserAttentions.class);
        assertNull(userResult.getAddTimes());
    }

    @Test
    public void testHessianBug() throws Exception {
        UserAttentions ua1 = new UserAttentions(1234L, new long[] {}, new long[] {}, 0, 0L);
        UserAttentions ua2 = new UserAttentions(1234L, new long[] {}, new long[] {}, 0);
        Hessian2Serialization s = new Hessian2Serialization();
        byte[] bytes1 = s.serialize(ua1);
        byte[] bytes2 = s.serialize(ua2);
        assertEquals(bytes1.length, bytes2.length);

        UserAttentions result = s.deserialize(bytes1, UserAttentions.class);
        assertEquals(ua2.getUid(), result.getUid());

    }

    @Test
    public void testHessianTimeStampBug() throws Exception {
        UserAttentions ua = new UserAttentions(1234L, new long[] {}, new long[] {}, 0, 0L);
        Timestamp timestamp = ua.getTimeStamp();

        Hessian2Serialization s = new Hessian2Serialization();
        byte[] b = s.serialize(ua);
        UserAttentions result = s.deserialize(b, UserAttentions.class);
        assertEquals(result.getTimeStamp().getTime(), timestamp.getTime());

        ua.setTimeStamp(null);

        b = s.serialize(ua);

        result = s.deserialize(b, UserAttentions.class);
        assertNull(result.getTimeStamp());
    }

    @Test
    public void testHessianDeserializeException() throws Exception {
        UserAttentions user = new UserAttentions();
        user.setAddTimes(null);
        try {
            hessian2Serialization.deserialize(hessian2Serialization.serialize(user), UnModel.class);

        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}


class UnModel {

}


class Model implements Serializable {
    private static final long serialVersionUID = -6654784635984161860L;

    private Map<String, SubModel> map = new HashMap<String, SubModel>();

    private Integer index = 10;

    public void add(String key, SubModel value) {
        map.put(key, value);
    }

    public String toString() {
        StringBuilder builder = new StringBuilder();
        builder.append(index).append("\n");

        for (Map.Entry<String, SubModel> entry : map.entrySet()) {
            builder.append(entry.getKey()).append("     ").append(entry.getValue().toString()).append("\n");
        }

        return builder.toString();
    }
}


class SubModel implements Serializable {
    private static final long serialVersionUID = 4797257086799637170L;

    private String name;
    private int age;
    private long[] addTimes = null; // add attention/fan/filter times

    public SubModel(String name, int age) {
        this.name = name;
        this.age = age;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    public String toString() {
        return name + ":" + age;
    }

    public long[] getAddTimes() {
        return addTimes;
    }

    public void setAddTimes(long[] addTimes) {
        this.addTimes = addTimes;
    }

    static class SerializationObject1 implements Serializable {
        private static final long serialVersionUID = -4917299054507959821L;
        public int a = 10;
        public long b = 20;
    }

    static class SerializationObject2 implements Serializable {
        private static final long serialVersionUID = 302091199018752193L;

        public int a = 10;
        public int c = 30;
    }

}