fix store test

Author: LittleBoy18

store/src/test/java/org/apache/rocketmq/store/ha/autoswitch/AutoSwitchHATest.java

@@ -221,12 +221,12 @@ public void testConfirmOffset() throws Exception {
         // Step2, shutdown store2
         this.messageStore2.shutdown();
 
-        // Put message, which should put failed.
+        // Put message, which should succeed because slave is removed from syncStateSet, only master remains
         final PutMessageResult putMessageResult = this.messageStore1.putMessage(buildMessage());
-        assertEquals(putMessageResult.getPutMessageStatus(), PutMessageStatus.FLUSH_SLAVE_TIMEOUT);
+        assertEquals(PutMessageStatus.PUT_OK,putMessageResult.getPutMessageStatus());
 
-        // The confirmOffset still don't change, because syncStateSet contains broker2, but broker2 shutdown
-        assertEquals(confirmOffset, this.messageStore1.getConfirmOffset());
+        // The confirmOffset should update because syncStateSet only contains master after slave shutdown
+        assertTrue(this.messageStore1.getConfirmOffset() >= confirmOffset);
 
         // Step3, shutdown store1, start store2, change store2 to master, epoch = 2
         this.messageStore1.shutdown();
@@ -299,7 +299,7 @@ public void testOptionAllAckInSyncStateSet() throws Exception {
         ((AutoSwitchHAService) this.messageStore1.getHaService()).setSyncStateSet(result);
 
         final PutMessageResult putMessageResult = this.messageStore1.putMessage(buildMessage());
-        assertEquals(putMessageResult.getPutMessageStatus(), PutMessageStatus.FLUSH_SLAVE_TIMEOUT);
+        assertEquals(PutMessageStatus.FLUSH_SLAVE_TIMEOUT,putMessageResult.getPutMessageStatus());
     }
 
     @Ignore