controller-server/src/test/java/com/yahoo/vespa/hosted/controller/restapi/ApplicationRequestToDiscFilterRequestWrapper.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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148

// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.controller.restapi;

import com.yahoo.application.container.handler.Request;
import com.yahoo.jdisc.http.HttpRequest;
import com.yahoo.jdisc.http.filter.DiscFilterRequest;

import java.net.URI;
import java.security.Principal;
import java.security.cert.X509Certificate;
import java.util.Collections;
import java.util.Enumeration;
import java.util.List;
import java.util.Map;

import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;

/**
 * Wraps an {@link Request} into a {@link DiscFilterRequest}. Only a few methods are supported.
 * Changes are not propagated; updated request instance must be retrieved through {@link #getUpdatedRequest()}.
 *
 * @author bjorncs
 */
public class ApplicationRequestToDiscFilterRequestWrapper extends DiscFilterRequest {

    private final Request request;
    private final List<X509Certificate> clientCertificateChain;
    private Principal userPrincipal;

    public ApplicationRequestToDiscFilterRequestWrapper(Request request) {
        this(request, Collections.emptyList());
    }

    public ApplicationRequestToDiscFilterRequestWrapper(Request request, List<X509Certificate> clientCertificateChain) {
        super(createDummyHttpRequest(request));
        this.request = request;
        this.userPrincipal = request.getUserPrincipal().orElse(null);
        this.clientCertificateChain = clientCertificateChain;
    }

    private static HttpRequest createDummyHttpRequest(Request req) {
        HttpRequest dummy = mock(HttpRequest.class, invocation -> { throw new UnsupportedOperationException(); });
        doReturn(URI.create(req.getUri()).normalize()).when(dummy).getUri();
        doNothing().when(dummy).copyHeaders(any());
        doReturn(Map.of()).when(dummy).parameters();
        return dummy;
    }

    public Request getUpdatedRequest() {
        Request updatedRequest = new Request(this.request.getUri(), this.request.getBody(), this.request.getMethod(), this.userPrincipal);
        this.request.getHeaders().forEach(updatedRequest.getHeaders()::put);
        updatedRequest.getAttributes().putAll(this.request.getAttributes());
        return updatedRequest;
    }

    @Override
    public String getMethod() {
        return request.getMethod().name();
    }

    @Override
    public String getParameter(String name) {
        throw new UnsupportedOperationException();
    }

    @Override
    public Enumeration<String> getParameterNames() {
        throw new UnsupportedOperationException();
    }

    @Override
    public void addHeader(String name, String value) {
        throw new UnsupportedOperationException();
    }

    @Override
    public String getHeader(String name) {
        return request.getHeaders().getFirst(name);
    }

    @Override
    public Enumeration<String> getHeaderNames() {
        throw new UnsupportedOperationException();
    }

    @Override
    public List<String> getHeaderNamesAsList() {
        throw new UnsupportedOperationException();
    }

    @Override
    public Enumeration<String> getHeaders(String name) {
        throw new UnsupportedOperationException();
    }

    @Override
    public List<String> getHeadersAsList(String name) {
        throw new UnsupportedOperationException();
    }

    @Override
    public void removeHeaders(String name) {
        throw new UnsupportedOperationException();
    }

    @Override
    public void setHeaders(String name, String value) {
        throw new UnsupportedOperationException();
    }

    @Override
    public void setHeaders(String name, List<String> values) {
        throw new UnsupportedOperationException();
    }

    @Override
    public Principal getUserPrincipal() {
        return this.userPrincipal;
    }

    @Override
    public void setUserPrincipal(Principal principal) {
        this.userPrincipal = principal;
    }

    @Override
    public List<X509Certificate> getClientCertificateChain() {
        return clientCertificateChain;
    }

    @Override
    public void clearCookies() {
        throw new UnsupportedOperationException();
    }

    @Override
    public Object getAttribute(String name) {
        return request.getAttributes().get(name);
    }

    @Override
    public void setAttribute(String name, Object value) {
        request.getAttributes().put(name, value);
    }
}