Add WithTestExtension option

MarcoPolo · MarcoPolo · commit 996ce03c32a1 · 2025-08-12T14:30:46.000-07:00
diff --git a/extensions.go b/extensions.go
@@ -9,10 +9,10 @@ type PeerExtensions struct {
 	TestExtension bool
 }
 
-func WithPeerExtensions(exts PeerExtensions) Option {
+func WithTestExtension() Option {
 	return func(ps *PubSub) error {
 		if rt, ok := ps.rt.(*GossipSubRouter); ok {
-			rt.extensions.myExtensions = exts
+			rt.extensions.myExtensions.TestExtension = true
 		}
 		return nil
 	}

Original file line number	Diff line number	Diff line change
`@@ -9,10 +9,10 @@ type PeerExtensions struct {`
`9`	`9`	`TestExtension bool`
`10`	`10`	`}`
`11`	`11`
`12`		`-func WithPeerExtensions(exts PeerExtensions) Option {`
	`12`	`+func WithTestExtension() Option {`
`13`	`13`	`return func(ps *PubSub) error {`
`14`	`14`	`if rt, ok := ps.rt.(*GossipSubRouter); ok {`
`15`		`- rt.extensions.myExtensions = exts`
	`15`	`+ rt.extensions.myExtensions.TestExtension = true`
`16`	`16`	`}`
`17`	`17`	`return nil`
`18`	`18`	`}`