/*
* Copyright (C) 2012 Facebook, 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.facebook.swift.codec;
import com.facebook.swift.codec.internal.EnumThriftCodec;
import com.facebook.swift.codec.internal.ThriftCodecFactory;
import com.facebook.swift.codec.internal.coercion.DefaultJavaCoercions;
import com.facebook.swift.codec.metadata.ThriftCatalog;
import com.facebook.swift.codec.metadata.ThriftEnumMetadata;
import com.facebook.swift.codec.metadata.ThriftEnumMetadataBuilder;
import com.facebook.swift.codec.metadata.ThriftStructMetadata;
import com.facebook.swift.codec.metadata.ThriftType;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import org.apache.thrift.protocol.TCompactProtocol;
import org.apache.thrift.transport.TMemoryBuffer;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
import java.nio.ByteBuffer;
import static com.facebook.swift.codec.metadata.ThriftType.BOOL;
import static com.facebook.swift.codec.metadata.ThriftType.BYTE;
import static com.facebook.swift.codec.metadata.ThriftType.DOUBLE;
import static com.facebook.swift.codec.metadata.ThriftType.I16;
import static com.facebook.swift.codec.metadata.ThriftType.I32;
import static com.facebook.swift.codec.metadata.ThriftType.I64;
import static com.facebook.swift.codec.metadata.ThriftType.STRING;
import static com.facebook.swift.codec.metadata.ThriftType.enumType;
import static com.facebook.swift.codec.metadata.ThriftType.list;
import static com.facebook.swift.codec.metadata.ThriftType.map;
import static com.facebook.swift.codec.metadata.ThriftType.set;
import static com.google.common.base.Charsets.UTF_8;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertNotNull;
import static org.testng.Assert.fail;
public class TestThriftCodecManager
{
private ThriftCodecManager codecManager;
private ThriftType fruitType;
@BeforeMethod
protected void setUp()
throws Exception
{
codecManager = new ThriftCodecManager(new ThriftCodecFactory()
{
@Override
public <T> ThriftCodec<T> generateThriftTypeCodec(ThriftCodecManager codecManager, ThriftStructMetadata<T> metadata)
{
throw new UnsupportedOperationException();
}
});
ThriftCatalog catalog = codecManager.getCatalog();
catalog.addDefaultCoercions(DefaultJavaCoercions.class);
fruitType = catalog.getThriftType(Fruit.class);
codecManager.addCodec(new EnumThriftCodec<>(fruitType));
}
@Test
public void testBasicTypes()
throws Exception
{
testRoundTripSerialize(true);
testRoundTripSerialize(false);
testRoundTripSerialize((byte) 100);
testRoundTripSerialize((short) 1000);
testRoundTripSerialize(10000);
testRoundTripSerialize((long) 10000000);
testRoundTripSerialize(42.42d);
testRoundTripSerialize("some string");
}
@Test
public void testBasicThriftTypes()
throws Exception
{
testRoundTripSerialize(BOOL, true);
testRoundTripSerialize(BOOL, false);
testRoundTripSerialize(BYTE, (byte) 100);
testRoundTripSerialize(I16, (short) 1000);
testRoundTripSerialize(I32, 10000);
testRoundTripSerialize(I64, (long) 10000000);
testRoundTripSerialize(DOUBLE, 42.42d);
testRoundTripSerialize(STRING, toByteBuffer("some string"));
}
@Test
public void testEnum()
throws Exception
{
ThriftEnumMetadata<Fruit> fruitEnumMetadata = new ThriftEnumMetadataBuilder<>(Fruit.class).build();
ThriftEnumMetadata<Letter> letterEnumMetadata = new ThriftEnumMetadataBuilder<>(Letter.class).build();
testRoundTripSerialize(Fruit.CHERRY);
testRoundTripSerialize(Letter.C);
testRoundTripSerialize(enumType(fruitEnumMetadata), Fruit.CHERRY);
testRoundTripSerialize(enumType(letterEnumMetadata), Letter.C);
testRoundTripSerialize(list(enumType(fruitEnumMetadata)), ImmutableList.copyOf(Fruit.values()));
testRoundTripSerialize(list(enumType(letterEnumMetadata)), ImmutableList.copyOf(Letter.values()));
}
@Test
public void testCollectionThriftTypes()
throws Exception
{
testRoundTripSerialize(set(STRING), ImmutableSet.of(toByteBuffer("some string"), toByteBuffer("another string")));
testRoundTripSerialize(list(STRING), ImmutableList.of(toByteBuffer("some string"), toByteBuffer("another string")));
testRoundTripSerialize(map(STRING, STRING), ImmutableMap.of(toByteBuffer("1"), toByteBuffer("one"), toByteBuffer("2"), toByteBuffer("two")));
}
@Test
public void testCoercedCollection()
throws Exception
{
testRoundTripSerialize(set(I32.coerceTo(Integer.class)), ImmutableSet.of(1, 2, 3));
testRoundTripSerialize(list(I32.coerceTo(Integer.class)), ImmutableList.of(4, 5, 6));
testRoundTripSerialize(map(I32.coerceTo(Integer.class), I32.coerceTo(Integer.class)), ImmutableMap.of(1, 2, 2, 4, 3, 9));
}
@Test
public void testAddStructCodec()
throws Exception
{
BonkField bonk = new BonkField("message", 42);
// no codec for BonkField so this will fail
try {
testRoundTripSerialize(bonk);
fail("Expected exception");
}
catch (Exception ignored) {
}
// add the codec
ThriftType type = codecManager.getCatalog().getThriftType(BonkField.class);
codecManager.addCodec(new BonkFieldThriftCodec(type));
// try again
testRoundTripSerialize(bonk);
}
@Test
public void testAddUnionCodec()
throws Exception
{
UnionField union = new UnionField();
union._id= 1;
union.stringValue = "Hello, World";
// no codec for UnionField so this will fail
try {
testRoundTripSerialize(union);
fail("Expected exception");
}
catch (Exception ignored) {
}
// add the codec
ThriftType type = codecManager.getCatalog().getThriftType(UnionField.class);
codecManager.addCodec(new UnionFieldThriftCodec(type, codecManager.getCodec(Fruit.class)));
// try again
testRoundTripSerialize(union);
union = new UnionField();
union._id = 2;
union.longValue = 4815162342L;
testRoundTripSerialize(union);
union = new UnionField();
union._id = 3;
union.fruitValue = Fruit.BANANA;
testRoundTripSerialize(union);
}
private <T> void testRoundTripSerialize(T value)
throws Exception
{
// write value
TMemoryBuffer transport = new TMemoryBuffer(10 * 1024);
TCompactProtocol protocol = new TCompactProtocol(transport);
codecManager.write((Class<T>) value.getClass(), value, protocol);
// read value back
T copy = codecManager.read((Class<T>) value.getClass(), protocol);
assertNotNull(copy);
// verify they are the same
assertEquals(copy, value);
}
private <T> void testRoundTripSerialize(ThriftType type, T value)
throws Exception
{
// write value
TMemoryBuffer transport = new TMemoryBuffer(10 * 1024);
TCompactProtocol protocol = new TCompactProtocol(transport);
codecManager.write(type, value, protocol);
// read value back
T copy = (T) codecManager.read(type, protocol);
assertNotNull(copy);
// verify they are the same
assertEquals(copy, value);
}
private ByteBuffer toByteBuffer(String string)
{
return ByteBuffer.wrap(string.getBytes(UTF_8));
}
}