swamp: add options to swamp config (#728)

* swamp: add options to swamp config
celestiaorg · May 19, 2022 · d99e946 · d99e946
1 parent 0466d9c
commit d99e946
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 6 deletions.
diff --git a/node/tests/swamp/config.go b/node/tests/swamp/config.go
@@ -26,3 +26,13 @@ func DefaultComponents() *Components {
 		CoreCfg: tnCfg,
 	}
 }
+
+// Option for modifying Swamp's Config.
+type Option func(*Components)
+
+// WithBlockInterval sets a custom interval for block creation.
+func WithBlockInterval(interval time.Duration) Option {
+	return func(c *Components) {
+		c.CoreCfg.Consensus.CreateEmptyBlocksInterval = interval
+	}
+}
diff --git a/node/tests/swamp/swamp.go b/node/tests/swamp/swamp.go
@@ -47,11 +47,16 @@ type Swamp struct {
 }
 
 // NewSwamp creates a new instance of Swamp.
-func NewSwamp(t *testing.T, ic *Components) *Swamp {
+func NewSwamp(t *testing.T, options ...Option) *Swamp {
 	if testing.Verbose() {
 		log.SetDebugLogging()
 	}
 
+	ic := DefaultComponents()
+	for _, option := range options {
+		option(ic)
+	}
+
 	var err error
 	ctx := context.Background()
 

diff --git a/node/tests/sync_test.go b/node/tests/sync_test.go
@@ -28,7 +28,7 @@ Steps:
 6. Check LN is synced to height 30
 */
 func TestSyncLightWithBridge(t *testing.T) {
-	sw := swamp.NewSwamp(t, swamp.DefaultComponents())
+	sw := swamp.NewSwamp(t)
 
 	bridge := sw.NewBridgeNode()
 
@@ -76,7 +76,7 @@ Steps:
 9. Check LN is synced to height 40
 */
 func TestSyncStartStopLightWithBridge(t *testing.T) {
-	sw := swamp.NewSwamp(t, swamp.DefaultComponents())
+	sw := swamp.NewSwamp(t)
 
 	bridge := sw.NewBridgeNode()
 
@@ -128,7 +128,7 @@ Steps:
 6. Check FN is synced to height 30
 */
 func TestSyncFullWithBridge(t *testing.T) {
-	sw := swamp.NewSwamp(t, swamp.DefaultComponents())
+	sw := swamp.NewSwamp(t)
 
 	bridge := sw.NewBridgeNode()
 
@@ -174,7 +174,7 @@ Steps:
 9. Check LN is synced to height 50
 */
 func TestSyncLightWithFull(t *testing.T) {
-	sw := swamp.NewSwamp(t, swamp.DefaultComponents())
+	sw := swamp.NewSwamp(t)
 
 	bridge := sw.NewBridgeNode()
 
@@ -236,7 +236,7 @@ Steps:
 9. Check LN is synced to height 50
 */
 func TestSyncLightWithTrustedPeers(t *testing.T) {
-	sw := swamp.NewSwamp(t, swamp.DefaultComponents())
+	sw := swamp.NewSwamp(t)
 
 	bridge := sw.NewBridgeNode()