Spring/spring-framework
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Reliable content length check in ResourceHandlerFunctionTests

Browse Source
This commit is contained in:
Juergen Hoeller
2016-12-12 09:57:57 +01:00
parent f7ea469897
commit 66e6b35e9f
1 changed files with 6 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
spring-web-reactive/src/test/java/org/springframework/web/reactive/function/ResourceHandlerFunctionTests.java
View File

@@ -20,7 +20,6 @@ import java.io.IOException;
import java.nio.file.Files; import java.nio.file.Files;
import java.util.EnumSet; import java.util.EnumSet;
import org.junit.Before;
import org.junit.Test; import org.junit.Test;
import reactor.core.publisher.Mono; import reactor.core.publisher.Mono;
import reactor.test.StepVerifier; import reactor.test.StepVerifier;
@@ -36,24 +35,17 @@ import org.springframework.web.server.ServerWebExchange;
import org.springframework.web.server.adapter.DefaultServerWebExchange; import org.springframework.web.server.adapter.DefaultServerWebExchange;
import org.springframework.web.server.session.MockWebSessionManager; import org.springframework.web.server.session.MockWebSessionManager;
import static org.junit.Assert.assertArrayEquals; import static org.junit.Assert.*;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
/** /**
* @author Arjen Poutsma * @author Arjen Poutsma
*/ */
public class ResourceHandlerFunctionTests { public class ResourceHandlerFunctionTests {
private Resource resource; private final Resource resource = new ClassPathResource("response.txt", getClass());
private ResourceHandlerFunction handlerFunction; private final ResourceHandlerFunction handlerFunction = new ResourceHandlerFunction(this.resource);
@Before
public void createResource() {
this.resource = new ClassPathResource("response.txt", getClass());
this.handlerFunction = new ResourceHandlerFunction(this.resource);
}
@Test @Test
public void get() throws IOException { public void get() throws IOException {
@@ -88,7 +80,7 @@ public class ResourceHandlerFunctionTests {
.expectComplete() .expectComplete()
.verify(); .verify();
assertEquals(MediaType.TEXT_PLAIN, mockResponse.getHeaders().getContentType()); assertEquals(MediaType.TEXT_PLAIN, mockResponse.getHeaders().getContentType());
assertEquals(49, mockResponse.getHeaders().getContentLength()); assertEquals(this.resource.contentLength(), mockResponse.getHeaders().getContentLength());
} }
@Test @Test
@@ -116,7 +108,7 @@ public class ResourceHandlerFunctionTests {
.expectComplete() .expectComplete()
.verify(); .verify();
assertEquals(MediaType.TEXT_PLAIN, mockResponse.getHeaders().getContentType()); assertEquals(MediaType.TEXT_PLAIN, mockResponse.getHeaders().getContentType());
assertEquals(49, mockResponse.getHeaders().getContentLength()); assertEquals(this.resource.contentLength(), mockResponse.getHeaders().getContentLength());
} }
@Test @Test
@@ -148,4 +140,4 @@ public class ResourceHandlerFunctionTests {
assertNull(mockResponse.getBody()); assertNull(mockResponse.getBody());
} }
} }