test/jdk/com/sun/net/httpserver/simpleserver/RequestTest.java

/*
 * Copyright (c) 2021, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

/*
 * @test
 * @summary Tests for Request
 * @run testng RequestTest
 */

import java.io.InputStream;
import java.io.OutputStream;
import java.net.InetSocketAddress;
import java.net.URI;
import java.util.AbstractMap;
import java.util.List;
import java.util.Map;
import com.sun.net.httpserver.*;
import org.testng.annotations.Test;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertThrows;

public class RequestTest {

    @Test
    public void testAddToEmpty() {
        var headers = new Headers();
        Request request = new TestHttpExchange(headers);
        request = request.with("Foo", List.of("Bar"));
        assertEquals(request.getRequestHeaders().size(), 1);
        assertEquals(request.getRequestHeaders().get("Foo"), List.of("Bar"));
        assertReadOnly(request.getRequestHeaders());
    }

    @Test
    public void testAddition() {
        var headers = new Headers();
        headers.add("Foo", "Bar");
        Request request = new TestHttpExchange(headers);
        request = request.with("X-Foo", List.of("Bar"));
        assertEquals(request.getRequestHeaders().size(), 2);
        assertEquals(request.getRequestHeaders().get("Foo"), List.of("Bar"));
        assertEquals(request.getRequestHeaders().get("X-Foo"), List.of("Bar"));
        assertReadOnly(request.getRequestHeaders());
    }

    @Test
    public void testAddWithExisting() {
        final String headerName = "Foo";
        var headers = new Headers();
        headers.add(headerName, "Bar");
        Request request = new TestHttpExchange(headers);
        request = request.with(headerName, List.of("blahblahblah"));
        assertEquals(request.getRequestHeaders().size(), 1);
        assertEquals(request.getRequestHeaders().get(headerName), List.of("Bar"));
        assertReadOnly(request.getRequestHeaders());
    }

    @Test
    public void testAddSeveral() {
        var headers = new Headers();
        headers.add("Foo", "Bar");
        Request request = new TestHttpExchange(headers);
        request = request.with("Larry", List.of("a"))
                         .with("Curly", List.of("b"))
                         .with("Moe",   List.of("c"));
        assertEquals(request.getRequestHeaders().size(), 4);
        assertEquals(request.getRequestHeaders().getFirst("Foo"), "Bar");
        assertEquals(request.getRequestHeaders().getFirst("Larry"), "a");
        assertEquals(request.getRequestHeaders().getFirst("Curly"), "b");
        assertEquals(request.getRequestHeaders().getFirst("Moe"  ), "c");
        assertReadOnly(request.getRequestHeaders());
    }

    static final Class<UnsupportedOperationException> UOP = UnsupportedOperationException.class;

    static void assertReadOnly(Headers headers) {
        assertUnsupportedOperation(headers);
        assertUnmodifiableCollection(headers);
        assertUnmodifiableList(headers);
    }

    static void assertUnsupportedOperation(Headers headers) {
        assertThrows(UOP, () -> headers.add("a", "b"));
        assertThrows(UOP, () -> headers.compute("c", (k, v) -> List.of("c")));
        assertThrows(UOP, () -> headers.computeIfAbsent("d", k -> List.of("d")));
        assertThrows(UOP, () -> headers.computeIfPresent("Foo", (k, v) -> null));
        assertThrows(UOP, () -> headers.merge("e", List.of("e"), (k, v) -> List.of("e")));
        assertThrows(UOP, () -> headers.put("f", List.of("f")));
        assertThrows(UOP, () -> headers.putAll(Map.of()));
        assertThrows(UOP, () -> headers.putIfAbsent("g", List.of("g")));
        assertThrows(UOP, () -> headers.remove("h"));
        assertThrows(UOP, () -> headers.replace("i", List.of("i")));
        assertThrows(UOP, () -> headers.replace("j", List.of("j"), List.of("j")));
        assertThrows(UOP, () -> headers.replaceAll((k, v) -> List.of("k")));
        assertThrows(UOP, () -> headers.set("l", "m"));
        assertThrows(UOP, () -> headers.clear());
    }

    static void assertUnmodifiableCollection(Headers headers) {
        var entry = new AbstractMap.SimpleEntry<>("n", List.of("n"));

        assertThrows(UOP, () -> headers.values().remove(List.of("Bar")));
        assertThrows(UOP, () -> headers.values().removeAll(List.of("Bar")));
        assertThrows(UOP, () -> headers.keySet().remove("Foo"));
        assertThrows(UOP, () -> headers.keySet().removeAll(List.of("Foo")));
        assertThrows(UOP, () -> headers.entrySet().remove(entry));
        assertThrows(UOP, () -> headers.entrySet().removeAll(List.of(entry)));
    }

    static void assertUnmodifiableList(Headers headers) {
        assertThrows(UOP, () -> headers.get("Foo").remove(0));
        assertThrows(UOP, () -> headers.get("foo").remove(0));
        assertThrows(UOP, () -> headers.values().stream().findFirst().orElseThrow().remove(0));
        assertThrows(UOP, () -> headers.entrySet().stream().findFirst().orElseThrow().getValue().remove(0));
    }

    static class TestHttpExchange extends StubHttpExchange {
        final Headers headers;
        TestHttpExchange(Headers headers) {
            this.headers = headers;
        }
        @Override
        public Headers getRequestHeaders() {
            return headers;
        }
    }

    static class StubHttpExchange extends HttpExchange {
        @Override public Headers getRequestHeaders() { return null; }
        @Override public Headers getResponseHeaders() { return null; }
        @Override public URI getRequestURI() { return null; }
        @Override public String getRequestMethod() { return null; }
        @Override public HttpContext getHttpContext() { return null; }
        @Override public void close() { }
        @Override public InputStream getRequestBody() { return null; }
        @Override public OutputStream getResponseBody() { return null; }
        @Override public void sendResponseHeaders(int rCode, long responseLength) { }
        @Override public InetSocketAddress getRemoteAddress() { return null; }
        @Override public int getResponseCode() { return 0; }
        @Override public InetSocketAddress getLocalAddress() { return null; }
        @Override public String getProtocol() { return null; }
        @Override public Object getAttribute(String name) { return null; }
        @Override public void setAttribute(String name, Object value) { }
        @Override public void setStreams(InputStream i, OutputStream o) { }
        @Override public HttpPrincipal getPrincipal() { return null; }
    }
}