org.hibernate.processor.validation.Mocker Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hibernate-jpamodelgen Show documentation

Annotation Processor to generate JPA 2 static metamodel classes

There is a newer version: 7.0.0.Beta4

Show newest version

/*
 * Hibernate, Relational Persistence for Idiomatic Java
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later.
 * See the lgpl.txt file in the root directory or .
 */
package org.hibernate.processor.validation;

import net.bytebuddy.ByteBuddy;

import java.lang.reflect.Constructor;
import java.util.HashMap;
import java.util.Map;
import java.util.function.Supplier;

import static net.bytebuddy.implementation.FixedValue.value;
import static net.bytebuddy.matcher.ElementMatchers.isAbstract;
import static net.bytebuddy.matcher.ElementMatchers.returns;

/**
 * @author Gavin King
 */
@FunctionalInterface
public interface Mocker {

	T make(Object... args);

	Map, Class> mocks = new HashMap<>();

	static  Supplier nullary(Class clazz) {
		try {
			Class mock = load(clazz);
			return () -> {
				try {
					return mock.newInstance();
				}
				catch (Exception e) {
					throw new RuntimeException(e);
				}
			};
		}
		catch (Exception e) {
			throw new RuntimeException(e);
		}
	}

	@SuppressWarnings("unchecked")
	static  Mocker variadic(Class clazz) {
		Constructor[] constructors = load(clazz).getDeclaredConstructors();
		if (constructors.length>1) {
			throw new RuntimeException("more than one constructor for " + clazz);
		}
		Constructor constructor = constructors[0];
		return (args) -> {
			try {
				return (T) constructor.newInstance(args);
			}
			catch (Exception e) {
				throw new RuntimeException(e);
			}
		};
	}

	@SuppressWarnings("unchecked")
	static  Class load(Class clazz) {
		if (mocks.containsKey(clazz)) {
			return (Class) mocks.get(clazz);
		}
		Class mock =
				new ByteBuddy()
						.subclass(clazz)
						.method(returns(String.class).and(isAbstract()))
								.intercept(value(""))
						.method(returns(boolean.class).and(isAbstract()))
								.intercept(value(false))
						.method(returns(int.class).and(isAbstract()))
								.intercept(value(0))
						.method(returns(long.class).and(isAbstract()))
								.intercept(value(0L))
						.method(returns(int[].class).and(isAbstract()))
								.intercept(value(new int[0]))
						.method(returns(String[].class).and(isAbstract()))
								.intercept(value(new String[0]))
						.make()
						.load(clazz.getClassLoader())
						.getLoaded();
		mocks.put(clazz,mock);
		return mock;
	}
}