org.eclipse.edc.junit.extensions.RuntimeExtension Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of junit Show documentation
edc :: junit
There is a newer version: 0.10.1
/*
 *  Copyright (c) 2024 Bayerische Motoren Werke Aktiengesellschaft (BMW AG)
 *
 *  This program and the accompanying materials are made available under the
 *  terms of the Apache License, Version 2.0 which is available at
 *  https://www.apache.org/licenses/LICENSE-2.0
 *
 *  SPDX-License-Identifier: Apache-2.0
 *
 *  Contributors:
 *       Bayerische Motoren Werke Aktiengesellschaft (BMW AG) - initial API and implementation
 *
 */

package org.eclipse.edc.junit.extensions;

import org.eclipse.edc.spi.system.ConfigurationExtension;
import org.eclipse.edc.spi.system.SystemExtension;
import org.eclipse.edc.spi.system.configuration.ConfigFactory;
import org.junit.jupiter.api.extension.ExtensionContext;
import org.junit.jupiter.api.extension.ParameterContext;
import org.junit.jupiter.api.extension.ParameterResolutionException;
import org.junit.jupiter.api.extension.ParameterResolver;

import java.util.Map;

import static org.eclipse.edc.util.types.Cast.cast;

public abstract class RuntimeExtension implements ParameterResolver {

    protected final EmbeddedRuntime runtime;

    protected RuntimeExtension(EmbeddedRuntime runtime) {
        this.runtime = runtime;
    }

    @Override
    public boolean supportsParameter(ParameterContext parameterContext, ExtensionContext extensionContext) throws ParameterResolutionException {
        var type = parameterContext.getParameter().getParameterizedType();
        if (type.equals(RuntimeExtension.class)) {
            return true;
        } else if (type.equals(EmbeddedRuntime.class)) {
            return true;
        } else if (type instanceof Class) {
            return runtime.getContext().hasService(cast(type));
        }
        return false;
    }

    @Override
    public Object resolveParameter(ParameterContext parameterContext, ExtensionContext extensionContext) throws ParameterResolutionException {
        var type = parameterContext.getParameter().getParameterizedType();
        if (type.equals(RuntimeExtension.class)) {
            return this;
        } else if (type.equals(EmbeddedRuntime.class)) {
            return runtime;
        } else if (type instanceof Class) {
            return runtime.getContext().getService(cast(type));
        }
        return null;
    }

    /**
     * Registers a mock service with the runtime. Note that the mock will overwrite any service already registered with the context.
     * This is intentional. A warning will be logged to STDOUT if the mock actually replaces an existing service.
     *
     * @param mock the service mock
     */
    public  RuntimeExtension registerServiceMock(Class type, T mock) {
        runtime.registerServiceMock(type, mock);
        return this;
    }

    /**
     * Registers a service extension with the runtime.
     */
    public  RuntimeExtension registerSystemExtension(Class type, SystemExtension extension) {
        runtime.registerSystemExtension(type, extension);
        return this;
    }

    public RuntimeExtension setConfiguration(Map configuration) {
        registerSystemExtension(ConfigurationExtension.class, (ConfigurationExtension) () -> ConfigFactory.fromMap(configuration));
        return this;
    }

    public  T getService(Class clazz) {
        return runtime.getService(clazz);
    }
}