/*
 * Copyright 2014 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.gradle.integtests.tooling.fixture

import org.gradle.internal.UncheckedException
import org.gradle.tooling.GradleConnectionException
import org.gradle.tooling.ResultHandler
import org.gradle.tooling.internal.consumer.BlockingResultHandler

import java.util.concurrent.CountDownLatch
import java.util.concurrent.TimeUnit

class TestResultHandler implements ResultHandler<Object> {
    final latch = new CountDownLatch(1)
    GradleConnectionException failure

    void onComplete(Object result) {
        latch.countDown()
    }

    void onFailure(GradleConnectionException failure) {
        this.failure = failure
        latch.countDown()
    }

    def finished() {
        finished(20)
    }

    def finished(int seconds) {
        if (!latch.await(seconds, TimeUnit.SECONDS)) {
            throw new AssertionError("Timeout waiting for operation to complete.")
        }
    }

    def failWithFailure() {
        if (failure) {
            throw UncheckedException.throwAsUncheckedException(BlockingResultHandler.attachCallerThreadStackTrace(failure))
        }
    }

    def assertNoFailure() {
        if (failure == null) {
            return true
        } else {
            throw failure
        }
    }

    def assertFailedWith(Class<? extends GradleConnectionException> exceptionType) {
        if (failure?.getClass() == exceptionType) {
            return true
        }
        if (failure) {
            throw failure
        }
        false
    }
}
