001/**
002 * Copyright (C) 2006-2021 Talend Inc. - www.talend.com
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package org.talend.sdk.component.junit5.environment;
017
018import static org.junit.platform.commons.util.AnnotationUtils.findAnnotation;
019import static org.junit.platform.commons.util.AnnotationUtils.isAnnotated;
020
021import java.util.stream.Stream;
022
023import org.junit.jupiter.api.extension.ExtensionContext;
024import org.junit.jupiter.api.extension.TestTemplateInvocationContext;
025import org.junit.jupiter.api.extension.TestTemplateInvocationContextProvider;
026import org.talend.sdk.component.junit.environment.DecoratingEnvironmentProvider;
027import org.talend.sdk.component.junit.environment.EnvironmentsConfigurationParser;
028import org.talend.sdk.component.junit5.ComponentExtension;
029import org.talend.sdk.component.junit5.WithComponents;
030
031public class EnvironmentsExtension implements TestTemplateInvocationContextProvider {
032
033    @Override
034    public boolean supportsTestTemplate(final ExtensionContext context) {
035        return isAnnotated(context.getTestMethod(), EnvironmentalTest.class);
036    }
037
038    @Override
039    public Stream<TestTemplateInvocationContext> provideTestTemplateInvocationContexts(final ExtensionContext context) {
040        final String format = findAnnotation(context.getRequiredTestMethod(), EnvironmentalTest.class).get().format();
041        return new EnvironmentsConfigurationParser(context.getRequiredTestClass())
042                .stream()
043                .map(e -> new EnvironmentalContext(e,
044                        format
045                                .replace("${displayName}", context.getDisplayName())
046                                .replace("${environment}", DecoratingEnvironmentProvider.class.cast(e).getName()),
047                        createComponentExtension(context)));
048    }
049
050    private ComponentExtension createComponentExtension(final ExtensionContext context) {
051        return context
052                .getElement()
053                .map(it -> it.isAnnotationPresent(WithComponents.class))
054                .filter(it -> it)
055                .orElseGet(() -> context
056                        .getParent()
057                        .flatMap(it -> it.getElement().map(e -> e.isAnnotationPresent(WithComponents.class)))
058                        .orElse(false))
059                                ? context
060                                        .getStore(ExtensionContext.Namespace.create(ComponentExtension.class.getName()))
061                                        .get(ComponentExtension.class.getName() + ".instance", ComponentExtension.class)
062                                : null;
063    }
064}