PojoSubclassSerializerTest.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 org.apache.flink.api.java.typeutils.runtime;

import org.apache.flink.api.common.ExecutionConfig;
import org.apache.flink.api.common.typeinfo.TypeInformation;
import org.apache.flink.api.common.typeutils.SerializerTestBase;
import org.apache.flink.api.common.typeutils.TypeSerializer;
import org.apache.flink.api.java.typeutils.TypeExtractor;

import org.junit.Test;

import java.util.Objects;
import java.util.Random;

/**
 * A test for the {@link PojoSerializer}.
 */
public class PojoSubclassSerializerTest extends SerializerTestBase<PojoSubclassSerializerTest.TestUserClassBase> {
	private TypeInformation<TestUserClassBase> type = TypeExtractor.getForClass(TestUserClassBase.class);

	@Override
	protected TypeSerializer<TestUserClassBase> createSerializer() {
		// only register one of the three child classes, the third child class is NO POJO
		ExecutionConfig conf = new ExecutionConfig();
		conf.registerPojoType(TestUserClass1.class);
		TypeSerializer<TestUserClassBase> serializer = type.createSerializer(conf);
		assert(serializer instanceof PojoSerializer);
		return serializer;
	}

	@Override
	protected int getLength() {
		return -1;
	}

	@Override
	protected Class<TestUserClassBase> getTypeClass() {
		return TestUserClassBase.class;
	}

	@Override
	protected TestUserClassBase[] getTestData() {
		Random rnd = new Random(874597969123412341L);

		return new TestUserClassBase[]{
				new TestUserClass1(rnd.nextInt(), "foo", rnd.nextLong()),
				new TestUserClass2(rnd.nextInt(), "bar", rnd.nextFloat()),
				new TestUserClass3(rnd.nextInt(), "bar", rnd.nextFloat())
		};

	}

	@Override
	@Test
	public void testInstantiate() {
		// don't do anything, since the PojoSerializer with subclass will return null
	}

	// User code class for testing the serializer
	public static abstract class TestUserClassBase {
		public int dumm1;
		public String dumm2;


		public TestUserClassBase() {
		}

		public TestUserClassBase(int dumm1, String dumm2) {
			this.dumm1 = dumm1;
			this.dumm2 = dumm2;
		}

		@Override
		public int hashCode() {
			return Objects.hash(dumm1, dumm2);
		}

		@Override
		public boolean equals(Object other) {
			if (!(other instanceof TestUserClassBase)) {
				return false;
			}
			TestUserClassBase otherTUC = (TestUserClassBase) other;
			if (dumm1 != otherTUC.dumm1) {
				return false;
			}
			if (!dumm2.equals(otherTUC.dumm2)) {
				return false;
			}
			return true;
		}
	}

	public static class TestUserClass1 extends TestUserClassBase {
		public long dumm3;

		public TestUserClass1() {
		}

		public TestUserClass1(int dumm1, String dumm2, long dumm3) {
			super(dumm1, dumm2);
			this.dumm3 = dumm3;
		}

		@Override
		public boolean equals(Object other) {
			if (!(other instanceof TestUserClass1)) {
				return false;
			}
			TestUserClass1 otherTUC = (TestUserClass1) other;
			if (dumm1 != otherTUC.dumm1) {
				return false;
			}
			if (!dumm2.equals(otherTUC.dumm2)) {
				return false;
			}
			if (dumm3 != otherTUC.dumm3) {
				return false;
			}
			return true;
		}
	}

	public static class TestUserClass2 extends TestUserClassBase {
		public float dumm4;

		public TestUserClass2() {
		}

		public TestUserClass2(int dumm1, String dumm2, float dumm4) {
			super(dumm1, dumm2);
			this.dumm4 = dumm4;
		}

		@Override
		public boolean equals(Object other) {
			if (!(other instanceof TestUserClass2)) {
				return false;
			}
			TestUserClass2 otherTUC = (TestUserClass2) other;
			if (dumm1 != otherTUC.dumm1) {
				return false;
			}
			if (!dumm2.equals(otherTUC.dumm2)) {
				return false;
			}
			if (dumm4 != otherTUC.dumm4) {
				return false;
			}
			return true;
		}
	}

	public static class TestUserClass3 extends TestUserClassBase {
		public float dumm4;

		public TestUserClass3(int dumm1, String dumm2, float dumm4) {
			super(dumm1, dumm2);
			this.dumm4 = dumm4;
		}

		@Override
		public boolean equals(Object other) {
			if (!(other instanceof TestUserClass3)) {
				return false;
			}
			TestUserClass3 otherTUC = (TestUserClass3) other;
			if (dumm1 != otherTUC.dumm1) {
				return false;
			}
			if (!dumm2.equals(otherTUC.dumm2)) {
				return false;
			}
			if (dumm4 != otherTUC.dumm4) {
				return false;
			}
			return true;
		}
	}
}