Merge pull request #525 from iotaledger/fix/testsuite-tracking-filter…

…ed-blocks Fix tracking filtered blocks in the Testsuite
iotaledger · Nov 15, 2023 · d070e81 · d070e81
2 parents 3ab0b0a + 4c0ac56
commit d070e81
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pkg/testsuite/mock/node.go b/pkg/testsuite/mock/node.go
@@ -150,6 +150,13 @@ func (n *Node) hookEvents() {
 	events.CandidateEngineActivated.Hook(func(e *engine.Engine) { n.candidateEngineActivatedCount.Add(1) })
 
 	events.MainEngineSwitched.Hook(func(e *engine.Engine) { n.mainEngineSwitchedCount.Add(1) })
+
+	n.Protocol.Events.Engine.CommitmentFilter.BlockFiltered.Hook(func(event *commitmentfilter.BlockFilteredEvent) {
+		n.mutex.Lock()
+		defer n.mutex.Unlock()
+
+		n.filteredBlockEvents = append(n.filteredBlockEvents, event)
+	})
 }
 
 func (n *Node) hookLogging(failOnBlockFiltered bool) {