2525import java .util .Collections ;
2626import java .util .concurrent .BlockingDeque ;
2727import java .util .concurrent .LinkedBlockingDeque ;
28- import java .util .concurrent .Phaser ;
2928import java .util .concurrent .TimeUnit ;
3029
3130import org .junit .jupiter .api .AfterEach ;
3231import org .junit .jupiter .api .BeforeEach ;
3332
34- import org .springframework .core .task .SimpleAsyncTaskExecutor ;
35- import org .springframework .core .task .SyncTaskExecutor ;
3633import org .springframework .data .redis .ObjectFactory ;
3734import org .springframework .data .redis .connection .RedisConnectionFactory ;
3835import org .springframework .data .redis .connection .jedis .JedisConnectionFactory ;
@@ -82,33 +79,18 @@ public static Collection<Object[]> testParams() {
8279 }
8380
8481 @ BeforeEach
85- void setUp () throws Exception {
82+ void setUp () {
8683 bag .clear ();
8784
8885 adapter .setSerializer (template .getValueSerializer ());
8986 adapter .afterPropertiesSet ();
9087
91- Phaser phaser = new Phaser (1 );
92-
9388 container = new RedisMessageListenerContainer ();
9489 container .setConnectionFactory (template .getConnectionFactory ());
9590 container .setBeanName ("container" );
9691 container .addMessageListener (adapter , Arrays .asList (new ChannelTopic (CHANNEL )));
97- container .setTaskExecutor (new SyncTaskExecutor ());
98- container .setSubscriptionExecutor (new SimpleAsyncTaskExecutor () {
99- @ Override
100- protected void doExecute (Runnable task ) {
101- super .doExecute (() -> {
102- phaser .arriveAndDeregister ();
103- task .run ();
104- });
105- }
106- });
10792 container .afterPropertiesSet ();
10893 container .start ();
109-
110- phaser .arriveAndAwaitAdvance ();
111- Thread .sleep (250 );
11294 }
11395
11496 @ AfterEach
@@ -130,17 +112,18 @@ void testContainerSubscribe() {
130112 T payload1 = getT ();
131113 T payload2 = getT ();
132114
133- assertThat ( template .convertAndSend (CHANNEL , payload1 )). isEqualTo ( 1L );
134- assertThat ( template .convertAndSend (CHANNEL , payload2 )). isEqualTo ( 1L );
115+ template .convertAndSend (CHANNEL , payload1 );
116+ template .convertAndSend (CHANNEL , payload2 );
135117
136118 await ().atMost (Duration .ofSeconds (2 )).until (() -> bag .contains (payload1 ) && bag .contains (payload2 ));
137119 }
138120
139121 @ ParameterizedRedisTest
140122 void testMessageBatch () throws Exception {
123+
141124 int COUNT = 10 ;
142125 for (int i = 0 ; i < COUNT ; i ++) {
143- assertThat ( template .convertAndSend (CHANNEL , getT ())). isEqualTo ( 1L );
126+ template .convertAndSend (CHANNEL , getT ());
144127 }
145128
146129 for (int i = 0 ; i < COUNT ; i ++) {
@@ -155,8 +138,8 @@ void testContainerUnsubscribe() throws Exception {
155138 T payload2 = getT ();
156139
157140 container .removeMessageListener (adapter , new ChannelTopic (CHANNEL ));
158- assertThat ( template .convertAndSend (CHANNEL , payload1 )). isEqualTo ( 1L );
159- assertThat ( template .convertAndSend (CHANNEL , payload2 )). isEqualTo ( 1L );
141+ template .convertAndSend (CHANNEL , payload1 );
142+ template .convertAndSend (CHANNEL , payload2 );
160143
161144 assertThat (bag .poll (200 , TimeUnit .MILLISECONDS )).isNull ();
162145 }
0 commit comments