🚨test: Use TestProbe for resetting StorageExtension

Author: negokaz

src/test/scala/lerna/akka/entityreplication/raft/RaftActorSnapshotSynchronizationSpec.scala

@@ -33,10 +33,11 @@ class RaftActorSnapshotSynchronizationSpec
   override def beforeEach(): Unit = {
     super.beforeEach()
     // clear storage
+    val probe   = TestProbe()
     val storage = StorageExtension(system)
-    storage.journalStorage ! InMemoryJournalStorage.ClearJournal
-    storage.snapshotStorage ! InMemorySnapshotStorage.ClearSnapshots
-    receiveWhile(messages = 2) {
+    probe.send(storage.journalStorage, InMemoryJournalStorage.ClearJournal)
+    probe.send(storage.snapshotStorage, InMemorySnapshotStorage.ClearSnapshots)
+    probe.receiveWhile(messages = 2) {
       case _: Status.Success => Done
     } should have length 2
     // reset SnapshotStore

src/test/scala/lerna/akka/entityreplication/raft/snapshot/sync/SnapshotSyncManagerSpec.scala

@@ -59,12 +59,13 @@ class SnapshotSyncManagerSpec extends TestKit(ActorSystem()) with ActorSpec with
   override def beforeEach(): Unit = {
     super.beforeEach()
     // clear storage
+    val probe   = TestProbe()
     val storage = StorageExtension(system)
-    storage.journalStorage ! InMemoryJournalStorage.ClearJournal
-    storage.snapshotStorage ! InMemorySnapshotStorage.ClearSnapshots
-    receiveWhile(messages = 2) {
+    probe.send(storage.journalStorage, InMemoryJournalStorage.ClearJournal)
+    probe.send(storage.snapshotStorage, InMemorySnapshotStorage.ClearSnapshots)
+    probe.receiveWhile(messages = 2) {
       case _: Status.Success => Done
-    }
+    } should have length 2
     // reset TestKits
     srcRaftSnapshotStoreTestKit.reset()
     dstRaftSnapshotStoreTestKit.reset()