108 lines
4.2 KiB
Java
108 lines
4.2 KiB
Java
/*
|
|
* Copyright (C) 2009 Google 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.google.gson.functional;
|
|
|
|
import static com.google.common.truth.Truth.assertThat;
|
|
|
|
import com.google.gson.Gson;
|
|
import com.google.gson.GsonBuilder;
|
|
import com.google.gson.JsonArray;
|
|
import com.google.gson.JsonElement;
|
|
import com.google.gson.JsonObject;
|
|
import com.google.gson.JsonSerializationContext;
|
|
import com.google.gson.JsonSerializer;
|
|
import com.google.gson.common.TestTypes.Base;
|
|
import com.google.gson.common.TestTypes.BaseSerializer;
|
|
import com.google.gson.common.TestTypes.ClassWithBaseArrayField;
|
|
import com.google.gson.common.TestTypes.ClassWithBaseField;
|
|
import com.google.gson.common.TestTypes.Sub;
|
|
import com.google.gson.common.TestTypes.SubSerializer;
|
|
import java.lang.reflect.Type;
|
|
import org.junit.Test;
|
|
|
|
/**
|
|
* Functional Test exercising custom serialization only. When test applies to both
|
|
* serialization and deserialization then add it to CustomTypeAdapterTest.
|
|
*
|
|
* @author Inderjeet Singh
|
|
*/
|
|
public class CustomSerializerTest {
|
|
|
|
@Test
|
|
public void testBaseClassSerializerInvokedForBaseClassFields() {
|
|
Gson gson = new GsonBuilder()
|
|
.registerTypeAdapter(Base.class, new BaseSerializer())
|
|
.registerTypeAdapter(Sub.class, new SubSerializer())
|
|
.create();
|
|
ClassWithBaseField target = new ClassWithBaseField(new Base());
|
|
JsonObject json = (JsonObject) gson.toJsonTree(target);
|
|
JsonObject base = json.get("base").getAsJsonObject();
|
|
assertThat(base.get(Base.SERIALIZER_KEY).getAsString()).isEqualTo(BaseSerializer.NAME);
|
|
}
|
|
|
|
@Test
|
|
public void testSubClassSerializerInvokedForBaseClassFieldsHoldingSubClassInstances() {
|
|
Gson gson = new GsonBuilder()
|
|
.registerTypeAdapter(Base.class, new BaseSerializer())
|
|
.registerTypeAdapter(Sub.class, new SubSerializer())
|
|
.create();
|
|
ClassWithBaseField target = new ClassWithBaseField(new Sub());
|
|
JsonObject json = (JsonObject) gson.toJsonTree(target);
|
|
JsonObject base = json.get("base").getAsJsonObject();
|
|
assertThat(base.get(Base.SERIALIZER_KEY).getAsString()).isEqualTo(SubSerializer.NAME);
|
|
}
|
|
|
|
@Test
|
|
public void testSubClassSerializerInvokedForBaseClassFieldsHoldingArrayOfSubClassInstances() {
|
|
Gson gson = new GsonBuilder()
|
|
.registerTypeAdapter(Base.class, new BaseSerializer())
|
|
.registerTypeAdapter(Sub.class, new SubSerializer())
|
|
.create();
|
|
ClassWithBaseArrayField target = new ClassWithBaseArrayField(new Base[] {new Sub(), new Sub()});
|
|
JsonObject json = (JsonObject) gson.toJsonTree(target);
|
|
JsonArray array = json.get("base").getAsJsonArray();
|
|
for (JsonElement element : array) {
|
|
JsonElement serializerKey = element.getAsJsonObject().get(Base.SERIALIZER_KEY);
|
|
assertThat(serializerKey.getAsString()).isEqualTo(SubSerializer.NAME);
|
|
}
|
|
}
|
|
|
|
@Test
|
|
public void testBaseClassSerializerInvokedForBaseClassFieldsHoldingSubClassInstances() {
|
|
Gson gson = new GsonBuilder()
|
|
.registerTypeAdapter(Base.class, new BaseSerializer())
|
|
.create();
|
|
ClassWithBaseField target = new ClassWithBaseField(new Sub());
|
|
JsonObject json = (JsonObject) gson.toJsonTree(target);
|
|
JsonObject base = json.get("base").getAsJsonObject();
|
|
assertThat(base.get(Base.SERIALIZER_KEY).getAsString()).isEqualTo(BaseSerializer.NAME);
|
|
}
|
|
|
|
@Test
|
|
public void testSerializerReturnsNull() {
|
|
Gson gson = new GsonBuilder()
|
|
.registerTypeAdapter(Base.class, new JsonSerializer<Base>() {
|
|
@Override public JsonElement serialize(Base src, Type typeOfSrc, JsonSerializationContext context) {
|
|
return null;
|
|
}
|
|
})
|
|
.create();
|
|
JsonElement json = gson.toJsonTree(new Base());
|
|
assertThat(json.isJsonNull()).isTrue();
|
|
}
|
|
}
|