io.datakernel.test.TestUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of datakernel-test Show documentation
Utils for test purposes, custom DataKernelRunner and custom JUnit rules.
The newest version!
/*
 * Copyright (C) 2015-2018 SoftIndex LLC.
 *
 * 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 io.datakernel.test;

import com.mysql.cj.jdbc.MysqlDataSource;
import io.datakernel.async.callback.Callback;
import org.jetbrains.annotations.Nullable;

import javax.sql.DataSource;
import java.io.*;
import java.net.ServerSocket;
import java.nio.charset.StandardCharsets;
import java.sql.SQLException;
import java.util.Properties;
import java.util.function.BiFunction;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;

public final class TestUtils {
	private static int activePromises = 0;

	public static synchronized int getFreePort() {
		try (ServerSocket s = new ServerSocket(0)) {
			return s.getLocalPort();
		} catch (IOException e) {
			throw new AssertionError(e);
		}
	}

	public static DataSource dataSource(String databasePropertiesPath) throws IOException, SQLException {
		Properties properties = new Properties();
		properties.load(new InputStreamReader(new BufferedInputStream(new FileInputStream(new File(databasePropertiesPath))), StandardCharsets.UTF_8));

		MysqlDataSource dataSource = new MysqlDataSource();
		dataSource.setUrl("jdbc:mysql://" + properties.getProperty("dataSource.serverName") + '/' + properties.getProperty("dataSource.databaseName"));
		dataSource.setUser(properties.getProperty("dataSource.user"));
		dataSource.setPassword(properties.getProperty("dataSource.password"));
		dataSource.setServerTimezone(properties.getProperty("dataSource.timeZone"));
		dataSource.setAllowMultiQueries(true);
		return dataSource;
	}

	public static  Callback assertComplete(ThrowingConsumer consumer) {
		activePromises++;
		return (t, e) -> {
			activePromises--;
			if (e != null) {
				if (e instanceof AssertionError) {
					throw (AssertionError) e;
				}
				throw new AssertionError(e);
			}
			try {
				consumer.accept(t);
			} catch (Throwable e2) {
				if (e2 instanceof AssertionError) {
					throw (AssertionError) e2;
				}
				throw new AssertionError(e2);
			}
		};
	}

	public static  Callback assertComplete() {
		return assertComplete($ -> {});
	}

	public static int getActivePromises() {
		return activePromises;
	}

	public static void clearActivePromises() {
		activePromises = 0;
	}

	@FunctionalInterface
	public interface ThrowingSupplier {
		T get() throws Throwable;
	}

	public static  Supplier asserting(ThrowingSupplier supplier) {
		return () -> {
			try {
				return supplier.get();
			} catch (AssertionError e) {
				throw e;
			} catch (Throwable e) {
				throw new AssertionError(e);
			}
		};
	}

	@FunctionalInterface
	public interface ThrowingConsumer {
		void accept(T t) throws Throwable;
	}

	public static  Consumer asserting(ThrowingConsumer consumer) {
		return x -> {
			try {
				consumer.accept(x);
			} catch (AssertionError e) {
				throw e;
			} catch (Throwable e) {
				throw new AssertionError(e);
			}
		};
	}

	@FunctionalInterface
	public interface ThrowingConsumerEx {
		void accept(T result, @Nullable Throwable e) throws Throwable;
	}

	public static  Callback asserting(ThrowingConsumerEx consumer) {
		return (x, y) -> {
			try {
				consumer.accept(x, y);
			} catch (RuntimeException | Error e) {
				throw e;
			} catch (Throwable throwable) {
				throw new AssertionError(throwable);
			}
		};
	}

	@FunctionalInterface
	public interface ThrowingFunction {
		R apply(T t) throws Throwable;
	}

	public static  Function asserting(ThrowingFunction function) {
		return x -> {
			try {
				return function.apply(x);
			} catch (AssertionError e) {
				throw e;
			} catch (Throwable e) {
				throw new AssertionError(e);
			}
		};
	}

	@FunctionalInterface
	public interface ThrowingBiFunction {
		R apply(T t, U u) throws Throwable;
	}

	public static  BiFunction asserting(ThrowingBiFunction function) {
		return (x, y) -> {
			try {
				return function.apply(x, y);
			} catch (AssertionError e) {
				throw e;
			} catch (Throwable e) {
				throw new AssertionError(e);
			}
		};
	}
}