top.infra.test.contract.ContractRule Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-boot-test-support Show documentation
The newest version!
package top.infra.test.contract;

import org.junit.rules.TestRule;
import org.junit.runner.Description;
import org.junit.runners.model.MultipleFailureException;
import org.junit.runners.model.Statement;

import java.util.ArrayList;
import java.util.List;

public class ContractRule implements TestRule {

    private static final String EUREKA_CLIENT_ENABLED = "eureka.client.enabled";
    private static final String SPRING_CLOUD_CONFIG_DISCOVERY_ENABLED = "spring.cloud.config.discovery.enabled";
    private static final String SPRING_CLOUD_CONFIG_FAILFAST = "spring.cloud.config.failFast";
    private String eurekaClientEnabled = null;
    private String springCloudConfigDiscoveryEnabled = null;
    private String springCloudConfigFailFast = null;

    @Override
    public Statement apply(final Statement base, final Description description) {
        return new Statement() {

            @Override
            public void evaluate() throws Throwable {

                List errors = new ArrayList<>();

                try {
                    starting(description);
                    base.evaluate();
                    succeeded(description);
                } catch (Throwable e) {
                    errors.add(e);
                    failed(e, description);
                } finally {
                    finished(description);
                }

                MultipleFailureException.assertEmpty(errors);
            }
        };
    }

    protected void starting(Description description) {
        this.backupProperties();

        this.setProperties();
    }

    protected void succeeded(Description description) {
    }

    protected void failed(Throwable e, Description description) {
    }

    protected void finished(Description description) {
        this.restoreProperties();
    }

    private void backupProperties() {
        this.eurekaClientEnabled = System.getProperty(EUREKA_CLIENT_ENABLED);
        this.springCloudConfigDiscoveryEnabled = System.getProperty(SPRING_CLOUD_CONFIG_DISCOVERY_ENABLED);
        this.springCloudConfigFailFast = System.getProperty(SPRING_CLOUD_CONFIG_FAILFAST);
    }

    private void restoreProperties() {
        if (this.eurekaClientEnabled != null) {
            System.setProperty(EUREKA_CLIENT_ENABLED, this.eurekaClientEnabled);
        } else {
            System.getProperties().remove(EUREKA_CLIENT_ENABLED);
        }

        if (this.springCloudConfigDiscoveryEnabled != null) {
            System.setProperty(SPRING_CLOUD_CONFIG_DISCOVERY_ENABLED, this.springCloudConfigDiscoveryEnabled);
        } else {
            System.getProperties().remove(SPRING_CLOUD_CONFIG_DISCOVERY_ENABLED);
        }

        if (this.springCloudConfigFailFast != null) {
            System.setProperty(SPRING_CLOUD_CONFIG_FAILFAST, this.springCloudConfigFailFast);
        } else {
            System.getProperties().remove(SPRING_CLOUD_CONFIG_FAILFAST);
        }
    }

    private void setProperties() {
        System.setProperty(EUREKA_CLIENT_ENABLED, "false");
        System.setProperty(SPRING_CLOUD_CONFIG_DISCOVERY_ENABLED, "false");
        System.setProperty(SPRING_CLOUD_CONFIG_FAILFAST, "false");
    }
}