http-utils/src/test/java/ai/vespa/util/http/VespaHttpClientBuilderTest.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

// Copyright 2019 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package ai.vespa.util.http;

import ai.vespa.util.http.VespaHttpClientBuilder.HttpToHttpsRewritingRequestInterceptor;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.protocol.BasicHttpContext;
import org.junit.Test;

import java.net.URI;

import static org.assertj.core.api.AssertionsForClassTypes.assertThat;

/**
 * @author bjorncs
 */
public class VespaHttpClientBuilderTest {

    @Test
    public void request_interceptor_modifies_scheme_of_requests() {
        verifyProcessedUriMatchesExpectedOutput("http://dummyhostname:8080/a/path/to/resource?query=value",
                                                "https://dummyhostname:8080/a/path/to/resource?query=value");
    }

    @Test
    public void request_interceptor_add_handles_implicit_http_port() {
        verifyProcessedUriMatchesExpectedOutput("http://dummyhostname/a/path/to/resource?query=value",
                                                "https://dummyhostname:80/a/path/to/resource?query=value");
    }

    private static void verifyProcessedUriMatchesExpectedOutput(String inputUri, String expectedOutputUri) {
        var interceptor = new HttpToHttpsRewritingRequestInterceptor();
        HttpGet request = new HttpGet(inputUri);
        interceptor.process(request, new BasicHttpContext());
        URI modifiedUri = request.getURI();
        URI expectedUri = URI.create(expectedOutputUri);
        assertThat(modifiedUri).isEqualTo(expectedUri);
    }

}