|
19 | 19 | import org.springframework.test.context.ActiveProfiles; |
20 | 20 | import org.springframework.test.context.junit4.SpringRunner; |
21 | 21 |
|
| 22 | +import java.util.HashSet; |
22 | 23 | import java.util.List; |
23 | 24 | import java.util.Map; |
24 | 25 | import java.util.Set; |
25 | | -import java.util.stream.Collectors; |
26 | 26 |
|
27 | 27 | @RunWith(SpringRunner.class) |
28 | 28 | @SpringBootTest |
29 | 29 | @ActiveProfiles("test") |
30 | 30 | @Import(TestProducerConfiguration.class) |
31 | 31 | class PulsarJavaSpringBootStarterApplicationTests { |
32 | 32 |
|
33 | | - @Autowired |
34 | | - private ConsumerBuilder consumerBuilder; |
| 33 | + @Autowired |
| 34 | + private ConsumerBuilder consumerBuilder; |
35 | 35 |
|
36 | | - @Autowired |
37 | | - private ConsumerCollector consumerCollector; |
| 36 | + @Autowired |
| 37 | + private ConsumerCollector consumerCollector; |
38 | 38 |
|
39 | | - @Autowired |
40 | | - private ProducerFactory producerFactory; |
| 39 | + @Autowired |
| 40 | + private ProducerFactory producerFactory; |
41 | 41 |
|
42 | | - @Autowired |
43 | | - private PulsarTemplate<MyMsg> producer; |
| 42 | + @Autowired |
| 43 | + private PulsarTemplate<MyMsg> producer; |
44 | 44 |
|
45 | | - @Test |
46 | | - void testProducerSendMethod() throws PulsarClientException { |
47 | | - producer.send("topic-one", new MyMsg("bb")); |
48 | | - } |
| 45 | + @Test |
| 46 | + void testProducerSendMethod() throws PulsarClientException { |
| 47 | + producer.send("topic-one", new MyMsg("bb")); |
| 48 | + } |
49 | 49 |
|
50 | | - @Before |
51 | | - public void init() throws InterruptedException { |
52 | | - Thread.sleep(100); |
53 | | - } |
| 50 | + @Before |
| 51 | + public void init() throws InterruptedException { |
| 52 | + Thread.sleep(100); |
| 53 | + } |
54 | 54 |
|
55 | | - @Test |
56 | | - void testConsumerRegistration1() throws Exception { |
57 | | - final List<Consumer> consumers = consumerBuilder.getConsumers(); |
| 55 | + @Test |
| 56 | + void testConsumerRegistration1() throws Exception { |
| 57 | + final List<Consumer> consumers = consumerBuilder.getConsumers(); |
58 | 58 |
|
59 | | - Assert.assertEquals(1, consumers.size()); |
| 59 | + Assert.assertEquals(1, consumers.size()); |
60 | 60 |
|
61 | | - final Consumer consumer = consumers.stream().findFirst().orElseThrow(Exception::new); |
| 61 | + final Consumer consumer = consumers.stream().findFirst().orElseThrow(Exception::new); |
62 | 62 |
|
63 | | - Assert.assertNotNull(consumer); |
64 | | - Assert.assertEquals("mock-topic", consumer.getTopic()); |
65 | | - } |
| 63 | + Assert.assertNotNull(consumer); |
| 64 | + Assert.assertEquals("mock-topic", consumer.getTopic()); |
| 65 | + } |
66 | 66 |
|
67 | | - @Test |
68 | | - void testConsumerRegistration2() throws Exception { |
69 | | - final Class<TestConsumerConfiguration> clazz = TestConsumerConfiguration.class; |
70 | | - final String descriptor = clazz.getName() + "#" + clazz.getDeclaredMethods()[0].getName(); |
71 | | - final ConsumerHolder consumerHolder = consumerCollector.getConsumer(descriptor).orElseThrow(Exception::new); |
| 67 | + @Test |
| 68 | + void testConsumerRegistration2() throws Exception { |
| 69 | + final Class<TestConsumerConfiguration> clazz = TestConsumerConfiguration.class; |
| 70 | + final String descriptor = clazz.getName() + "#" + clazz.getDeclaredMethods()[0].getName(); |
| 71 | + final ConsumerHolder consumerHolder = consumerCollector.getConsumer(descriptor).orElseThrow(Exception::new); |
72 | 72 |
|
73 | | - Assert.assertNotNull(consumerHolder); |
74 | | - Assert.assertEquals("mock-topic", consumerHolder.getAnnotation().topic()); |
75 | | - Assert.assertEquals(TestConsumerConfiguration.class, consumerHolder.getBean().getClass()); |
76 | | - Assert.assertEquals("mockTheListener", consumerHolder.getHandler().getName()); |
77 | | - } |
| 73 | + Assert.assertNotNull(consumerHolder); |
| 74 | + Assert.assertEquals("mock-topic", consumerHolder.getAnnotation().topic()); |
| 75 | + Assert.assertEquals(TestConsumerConfiguration.class, consumerHolder.getBean().getClass()); |
| 76 | + Assert.assertEquals("mockTheListener", consumerHolder.getHandler().getName()); |
| 77 | + } |
78 | 78 |
|
79 | | - @Test |
80 | | - void testProducerRegistration() { |
| 79 | + @Test |
| 80 | + void testProducerRegistration() { |
81 | 81 |
|
82 | | - final Map<String, ImmutablePair<Class<?>, Serialization>> topics = producerFactory.getTopics(); |
| 82 | + final Map<String, ImmutablePair<Class<?>, Serialization>> topics = producerFactory.getTopics(); |
83 | 83 |
|
84 | | - Assert.assertEquals(2, topics.size()); |
| 84 | + Assert.assertEquals(2, topics.size()); |
85 | 85 |
|
86 | | - final Set<String> topicNames = topics.entrySet().stream() |
87 | | - .map(Map.Entry::getKey).collect(Collectors.toSet()); |
| 86 | + final Set<String> topicNames = new HashSet<>(topics.keySet()); |
88 | 87 |
|
89 | | - Assert.assertTrue(topicNames.contains("topic-one")); |
90 | | - Assert.assertTrue(topicNames.contains("topic-two")); |
91 | | - } |
| 88 | + Assert.assertTrue(topicNames.contains("topic-one")); |
| 89 | + Assert.assertTrue(topicNames.contains("topic-two")); |
| 90 | + } |
92 | 91 | } |
0 commit comments