build.forTasks('hang')
            build.withCancellationToken(cancel.token())
            build.run(resultHandler)
            resultHandler.finished()
        }
        then:
        resultHandler.assertFailedWith(BuildCancelledException)
    }

    def "can cancel build after completion"() {
        buildFile << """
task thing
"""
        def cancel = GradleConnector.newCancellationTokenSource()

        build.run(resultHandler)
        sync.waitForAllPendingCalls(resultHandler)
        connector.disconnect()
        resultHandler.finished()

        then:
        resultHandler.assertFailedWith(BuildCancelledException)
    }

    def "disconnect before build starts"() {
        when:
        def connector = toolingApi.connector()
        connector.connect()
        connector.disconnect()

latch.await()
"""

    }

    void buildWasCancelled(TestResultHandler resultHandler, String failureMessage = 'Could not execute build using') {
        resultHandler.assertFailedWith(BuildCancelledException)
        assert resultHandler.failure.message.startsWith(failureMessage)

        if (targetIsGradle51OrLater()) {
            verifyBuildCancelledExceptionMessage(resultHandler)
        }

        }
    }

    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
        }

            build.run(resultHandler)
            sync.waitForAllPendingCalls(resultHandler)
            cancel.cancel()
            resultHandler.finished()
        }

        then:
        resultHandler.assertFailedWith(BuildCancelledException)
        resultHandler.failure.message.startsWith("Could not execute build using")
        if (targetDist.toolingApiHasCauseOnForcedCancel) {

            downloadHandle.waitUntilBlocked()
            tokenSource.cancel()
            handler.finished()
            downloadHandle.release()
        }

        then:
        handler.assertFailedWith(BuildCancelledException)
        handler.failure.message.contains('Distribution download cancelled.')

        and:
        events.buildOperations.size() == 1