Expose pool info for tests

filecoin-saturn · Aug 31, 2023 · 9989d73 · 9989d73
1 parent 5f2ab23
commit 9989d73
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/caboose.go b/caboose.go
@@ -164,6 +164,7 @@ func NewCaboose(config *Config) (*Caboose, error) {
 	if c.config.Harness != nil {
 		c.config.Harness.ActiveNodes = c.pool.ActiveNodes
 		c.config.Harness.AllNodes = c.pool.AllNodes
+		c.config.Harness.PoolController = c.pool
 	}
 
 	// start the pool

diff --git a/internal/state/state.go b/internal/state/state.go
@@ -3,4 +3,9 @@ package state
 type State struct {
 	ActiveNodes any
 	AllNodes    any
+	PoolController
+}
+
+type PoolController interface {
+	DoRefresh()
 }
diff --git a/pool.go b/pool.go
@@ -92,7 +92,7 @@ func (p *pool) Start() {
 	go p.checkPool()
 }
 
-func (p *pool) doRefresh() {
+func (p *pool) DoRefresh() {
 	newEP, err := p.loadPool()
 	if err == nil {
 		for _, n := range newEP {
@@ -107,21 +107,21 @@ func (p *pool) doRefresh() {
 	}
 }
 
-// refreshPool is a background thread triggering `doRefresh` every `config.PoolRefresh` interval.
+// refreshPool is a background thread triggering `DoRefresh` every `config.PoolRefresh` interval.
 func (p *pool) refreshPool() {
 	t := time.NewTimer(0)
 	started := sync.Once{}
 	for {
 		select {
 		case <-t.C:
-			p.doRefresh()
+			p.DoRefresh()
 			started.Do(func() {
 				close(p.started)
 			})
 
 			t.Reset(p.config.PoolRefresh)
 		case <-p.refresh:
-			p.doRefresh()
+			p.DoRefresh()
 			started.Do(func() {
 				close(p.started)
 			})