org.gradle.internal.resource.transport.http.AbstractHttpClientTest.groovy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gradle-api Show documentation

Gradle 6.9.1 API redistribution.

There is a newer version: 8.11.1

/*
 * Copyright 2016 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.internal.resource.transport.http

import org.apache.http.HttpEntity
import org.apache.http.ProtocolVersion
import org.apache.http.client.methods.CloseableHttpResponse
import org.apache.http.message.BasicStatusLine
import spock.lang.Specification

abstract class AbstractHttpClientTest extends Specification {

    MockedHttpResponse mockedHttpResponse() {
        CloseableHttpResponse response = Mock()
        mockedHttpResponse(response)
    }

    MockedHttpResponse mockedHttpResponse(CloseableHttpResponse response) {
        HttpEntity entity = Mock()
        InputStream content = Mock()

        new MockedHttpResponse(entity: entity, content: content, response: response)
    }

    static class MockedHttpResponse {
        HttpEntity entity
        InputStream content
        CloseableHttpResponse response
    }

    void assertIsClosedCorrectly(MockedHttpResponse mockedHttpResponse) {
        def response = mockedHttpResponse.response
        def entity = mockedHttpResponse.entity
        def content = mockedHttpResponse.content
        _ * response.getStatusLine() >> new BasicStatusLine(new ProtocolVersion("HTTP", 1, 1), 400, "I'm broken")
        1 * response.close()
        1 * response.getEntity() >> entity
        1 * entity.isStreaming() >> true
        1 * entity.content >> content
        1 * content.close()
    }
}