1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 package org.springframework.web.socket;
18
19 import org.junit.Before;
20
21 import org.springframework.http.HttpHeaders;
22 import org.springframework.http.server.ServerHttpAsyncRequestControl;
23 import org.springframework.http.server.ServerHttpRequest;
24 import org.springframework.http.server.ServerHttpResponse;
25 import org.springframework.http.server.ServletServerHttpRequest;
26 import org.springframework.http.server.ServletServerHttpResponse;
27 import org.springframework.mock.web.test.MockHttpServletRequest;
28 import org.springframework.mock.web.test.MockHttpServletResponse;
29
30
31
32
33
34
35 public abstract class AbstractHttpRequestTests {
36
37 protected ServerHttpRequest request;
38
39 protected ServerHttpResponse response;
40
41 protected MockHttpServletRequest servletRequest;
42
43 protected MockHttpServletResponse servletResponse;
44
45 protected ServerHttpAsyncRequestControl asyncControl;
46
47
48 @Before
49 public void setUp() {
50 resetRequestAndResponse();
51 }
52
53 protected void setRequest(String method, String requestUri) {
54 this.servletRequest.setMethod(method);
55 this.servletRequest.setRequestURI(requestUri);
56 }
57
58 protected void setOrigin(String origin) {
59 this.request.getHeaders().add(HttpHeaders.ORIGIN, origin);
60 }
61
62 protected void resetRequestAndResponse() {
63 resetRequest();
64 resetResponse();
65 this.asyncControl = this.request.getAsyncRequestControl(this.response);
66 }
67
68 protected void resetRequest() {
69 this.servletRequest = new MockHttpServletRequest();
70 this.servletRequest.setAsyncSupported(true);
71 this.request = new ServletServerHttpRequest(this.servletRequest);
72 }
73
74 protected void resetResponse() {
75 this.servletResponse = new MockHttpServletResponse();
76 this.response = new ServletServerHttpResponse(this.servletResponse);
77 }
78
79 }