Examples with MessageCollector org.springframework.cloud.stream.test.binder.MessageCollector used on opensource projects

Example 11 with MessageCollector

use of org.springframework.cloud.stream.test.binder.MessageCollector in project spring-cloud-stream by spring-cloud.

the class AvroSchemaMessageConverterTests method testSendMessageWithoutLocation.

@Test
public void testSendMessageWithoutLocation() throws Exception {
    ConfigurableApplicationContext sourceContext = SpringApplication.run(AvroSourceApplication.class, "--server.port=0", "--spring.jmx.enabled=false", "--spring.cloud.stream.schemaRegistryClient.enabled=false", "--spring.cloud.stream.bindings.output.contentType=avro/bytes");
    Source source = sourceContext.getBean(Source.class);
    User1 firstOutboundFoo = new User1();
    firstOutboundFoo.setName("foo" + UUID.randomUUID().toString());
    firstOutboundFoo.setFavoriteColor("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.schemaRegistryClient.enabled=false", "--spring.cloud.stream.bindings.output.contentType=avro/bytes");
    Source barSource = barSourceContext.getBean(Source.class);
    User2 firstOutboundUser2 = new User2();
    firstOutboundUser2.setFavoriteColor("foo" + UUID.randomUUID().toString());
    firstOutboundUser2.setFavoritePlace("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 secondUser2OutboundPojo = new User2();
    secondUser2OutboundPojo.setFavoriteColor("foo" + UUID.randomUUID().toString());
    secondUser2OutboundPojo.setFavoritePlace("foo" + UUID.randomUUID().toString());
    secondUser2OutboundPojo.setName("foo" + UUID.randomUUID().toString());
    source.output().send(MessageBuilder.withPayload(secondUser2OutboundPojo).build());
    Message<?> secondBarOutboundMessage = sourceMessageCollector.forChannel(source.output()).poll(1000, TimeUnit.MILLISECONDS);
    ConfigurableApplicationContext sinkContext = SpringApplication.run(AvroSinkApplication.class, "--server.port=0", "--spring.jmx.enabled=false", "--spring.cloud.stream.schemaRegistryClient.enabled=false");
    Sink sink = sinkContext.getBean(Sink.class);
    sink.input().send(outboundMessage);
    sink.input().send(barOutboundMessage);
    sink.input().send(secondBarOutboundMessage);
    List<User1> receivedUsers = sinkContext.getBean(AvroSinkApplication.class).receivedUsers;
    assertThat(receivedUsers).hasSize(3);
    assertThat(receivedUsers.get(0)).isNotSameAs(firstOutboundFoo);
    assertThat(receivedUsers.get(0).getFavoriteColor()).isEqualTo(firstOutboundFoo.getFavoriteColor());
    assertThat(receivedUsers.get(0).getName()).isEqualTo(firstOutboundFoo.getName());
    assertThat(receivedUsers.get(1)).isNotSameAs(firstOutboundUser2);
    assertThat(receivedUsers.get(1).getFavoriteColor()).isEqualTo(firstOutboundUser2.getFavoriteColor());
    assertThat(receivedUsers.get(1).getName()).isEqualTo(firstOutboundUser2.getName());
    assertThat(receivedUsers.get(2)).isNotSameAs(secondUser2OutboundPojo);
    assertThat(receivedUsers.get(2).getFavoriteColor()).isEqualTo(secondUser2OutboundPojo.getFavoriteColor());
    assertThat(receivedUsers.get(2).getName()).isEqualTo(secondUser2OutboundPojo.getName());
    sourceContext.close();
}

Example 12 with MessageCollector

use of org.springframework.cloud.stream.test.binder.MessageCollector in project spring-cloud-stream by spring-cloud.

the class AvroSchemaMessageConverterTests method testSendMessageWithLocation.

@Test
public void testSendMessageWithLocation() throws Exception {
    ConfigurableApplicationContext sourceContext = SpringApplication.run(AvroSourceApplication.class, "--server.port=0", "--spring.jmx.enabled=false", "--schemaLocation=classpath:schemas/users_v1.schema", "--spring.cloud.stream.schemaRegistryClient.enabled=false", "--spring.cloud.stream.bindings.output.contentType=avro/bytes");
    Source source = sourceContext.getBean(Source.class);
    User1 firstOutboundFoo = new User1();
    firstOutboundFoo.setName("foo" + UUID.randomUUID().toString());
    firstOutboundFoo.setFavoriteColor("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", "--schemaLocation=classpath:schemas/users_v1.schema", "--spring.cloud.stream.schemaRegistryClient.enabled=false", "--spring.cloud.stream.bindings.output.contentType=avro/bytes");
    Source barSource = barSourceContext.getBean(Source.class);
    User2 firstOutboundUser2 = new User2();
    firstOutboundUser2.setFavoriteColor("foo" + UUID.randomUUID().toString());
    firstOutboundUser2.setFavoritePlace("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 secondUser2OutboundPojo = new User2();
    secondUser2OutboundPojo.setFavoriteColor("foo" + UUID.randomUUID().toString());
    secondUser2OutboundPojo.setFavoritePlace("foo" + UUID.randomUUID().toString());
    secondUser2OutboundPojo.setName("foo" + UUID.randomUUID().toString());
    source.output().send(MessageBuilder.withPayload(secondUser2OutboundPojo).build());
    Message<?> secondBarOutboundMessage = sourceMessageCollector.forChannel(source.output()).poll(1000, TimeUnit.MILLISECONDS);
    ConfigurableApplicationContext sinkContext = SpringApplication.run(AvroSinkApplication.class, "--server.port=0", "--spring.jmx.enabled=false", "--spring.cloud.stream.schemaRegistryClient.enabled=false", "--schemaLocation=classpath:schemas/users_v1.schema");
    Sink sink = sinkContext.getBean(Sink.class);
    sink.input().send(outboundMessage);
    sink.input().send(barOutboundMessage);
    sink.input().send(secondBarOutboundMessage);
    List<User1> receivedUsers = sinkContext.getBean(AvroSinkApplication.class).receivedUsers;
    assertThat(receivedUsers).hasSize(3);
    assertThat(receivedUsers.get(0)).isNotSameAs(firstOutboundFoo);
    assertThat(receivedUsers.get(0).getFavoriteColor()).isEqualTo(firstOutboundFoo.getFavoriteColor());
    assertThat(receivedUsers.get(0).getName()).isEqualTo(firstOutboundFoo.getName());
    assertThat(receivedUsers.get(1)).isNotSameAs(firstOutboundUser2);
    assertThat(receivedUsers.get(1).getFavoriteColor()).isEqualTo(firstOutboundUser2.getFavoriteColor());
    assertThat(receivedUsers.get(1).getName()).isEqualTo(firstOutboundUser2.getName());
    assertThat(receivedUsers.get(2)).isNotSameAs(secondUser2OutboundPojo);
    assertThat(receivedUsers.get(2).getFavoriteColor()).isEqualTo(secondUser2OutboundPojo.getFavoriteColor());
    assertThat(receivedUsers.get(2).getName()).isEqualTo(secondUser2OutboundPojo.getName());
    sourceContext.close();
}

Example 13 with MessageCollector

use of org.springframework.cloud.stream.test.binder.MessageCollector in project spring-cloud-stream by spring-cloud.

the class AvroSchemaRegistryClientMessageConverterTests method testSendMessage.

@Test
public void testSendMessage() throws Exception {
    ConfigurableApplicationContext schemaRegistryServerContext = SpringApplication.run(SchemaRegistryServerApplication.class);
    ConfigurableApplicationContext sourceContext = SpringApplication.run(AvroSourceApplication.class, "--server.port=0", "--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());
    sinkContext.close();
    barSourceContext.close();
    sourceContext.close();
    schemaRegistryServerContext.close();
}

Example 14 with MessageCollector

use of org.springframework.cloud.stream.test.binder.MessageCollector in project spring-cloud-stream by spring-cloud.

the class StreamListenerReactiveMethodWithReturnTypeTests method sendMessageAndValidate.

@SuppressWarnings("unchecked")
private static void sendMessageAndValidate(ConfigurableApplicationContext context) throws InterruptedException {
    Processor processor = context.getBean(Processor.class);
    String sentPayload = "hello " + UUID.randomUUID().toString();
    processor.input().send(MessageBuilder.withPayload(sentPayload).setHeader("contentType", "text/plain").build());
    MessageCollector messageCollector = context.getBean(MessageCollector.class);
    Message<String> result = (Message<String>) messageCollector.forChannel(processor.output()).poll(1000, TimeUnit.MILLISECONDS);
    assertThat(result).isNotNull();
    assertThat(result.getPayload()).isEqualTo(sentPayload.toUpperCase());
}

Example 15 with MessageCollector

use of org.springframework.cloud.stream.test.binder.MessageCollector in project spring-cloud-stream by spring-cloud.

the class StreamListenerReactiveReturnWithMessageTests method sendMessageAndValidate.

@SuppressWarnings("unchecked")
private static void sendMessageAndValidate(ConfigurableApplicationContext context) throws InterruptedException {
    Processor processor = context.getBean(Processor.class);
    String sentPayload = "hello " + UUID.randomUUID().toString();
    processor.input().send(MessageBuilder.withPayload(sentPayload).setHeader("contentType", "text/plain").build());
    MessageCollector messageCollector = context.getBean(MessageCollector.class);
    Message<String> result = (Message<String>) messageCollector.forChannel(processor.output()).poll(1000, TimeUnit.MILLISECONDS);
    assertThat(result).isNotNull();
    assertThat(result.getPayload()).isEqualTo(sentPayload.toUpperCase());
}