import java.lang.annotation.Annotation;
import java.util.concurrent.TimeUnit;

public class TimeoutAdapter implements Timeout {

    private int timeout;
    private TimeUnit unit;

    TimeoutAdapter(IntegrationTestTimeout timeout) {
        this.timeout = timeout.value();
        this.unit = timeout.unit();
    }

    TimeoutAdapter(int timeout, TimeUnit unit) {
        this.timeout = timeout;
        this.unit = unit;
    }

@CompileStatic
class IntegrationTestTimeoutInterceptor extends TimeoutInterceptor {
    IntegrationTestTimeoutInterceptor(IntegrationTestTimeout timeout) {
        super(new TimeoutAdapter(timeout))
    }

    IntegrationTestTimeoutInterceptor(int timeoutSeconds) {
        super(new TimeoutAdapter(timeoutSeconds, TimeUnit.SECONDS))
    }

    @Override
    void intercept(final IMethodInvocation invocation) throws Throwable {
        try {