io.kroxylicious.testing.kafka.invm.InVMProvisioningStrategy Maven / Gradle / Ivy

Go to download
/*
 * Copyright Kroxylicious Authors.
 *
 * Licensed under the Apache Software License version 2.0, available at http://www.apache.org/licenses/LICENSE-2.0
 */
package io.kroxylicious.testing.kafka.invm;

import java.lang.annotation.Annotation;
import java.time.Duration;
import java.util.List;

import org.junit.jupiter.api.TestInfo;

import io.kroxylicious.testing.kafka.api.KafkaCluster;
import io.kroxylicious.testing.kafka.api.KafkaClusterProvisioningStrategy;
import io.kroxylicious.testing.kafka.common.KafkaClusterConfig;

/**
 * The in VM provisioning strategy.
 */
public class InVMProvisioningStrategy implements KafkaClusterProvisioningStrategy {

    /**
     * Instantiates a new In vm provisioning strategy.
     */
    public InVMProvisioningStrategy() {
    }

    @Override
    public boolean supportsAnnotation(Annotation constraint) {
        return KafkaClusterConfig.supportsConstraint(constraint.annotationType());
    }

    @Override
    public boolean supportsType(Class declarationType) {
        return declarationType.isAssignableFrom(InVMKafkaCluster.class);
    }

    @Override
    public KafkaCluster create(List constraints, Class declarationType, TestInfo testInfo) {
        KafkaClusterConfig config = KafkaClusterConfig.fromConstraints(constraints, testInfo);
        return new InVMKafkaCluster(config);
    }

    @Override
    public Duration estimatedProvisioningTimeMs(List constraints, Class declarationType) {
        return Duration.ofMillis(500);
    }
}