@@ -484,13 +484,6 @@ func TestGatewayServer(t *testing.T) {
484
484
485
485
t .Run ("Upstream" , func (t * testing.T ) {
486
486
uplinkCount := 0
487
- upEvents := map [string ]events.Channel {}
488
- for _ , event := range []string {"gs.up.receive" , "gs.down.tx.success" , "gs.down.tx.fail" , "gs.status.receive" } {
489
- ch := make (events.Channel , 5 )
490
- events .Subscribe (event , ch )
491
- defer events .Unsubscribe (event , ch )
492
- upEvents [event ] = ch
493
- }
494
487
for _ , tc := range []struct {
495
488
Name string
496
489
Up * ttnpb.GatewayUp
@@ -664,6 +657,23 @@ func TestGatewayServer(t *testing.T) {
664
657
t .Run (tc .Name , func (t * testing.T ) {
665
658
a := assertions .New (t )
666
659
660
+ upEvents := map [string ]events.Channel {}
661
+ for _ , event := range []string {"gs.up.receive" , "gs.down.tx.success" , "gs.down.tx.fail" , "gs.status.receive" } {
662
+ upEvents [event ] = make (events.Channel , 5 )
663
+ }
664
+ defer test .SetDefaultEventsPubSub (& test.MockEventPubSub {
665
+ PublishFunc : func (ev events.Event ) {
666
+ switch name := ev .Name (); name {
667
+ case "gs.up.receive" , "gs.down.tx.success" , "gs.down.tx.fail" , "gs.status.receive" :
668
+ go func () {
669
+ upEvents [name ] <- ev
670
+ }()
671
+ default :
672
+ t .Logf ("%s event published" , name )
673
+ }
674
+ },
675
+ })()
676
+
667
677
select {
668
678
case upCh <- tc .Up :
669
679
case <- time .After (timeout ):
0 commit comments