@Test(groups = "long")
public void serializeDeserializeCollectionsTest() {
List<String> l = Arrays.asList("foo", "bar");
DataType dt = DataType.list(DataType.text());
assertEquals(dt.deserialize(dt.serialize(l)), l);
try {
DataType.list(DataType.bigint()).serialize(l);
fail("This should not have worked");
} catch (InvalidTypeException e) {
/* That's what we want */
}
}
/** Generates the sample collections that will be used in testing */
private static HashMap<DataType, Object> getSampleCollections() {
HashMap<DataType, Object> sampleCollections = new HashMap<DataType, Object>();
HashMap<DataType, Object> setAndListCollection;
HashMap<DataType, HashMap<DataType, Object>> mapCollection;
for (DataType.Name dataTypeName : DATA_TYPE_NON_PRIMITIVE_NAMES) {
switch (dataTypeName) {
case LIST:
for (DataType typeArgument : DATA_TYPE_PRIMITIVES) {
if (exclude(typeArgument)) continue;
List<Object> list = new ArrayList<Object>();
for (int i = 0; i < 5; i++) {
list.add(SAMPLE_DATA.get(typeArgument));
}
setAndListCollection = new HashMap<DataType, Object>();
setAndListCollection.put(typeArgument, list);
sampleCollections.put(DataType.list(typeArgument), setAndListCollection);
}
break;
case SET:
for (DataType typeArgument : DATA_TYPE_PRIMITIVES) {
if (exclude(typeArgument)) continue;
Set<Object> set = new HashSet<Object>();
for (int i = 0; i < 5; i++) {
set.add(SAMPLE_DATA.get(typeArgument));
}
setAndListCollection = new HashMap<DataType, Object>();
setAndListCollection.put(typeArgument, set);
sampleCollections.put(DataType.set(typeArgument), setAndListCollection);
}
break;
case MAP:
for (DataType typeArgument1 : DATA_TYPE_PRIMITIVES) {
if (exclude(typeArgument1)) continue;
for (DataType typeArgument2 : DATA_TYPE_PRIMITIVES) {
if (exclude(typeArgument2)) continue;
HashMap<DataType, Object> map = new HashMap<DataType, Object>();
map.put(typeArgument1, SAMPLE_DATA.get(typeArgument2));
mapCollection = new HashMap<DataType, HashMap<DataType, Object>>();
mapCollection.put(typeArgument1, map);
sampleCollections.put(DataType.map(typeArgument1, typeArgument2), mapCollection);
}
}
break;
default:
throw new RuntimeException("Missing handling of " + dataTypeName);
}
}
return sampleCollections;
}
