Search in sources :

Example 11 with Sink

use of org.springframework.cloud.stream.messaging.Sink in project spring-cloud-stream by spring-cloud.

the class AvroStubSchemaRegistryClientMessageConverterTests method testSendMessage.

@Test
public void testSendMessage() throws Exception {
    ConfigurableApplicationContext sourceContext = SpringApplication.run(AvroSourceApplication.class, "--server.port=0", "--debug", "--spring.jmx.enabled=false", "--spring.cloud.stream.bindings.output.contentType=application/*+avro", "--spring.cloud.stream.schema.avro.dynamicSchemaGenerationEnabled=true");
    Source source = sourceContext.getBean(Source.class);
    User1 firstOutboundFoo = new User1();
    firstOutboundFoo.setFavoriteColor("foo" + UUID.randomUUID().toString());
    firstOutboundFoo.setName("foo" + UUID.randomUUID().toString());
    source.output().send(MessageBuilder.withPayload(firstOutboundFoo).build());
    MessageCollector sourceMessageCollector = sourceContext.getBean(MessageCollector.class);
    Message<?> outboundMessage = sourceMessageCollector.forChannel(source.output()).poll(1000, TimeUnit.MILLISECONDS);
    ConfigurableApplicationContext barSourceContext = SpringApplication.run(AvroSourceApplication.class, "--server.port=0", "--spring.jmx.enabled=false", "--spring.cloud.stream.bindings.output.contentType=application/vnd.user1.v1+avro", "--spring.cloud.stream.schema.avro.dynamicSchemaGenerationEnabled=true");
    Source barSource = barSourceContext.getBean(Source.class);
    User2 firstOutboundUser2 = new User2();
    firstOutboundUser2.setFavoriteColor("foo" + UUID.randomUUID().toString());
    firstOutboundUser2.setName("foo" + UUID.randomUUID().toString());
    barSource.output().send(MessageBuilder.withPayload(firstOutboundUser2).build());
    MessageCollector barSourceMessageCollector = barSourceContext.getBean(MessageCollector.class);
    Message<?> barOutboundMessage = barSourceMessageCollector.forChannel(barSource.output()).poll(1000, TimeUnit.MILLISECONDS);
    assertThat(barOutboundMessage).isNotNull();
    User2 secondBarOutboundPojo = new User2();
    secondBarOutboundPojo.setFavoriteColor("foo" + UUID.randomUUID().toString());
    secondBarOutboundPojo.setName("foo" + UUID.randomUUID().toString());
    source.output().send(MessageBuilder.withPayload(secondBarOutboundPojo).build());
    Message<?> secondBarOutboundMessage = sourceMessageCollector.forChannel(source.output()).poll(1000, TimeUnit.MILLISECONDS);
    ConfigurableApplicationContext sinkContext = SpringApplication.run(AvroSinkApplication.class, "--server.port=0", "--spring.jmx.enabled=false");
    Sink sink = sinkContext.getBean(Sink.class);
    sink.input().send(outboundMessage);
    sink.input().send(barOutboundMessage);
    sink.input().send(secondBarOutboundMessage);
    List<User2> receivedPojos = sinkContext.getBean(AvroSinkApplication.class).receivedPojos;
    assertThat(receivedPojos).hasSize(3);
    assertThat(receivedPojos.get(0)).isNotSameAs(firstOutboundFoo);
    assertThat(receivedPojos.get(0).getFavoriteColor()).isEqualTo(firstOutboundFoo.getFavoriteColor());
    assertThat(receivedPojos.get(0).getName()).isEqualTo(firstOutboundFoo.getName());
    assertThat(receivedPojos.get(0).getFavoritePlace()).isEqualTo("NYC");
    assertThat(receivedPojos.get(1)).isNotSameAs(firstOutboundUser2);
    assertThat(receivedPojos.get(1).getFavoriteColor()).isEqualTo(firstOutboundUser2.getFavoriteColor());
    assertThat(receivedPojos.get(1).getName()).isEqualTo(firstOutboundUser2.getName());
    assertThat(receivedPojos.get(1).getFavoritePlace()).isEqualTo("Boston");
    assertThat(receivedPojos.get(2)).isNotSameAs(secondBarOutboundPojo);
    assertThat(receivedPojos.get(2).getFavoriteColor()).isEqualTo(secondBarOutboundPojo.getFavoriteColor());
    assertThat(receivedPojos.get(2).getName()).isEqualTo(secondBarOutboundPojo.getName());
    assertThat(receivedPojos.get(2).getFavoritePlace()).isEqualTo(secondBarOutboundPojo.getFavoritePlace());
    sourceContext.close();
}
Also used : ConfigurableApplicationContext(org.springframework.context.ConfigurableApplicationContext) Sink(org.springframework.cloud.stream.messaging.Sink) MessageCollector(org.springframework.cloud.stream.test.binder.MessageCollector) Source(org.springframework.cloud.stream.messaging.Source) Test(org.junit.Test)

Example 12 with Sink

use of org.springframework.cloud.stream.messaging.Sink in project spring-cloud-stream by spring-cloud.

the class StreamListenerAsMetaAnnotationTests method testCustomAnnotation.

@Test
public void testCustomAnnotation() {
    ConfigurableApplicationContext context = SpringApplication.run(TestPojoWithCustomAnnotatedArguments.class, "--server.port=0");
    TestPojoWithCustomAnnotatedArguments testPojoWithAnnotatedArguments = context.getBean(TestPojoWithCustomAnnotatedArguments.class);
    Sink sink = context.getBean(Sink.class);
    String id = UUID.randomUUID().toString();
    sink.input().send(MessageBuilder.withPayload("{\"foo\":\"barbar" + id + "\"}").setHeader("contentType", "application/json").setHeader("testHeader", "testValue").setHeader("type", "foo").build());
    assertThat(testPojoWithAnnotatedArguments.receivedFoo).hasSize(1);
    assertThat(testPojoWithAnnotatedArguments.receivedFoo.get(0)).hasFieldOrPropertyWithValue("foo", "barbar" + id);
    context.close();
}
Also used : ConfigurableApplicationContext(org.springframework.context.ConfigurableApplicationContext) Sink(org.springframework.cloud.stream.messaging.Sink) Test(org.junit.Test)

Aggregations

Sink (org.springframework.cloud.stream.messaging.Sink)12 ConfigurableApplicationContext (org.springframework.context.ConfigurableApplicationContext)12 Test (org.junit.Test)11 MessageCollector (org.springframework.cloud.stream.test.binder.MessageCollector)5 Source (org.springframework.cloud.stream.messaging.Source)4 Map (java.util.Map)1 DirectChannel (org.springframework.integration.channel.DirectChannel)1 Message (org.springframework.messaging.Message)1 MessageChannel (org.springframework.messaging.MessageChannel)1