Search in sources :

Example 26 with Flux

use of reactor.core.publisher.Flux in project spring-framework by spring-projects.

the class ServerSentEventHttpMessageReaderTests method readString.

@Test
public void readString() {
    String body = "data:foo\ndata:bar\n\ndata:baz\n\n";
    MockServerHttpRequest request = MockServerHttpRequest.post("/").body(body);
    Flux<String> data = messageReader.read(ResolvableType.forClass(String.class), request, Collections.emptyMap()).cast(String.class);
    StepVerifier.create(data).expectNextMatches(elem -> elem.equals("foo\nbar")).expectNextMatches(elem -> elem.equals("baz")).expectComplete().verify();
}
Also used : Jackson2JsonDecoder(org.springframework.http.codec.json.Jackson2JsonDecoder) StepVerifier(reactor.test.StepVerifier) AbstractDataBufferAllocatingTestCase(org.springframework.core.io.buffer.AbstractDataBufferAllocatingTestCase) MediaType(org.springframework.http.MediaType) Assert.assertTrue(org.junit.Assert.assertTrue) Test(org.junit.Test) Mono(reactor.core.publisher.Mono) Flux(reactor.core.publisher.Flux) MockServerHttpRequest(org.springframework.mock.http.server.reactive.test.MockServerHttpRequest) Assert.assertFalse(org.junit.Assert.assertFalse) Duration(java.time.Duration) ResolvableType(org.springframework.core.ResolvableType) Collections(java.util.Collections) Assert.assertEquals(org.junit.Assert.assertEquals) MockServerHttpRequest(org.springframework.mock.http.server.reactive.test.MockServerHttpRequest) Test(org.junit.Test)

Example 27 with Flux

use of reactor.core.publisher.Flux in project spring-framework by spring-projects.

the class ServerSentEventHttpMessageReaderTests method readServerSentEventsWithMultipleChunks.

@Test
public void readServerSentEventsWithMultipleChunks() {
    MockServerHttpRequest request = MockServerHttpRequest.post("/").body(Flux.just(stringBuffer("id:c42\nev"), stringBuffer("ent:foo\nretry:123\n:bla\n:bla bla\n:bla bla bla\ndata:"), stringBuffer("bar\n\nid:c43\nevent:bar\nretry:456\ndata:baz\n\n")));
    Flux<ServerSentEvent> events = messageReader.read(ResolvableType.forClassWithGenerics(ServerSentEvent.class, String.class), request, Collections.emptyMap()).cast(ServerSentEvent.class);
    StepVerifier.create(events).consumeNextWith(event -> {
        assertEquals("c42", event.id().get());
        assertEquals("foo", event.event().get());
        assertEquals(Duration.ofMillis(123), event.retry().get());
        assertEquals("bla\nbla bla\nbla bla bla", event.comment().get());
        assertEquals("bar", event.data().get());
    }).consumeNextWith(event -> {
        assertEquals("c43", event.id().get());
        assertEquals("bar", event.event().get());
        assertEquals(Duration.ofMillis(456), event.retry().get());
        assertFalse(event.comment().isPresent());
        assertEquals("baz", event.data().get());
    }).expectComplete().verify();
}
Also used : Jackson2JsonDecoder(org.springframework.http.codec.json.Jackson2JsonDecoder) StepVerifier(reactor.test.StepVerifier) AbstractDataBufferAllocatingTestCase(org.springframework.core.io.buffer.AbstractDataBufferAllocatingTestCase) MediaType(org.springframework.http.MediaType) Assert.assertTrue(org.junit.Assert.assertTrue) Test(org.junit.Test) Mono(reactor.core.publisher.Mono) Flux(reactor.core.publisher.Flux) MockServerHttpRequest(org.springframework.mock.http.server.reactive.test.MockServerHttpRequest) Assert.assertFalse(org.junit.Assert.assertFalse) Duration(java.time.Duration) ResolvableType(org.springframework.core.ResolvableType) Collections(java.util.Collections) Assert.assertEquals(org.junit.Assert.assertEquals) MockServerHttpRequest(org.springframework.mock.http.server.reactive.test.MockServerHttpRequest) Test(org.junit.Test)

Example 28 with Flux

use of reactor.core.publisher.Flux in project spring-framework by spring-projects.

the class ServerSentEventHttpMessageReaderTests method readPojo.

@Test
public void readPojo() {
    MockServerHttpRequest request = MockServerHttpRequest.post("/").body("data:{\"foo\": \"foofoo\", \"bar\": \"barbar\"}\n\n" + "data:{\"foo\": \"foofoofoo\", \"bar\": \"barbarbar\"}\n\n");
    Flux<Pojo> data = messageReader.read(ResolvableType.forClass(Pojo.class), request, Collections.emptyMap()).cast(Pojo.class);
    StepVerifier.create(data).consumeNextWith(pojo -> {
        assertEquals("foofoo", pojo.getFoo());
        assertEquals("barbar", pojo.getBar());
    }).consumeNextWith(pojo -> {
        assertEquals("foofoofoo", pojo.getFoo());
        assertEquals("barbarbar", pojo.getBar());
    }).expectComplete().verify();
}
Also used : Jackson2JsonDecoder(org.springframework.http.codec.json.Jackson2JsonDecoder) StepVerifier(reactor.test.StepVerifier) AbstractDataBufferAllocatingTestCase(org.springframework.core.io.buffer.AbstractDataBufferAllocatingTestCase) MediaType(org.springframework.http.MediaType) Assert.assertTrue(org.junit.Assert.assertTrue) Test(org.junit.Test) Mono(reactor.core.publisher.Mono) Flux(reactor.core.publisher.Flux) MockServerHttpRequest(org.springframework.mock.http.server.reactive.test.MockServerHttpRequest) Assert.assertFalse(org.junit.Assert.assertFalse) Duration(java.time.Duration) ResolvableType(org.springframework.core.ResolvableType) Collections(java.util.Collections) Assert.assertEquals(org.junit.Assert.assertEquals) MockServerHttpRequest(org.springframework.mock.http.server.reactive.test.MockServerHttpRequest) Test(org.junit.Test)

Aggregations

Flux (reactor.core.publisher.Flux)28 Test (org.junit.Test)21 ResolvableType (org.springframework.core.ResolvableType)15 Collections (java.util.Collections)11 DataBuffer (org.springframework.core.io.buffer.DataBuffer)11 MediaType (org.springframework.http.MediaType)11 Mono (reactor.core.publisher.Mono)11 StepVerifier (reactor.test.StepVerifier)9 MockServerHttpRequest (org.springframework.mock.http.server.reactive.test.MockServerHttpRequest)8 Duration (java.time.Duration)7 List (java.util.List)7 Assert.assertFalse (org.junit.Assert.assertFalse)7 DefaultDataBufferFactory (org.springframework.core.io.buffer.DefaultDataBufferFactory)7 Assert (org.springframework.util.Assert)7 Map (java.util.Map)6 Assert.assertTrue (org.junit.Assert.assertTrue)6 AbstractDataBufferAllocatingTestCase (org.springframework.core.io.buffer.AbstractDataBufferAllocatingTestCase)6 DefaultDataBuffer (org.springframework.core.io.buffer.DefaultDataBuffer)6 Assert.assertEquals (org.junit.Assert.assertEquals)5 ReactiveHttpInputMessage (org.springframework.http.ReactiveHttpInputMessage)5