netbird/client/internal/lazyconn/inactivity/manager_test.go

package inactivity

import (
	"context"
	"fmt"
	"testing"
	"time"

	log "github.com/sirupsen/logrus"

	"github.com/netbirdio/netbird/client/internal/lazyconn"
	"github.com/netbirdio/netbird/util"
)

func init() {
	_ = util.InitLog("trace", "console")
}

func TestNewManager(t *testing.T) {
	for i, sc := range scenarios {
		timer := NewFakeTimer()
		newTicker = func(d time.Duration) Ticker {
			return newFakeTicker(d, timer)
		}

		t.Run(fmt.Sprintf("Scenario %d", i), func(t *testing.T) {
			mock := newMockWgInterface("peer1", sc.Data, timer)
			manager := NewManager(mock, nil)
			peerCfg := &lazyconn.PeerConfig{
				PublicKey: "peer1",
				Log:       log.WithField("peer", "peer1"),
			}
			manager.AddPeer(peerCfg)

			ctx, cancel := context.WithCancel(context.Background())
			defer cancel()

			manager.Start(ctx)

			var inactiveResult bool
			select {
			case <-manager.InactivePeersChan:
				inactiveResult = true
			default:
				inactiveResult = false
			}

			if inactiveResult != sc.ExpectedInactive {
				t.Errorf("Expected inactive peers: %v, got: %v", sc.ExpectedInactive, inactiveResult)
			}
		})
	}
}