dubbo-common/src/test/java/com/alibaba/dubbo/common/serialize/dubbo/BuilderTest.java

/*
 * Copyright 1999-2011 Alibaba Group.
 *  
 * 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.alibaba.dubbo.common.serialize.dubbo;

import static junit.framework.Assert.assertEquals;
import static junit.framework.Assert.assertNull;
import static junit.framework.Assert.assertTrue;

import java.io.Serializable;
import java.lang.reflect.Modifier;
import java.sql.Time;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

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

import com.alibaba.dubbo.common.io.Bytes;
import com.alibaba.dubbo.common.io.UnsafeByteArrayOutputStream;
import com.alibaba.dubbo.common.serialize.support.dubbo.Builder;

public class BuilderTest
{
    @Test
	public void testPrimaryTypeBuilder() throws Exception
	{
		System.out.println((new byte[2]).hashCode());
		Builder<String> builder = Builder.register(String.class);
		UnsafeByteArrayOutputStream os = new UnsafeByteArrayOutputStream();
		String v = "123";
		builder.writeTo(v, os);
		byte[] b = os.toByteArray();
		System.out.println(b.length+":"+Bytes.bytes2hex(b));
		v = builder.parseFrom(b);
		builder.writeTo(v, os);
		b = os.toByteArray();
		System.out.println(b.length+":"+Bytes.bytes2hex(b));
	}

    @Test
	public void testEnumBuilder() throws Exception
	{
		Builder<Type> builder = Builder.register(Type.class);
		UnsafeByteArrayOutputStream os = new UnsafeByteArrayOutputStream();
		Type v = Type.High;
		builder.writeTo(v, os);
		byte[] b = os.toByteArray();
		System.out.println(b.length+":"+Bytes.bytes2hex(b));
		v = builder.parseFrom(b);
	}

    @Test
	public void testThrowableBuilder() throws Exception
	{
		Builder<Throwable> builder = Builder.register(Throwable.class);
		Throwable th = new Throwable();
		UnsafeByteArrayOutputStream os = new UnsafeByteArrayOutputStream();
		builder.writeTo(th, os);
		byte[] b = os.toByteArray();
		System.out.println(b.length+":"+Bytes.bytes2hex(b));

		th = builder.parseFrom(b);
	}

    @Test
	public void testArrayClassBuilder() throws Exception
	{
		UnsafeByteArrayOutputStream os;

		byte[] b;

		Builder<Object[]> osb = Builder.register(Object[].class);
		os = new UnsafeByteArrayOutputStream();
		osb.writeTo(new Object[]{ new String[0] }, os);
		b = os.toByteArray();

		Builder<long[]> lsb = Builder.register(long[].class);
		os = new UnsafeByteArrayOutputStream();
		lsb.writeTo(new long[]{ 1,121232,-3,4,-5,61321432413l }, os);
		lsb.writeTo(new long[]{ 1,121232,-3,4,-5,61321432413l }, os);
		lsb.writeTo(new long[]{ 1,2,3,12131314,123132313135l,-6 }, os);
		b = os.toByteArray();
		long[] ls = lsb.parseFrom(b);
		assertEquals(ls.length, 6);

		Builder<byte[]> bsb = Builder.register(byte[].class);
		os = new UnsafeByteArrayOutputStream();
		bsb.writeTo("i am a string.".getBytes(), os);
		b = os.toByteArray();

		Builder<int[][]> iisb = Builder.register(int[][].class);
		os = new UnsafeByteArrayOutputStream();
		iisb.writeTo(new int[][]{ {1,2,3,4}, {5,6,7,8}, {9,10}, {122,123,444} }, os);
		b = os.toByteArray();
		int[][] iis = iisb.parseFrom(b);
		assertEquals(iis.length, 4);

		Builder<int[][][]> iiisb = Builder.register(int[][][].class);
		os = new UnsafeByteArrayOutputStream();
		iiisb.writeTo(new int[][][]{ 
			{{1,2,3,4}},
			{{5,6,7,8}},
			{{122,123,444}}
		}, os);
		b = os.toByteArray();
		int[][][] iii = iiisb.parseFrom(b);
		assertEquals(iii.length, 3);
	}

    @Test
	public void testObjectBuilder() throws Exception
	{
		UnsafeByteArrayOutputStream os = new UnsafeByteArrayOutputStream();
		Builder<Bean> BeanBuilder = Builder.register(Bean.class);

		Bean bean = new Bean();
		bean.name = "ql";
		bean.type = Type.High;
		bean.types = new Type[]{ Type.High, Type.High };
		BeanBuilder.writeTo(bean, os);

		byte[] b = os.toByteArray();
		System.out.println(b.length+":"+Bytes.bytes2hex(b));

		bean = BeanBuilder.parseFrom(b);
		assertNull(bean.time);
		assertEquals(bean.i, 123123);
		assertEquals(bean.ni, -12344);
		assertEquals(bean.d, 12.345);
		assertEquals(bean.nd, -12.345);
		assertEquals(bean.l, 1281447759383l);
		assertEquals(bean.nl, -13445l);
		assertEquals(bean.vl, 100l);
		assertEquals(bean.type, Type.High);
		assertEquals(bean.types.length, 2);
		assertEquals(bean.types[0], Type.High);
		assertEquals(bean.types[1], Type.High);
		assertEquals(bean.list.size(), 3);
		assertEquals(bean.list.get(0), 1);
		assertEquals(bean.list.get(1), 2);
		assertEquals(bean.list.get(2), 1308147);
	}

    @Test
	public void testInterfaceBuilder() throws Exception
	{
		UnsafeByteArrayOutputStream os = new UnsafeByteArrayOutputStream();
		Builder<TestDO> builder = Builder.register(TestDO.class);
		TestDO d = new TestDOImpl();
		builder.writeTo(d, os);

		byte[] b = os.toByteArray();

		d = builder.parseFrom(b);
		assertTrue(TestDO.class.isAssignableFrom(d.getClass()));
		assertEquals("name", d.getName());
		assertEquals(28, d.getArg());
		assertEquals(Type.High, d.getType());
	}

    @Test
	public void testGenericBuilder() throws Exception
	{
		UnsafeByteArrayOutputStream os = new UnsafeByteArrayOutputStream();
		Builder<Object> ob = Builder.register(Object.class);

		Object o = new Object();
		ob.writeTo(o, os);
		byte[] b = os.toByteArray();

		os = new UnsafeByteArrayOutputStream();
		Bean bean = new Bean();
		bean.name = "ql";
		bean.type = Type.High;
		bean.types = new Type[]{ Type.High, Type.High };
		ob.writeTo(bean, os);

		b = os.toByteArray();
		bean = (Bean)ob.parseFrom(b);
		assertEquals(bean.i, 123123);
		assertEquals(bean.ni, -12344);
		assertEquals(bean.d, 12.345);
		assertEquals(bean.nd, -12.345);
		assertEquals(bean.l, 1281447759383l);
		assertEquals(bean.nl, -13445l);
		assertEquals(bean.vl, 100l);
		assertEquals(bean.type, Type.High);
		assertEquals(bean.types.length, 2);
		assertEquals(bean.types[0], Type.High);
		assertEquals(bean.types[1], Type.High);
		assertEquals(bean.list.size(), 3);
		assertEquals(bean.list.get(0), 1);
		assertEquals(bean.list.get(1), 2);
		assertEquals(bean.list.get(2), 1308147);
	}

    @Test
	public void testObjectArrayBuilder() throws Exception
	{
		UnsafeByteArrayOutputStream os = new UnsafeByteArrayOutputStream();
		Builder<Object[]> builder = Builder.register(Object[].class);

		Object[] obj = new Object[5];
		obj[0] = "1234";
		obj[1] = new Double(109.23);
		obj[2] = "3455";
		obj[3] = null;
		obj[4] = Boolean.TRUE;

		builder.writeTo(obj, os);
		byte[] b = os.toByteArray();
		System.out.println("Object array:"+b.length+":"+Bytes.bytes2hex(b));

		Assert.assertArrayEquals(obj, builder.parseFrom(b));
	}

    // FIXME MyList的从ArrayList中继承来的属性size会在decode时设置好，再Add时就不对了！！
    @Ignore
    @Test
    @SuppressWarnings({ "rawtypes", "unchecked" })
    public void testBuilder_MyList() throws Exception
	{
        Builder<MyList> b1 = Builder.register(MyList.class);
		MyList list = new MyList();
		list.add(new boolean[]{ true,false });
		list.add(new int[]{ 1,2,3,4,5 });
		list.add("String");
		list.add(4);
		list.code = 4321;
		
		UnsafeByteArrayOutputStream os = new UnsafeByteArrayOutputStream();
		b1.writeTo(list, os);
		byte[] b = os.toByteArray();
		System.out.println(b.length+":"+Bytes.bytes2hex(b));
		MyList result = b1.parseFrom(b);

		assertEquals(4, result.size());
		assertEquals(result.code, 4321);
		assertEquals(result.id, "feedback");
	}
    
    @Test
    @SuppressWarnings({ "rawtypes", "unchecked" })
    public void testBuilder_MyMap() throws Exception
    {
        UnsafeByteArrayOutputStream os = new UnsafeByteArrayOutputStream();
        Builder<MyMap> b2 = Builder.register(MyMap.class);
        MyMap map = new MyMap();
        map.put("name", "qianlei");
        map.put("displayName", "钱磊");
        map.code = 4321;
        b2.writeTo(map, os);
        byte[] b = os.toByteArray();
        System.out.println(b.length+":"+Bytes.bytes2hex(b));
        
        map = b2.parseFrom(b);
        
        assertEquals(map.size(), 2);
        assertEquals(map.code, 4321);
        assertEquals(map.id, "feedback");
    }

    @Test
	@SuppressWarnings("unchecked")
	public void testSerializableBean() throws Exception
	{
		System.out.println("testSerializableBean");
		UnsafeByteArrayOutputStream os = new UnsafeByteArrayOutputStream();

		SerializableBean sb = new SerializableBean();
		Builder<SerializableBean> sbb = Builder.register(SerializableBean.class);
		sbb.writeTo(sb, os);

		byte[] b = os.toByteArray();
		System.out.println(b.length+":"+Bytes.bytes2hex(b));
		assertEquals(sbb.parseFrom(os.toByteArray()), sb);
	}

    @Test
	@SuppressWarnings("unchecked")
	public void testOthers() throws Exception
	{
		UnsafeByteArrayOutputStream os = new UnsafeByteArrayOutputStream();

		StringBuffer buf = new StringBuffer();
		for(int i=0;i<1024*32+32;i++)
			buf.append('A');
		Builder<String> sb = Builder.register(String.class);
		sb.writeTo(buf.toString(), os);
		assertEquals(sb.parseFrom(os.toByteArray()), buf.toString());

		os = new UnsafeByteArrayOutputStream();
		Builder<HashMap> builder = Builder.register(HashMap.class);
		Map services = new HashMap();
		HashMap map = new HashMap();
		services.put("test.service", "http://127.0.0.1:9010/test.service");
		map.put("name", "qianlei");
		map.put("password", "123455");
		map.put("services", services);

		builder.writeTo(map, os);
		byte[] b = os.toByteArray();
		System.out.println(b.length+":"+Bytes.bytes2hex(b));
		map = builder.parseFrom(b);
		assertTrue(map.size() > 0);
        assertEquals("http://127.0.0.1:9010/test.service", ((Map) map.get("services")).get("test.service"));

		services = new ConcurrentHashMap();
		services.put("test.service", "http://127.0.0.1:9010/test.service");
		map.put("services", services);

		os = new UnsafeByteArrayOutputStream();
		builder.writeTo(map, os);
		b = os.toByteArray();
		System.out.println(b.length+":"+Bytes.bytes2hex(b));
		map = builder.parseFrom(b);
		assertTrue(map.size() > 0);
		assertEquals("http://127.0.0.1:9010/test.service", ((Map) map.get("services")).get("test.service"));

		Node node1 = new Node();
		Node node0 = new Node();
		node0.value = "0";
		node0.next = node1;
		node1.value = "1";
		node1.prev = node0;
		// write.
		Builder<Node> nodebuilder = Builder.register(Node.class);
		os = new UnsafeByteArrayOutputStream();
		nodebuilder.writeTo(node0, os);
		b = os.toByteArray();
		System.out.println("Node:"+b.length+":"+Bytes.bytes2hex(b));
		// parse
		node0 = nodebuilder.parseFrom(b);
		assertEquals(node0, node0.prev);
		assertEquals(node0, node0.next.prev);
		assertEquals(node0.value, "0");
	}
    public static void main(String[] args) {
        System.out.println(Modifier.isPublic(String.class.getModifiers()));
    }
    @Test
	public void testWithFC() throws Exception
	{
		Builder<SimpleDO> builder = Builder.register(SimpleDO.class);
		UnsafeByteArrayOutputStream os = new UnsafeByteArrayOutputStream();

		SimpleDO sd = new SimpleDO();
		sd.a = 1;
		sd.b = 2;
		sd.c = 3;
		sd.str1 = "12345";
		sd.str2 = "54321";
		builder.writeTo(sd, os);
		byte[] b = os.toByteArray();
		System.out.println(b.length+":"+Bytes.bytes2hex(b));

		sd = builder.parseFrom(b);
		assertEquals(sd.a, 1);
		assertEquals(sd.b, 2);
		assertEquals(sd.c, 3);
		assertEquals(sd.str1, "124");
		System.out.println(sd.str2);
	}

	public enum Type
	{
		Lower, Normal, High;
	}

	static interface TestDO
	{
		String getName();
		void setName(String name);
		Type getType();
		void setType(Type t);
		int getArg();
		void setArg(int arg);
	}

	static class TestDOImpl implements TestDO, Serializable
	{
	    private static final long serialVersionUID = 1L;
		public String getName()
		{
			return "name";
		}
		public void setName(String name){}
		public Type getType()
		{
			return Type.High;
		}
		public void setType(Type t){}
		public int getArg()
		{
			return 28;
		}
		public void setArg(int arg){}
	}

	static class Bean implements Serializable
	{
	    private static final long serialVersionUID = 1L;
		public int vi = 0;
		public long vl = 100l;

		boolean b = true;
		boolean[] bs = {false, true};

		String s1 = "1234567890";
		String s2 = "1234567890一二三四五六七八九零";

		private int i = 123123, ni = -12344, is[] = {1,2,3,4,-1,-2,-3,-4};
        private short s = 12, ns = -76;
		private double d = 12.345, nd = -12.345;
		private long l = 1281447759383l, nl = -13445l;

		private Boolean B = Boolean.FALSE;
		private Integer I = -1234;
		private Double D = new Double(1.23);
		private String name = "qianlei";
		private Type type = Type.Lower, type1 = Type.Normal;
		private Type[] types = { Type.Lower, Type.Lower };

		private Time time = null;

		public Type getType()
		{
			return type;
		}

		public void setType(Type type)
		{
			this.type = type;
		}

		private ArrayList list = new ArrayList();
		{
			list.add(1);
			list.add(2);
			list.add(1308147);
		}
	}

	static class MyList<T> extends ArrayList<T>
	{
		private int code = 12345;
		private String id = "feedback";
	}

	static class MyMap<K, V> extends HashMap<K, V>
	{
		private int code = 12345;
		private String id = "feedback";
	}

	static class Node implements Serializable
	{
        private static final long serialVersionUID = 1L;
        Node prev = this;
		Node next = this;
		String value = "value";

		@Override
        public int hashCode() {
            final int prime = 31;
            int result = 1;
            result = prime * result + ((value == null) ? 0 : value.hashCode());
            return result;
        }

        @Override
        public boolean equals(Object obj) {
            if (this == obj) return true;
            if (obj == null) return false;
            if (getClass() != obj.getClass()) return false;
            Node other = (Node) obj;
            if (value == null) {
                if (other.value != null) return false;
            } else if (!value.equals(other.value)) return false;
            return true;
        }
	}

	static class SerializableBean implements Serializable
	{
		private static final long serialVersionUID = -8949681707161463700L;

		public int a = 0;
		public long b = 100l;
		boolean c = true;
		String s1 = "1234567890";
		String s2 = "1234567890一二三四五六七八九零";

		public int hashCode()
		{
			return s1.hashCode() ^ s2.hashCode();
		}

		public boolean equals(Object obj)
		{
			if( obj == null ) return false;
			if( obj == this ) return true;
			if( obj instanceof SerializableBean )
			{
				SerializableBean sb = (SerializableBean)obj;
				return this.a == sb.a && this.b == sb.b && this.c == sb.c && this.s1.equals(sb.s1) && this.s2.equals(sb.s2); 
			}

			return false;
		}
	}
}