core: Add stop function to all components (#216)

1 files changed, 7 insertions, 3 deletions

diff --git a/core/agreement.go b/core/agreement.go
index 2337b43..8618b5f 100644
--- a/core/agreement.go
+++ b/core/agreement.go
@@ -130,9 +130,7 @@ func newAgreement(
 		fastForward:    make(chan uint64, 1),
 		authModule:     authModule,
 	}
-	agreement.restart(notarySet, types.Position{
-		ChainID: math.MaxUint32,
-	})
+	agreement.stop()
 	return agreement
 }
 
@@ -203,6 +201,12 @@ func (a *agreement) restart(
 	}
 }
 
+func (a *agreement) stop() {
+	a.restart(make(map[types.NodeID]struct{}), types.Position{
+		ChainID: math.MaxUint32,
+	})
+}
+
 // clocks returns how many time this state is required.
 func (a *agreement) clocks() int {
 	return a.state.clocks()