From 606de993d3da0e6cec3d5e6f1c2a9126abec054b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=B5=B5=E5=BB=B6?= <1060026287@qq.com>
Date: Mon, 15 Feb 2021 21:01:21 +0800
Subject: [PATCH] [ISSUE #2645] Fix unit case NPE in
 DefaultMessageStoreShutDownTest#testDispatchBehindWhenShutdown

---
 .../rocketmq/store/DefaultMessageStoreShutDownTest.java    | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/store/src/test/java/org/apache/rocketmq/store/DefaultMessageStoreShutDownTest.java b/store/src/test/java/org/apache/rocketmq/store/DefaultMessageStoreShutDownTest.java
index e43bb4bc..db7d367e 100644
--- a/store/src/test/java/org/apache/rocketmq/store/DefaultMessageStoreShutDownTest.java
+++ b/store/src/test/java/org/apache/rocketmq/store/DefaultMessageStoreShutDownTest.java
@@ -40,10 +40,11 @@ public class DefaultMessageStoreShutDownTest {
 
     @Before
     public void init() throws Exception {
-        messageStore = spy(buildMessageStore());
-        boolean load = messageStore.load();
+        DefaultMessageStore store = buildMessageStore();
+        boolean load = store.load();
         assertTrue(load);
-        messageStore.start();
+        store.start();
+        messageStore = spy(store);
         when(messageStore.dispatchBehindBytes()).thenReturn(100L);
     }
 
-- 
GitLab