highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-08-27 06:01:48 +02:00
Code Issues Packages Projects Releases Wiki Activity

protofsm: add CustomPollInterval for mocking purposes

Browse Source 
Adding this makes a state machine easier to unit test, as the caller can
specify a custom polling interval.
This commit is contained in:
Olaoluwa Osuntokun
2024-02-29 16:08:54 -06:00
parent 35ea05d5dc
commit d805c0fc3c
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
protofsm/state_machine.go
View File

@@ -178,6 +178,10 @@ type StateMachineCfg[Event any, Env Environment] struct {
// MsgMapper is an optional message mapper that can be used to map // MsgMapper is an optional message mapper that can be used to map
// normal wire messages into FSM events. // normal wire messages into FSM events.
MsgMapper fn.Option[MsgMapper[Event]] MsgMapper fn.Option[MsgMapper[Event]]
// CustomPollInterval is an optional custom poll interval that can be
// used to set a quicker interval for tests.
CustomPollInterval fn.Option[time.Duration]
} }
// NewStateMachine creates a new state machine given a set of daemon adapters, // NewStateMachine creates a new state machine given a set of daemon adapters,
@@ -373,7 +377,9 @@ func (s *StateMachine[Event, Env]) executeDaemonEvent( //nolint:funlen
go func() { go func() {
defer s.wg.Done() defer s.wg.Done()
predicateTicker := time.NewTicker(pollInterval) predicateTicker := time.NewTicker(
s.cfg.CustomPollInterval.UnwrapOr(pollInterval),
)
defer predicateTicker.Stop() defer predicateTicker.Stop()
log.Infof("FSM(%v): waiting for send predicate to "+ log.Infof("FSM(%v): waiting for send predicate to "+