ipn/ipnlocal: rename misnamed DisablePortMapperForTest to DisablePortPollerForTest
I think this was originally a brain-o in 9380e2dfc6. It's
disabling the port _poller_, listing what open ports (i.e. services)
are open, not PMP/PCP/UPnP port mapping.
While there, drop in some more testenv.AssertInTest() in a few places.
Updates #cleanup
Change-Id: Ia6f755ad3544f855883b8a7bdcfc066e8649547b
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
This commit is contained in:
committed by
Brad Fitzpatrick
parent
34242df51b
commit
0bd4f4729b
@@ -1503,9 +1503,7 @@ func (b *LocalBackend) PeerCaps(src netip.Addr) tailcfg.PeerCapMap {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *LocalBackend) GetFilterForTest() *filter.Filter {
|
func (b *LocalBackend) GetFilterForTest() *filter.Filter {
|
||||||
if !testenv.InTest() {
|
testenv.AssertInTest()
|
||||||
panic("GetFilterForTest called outside of test")
|
|
||||||
}
|
|
||||||
nb := b.currentNode()
|
nb := b.currentNode()
|
||||||
return nb.filterAtomic.Load()
|
return nb.filterAtomic.Load()
|
||||||
}
|
}
|
||||||
@@ -2304,9 +2302,10 @@ func (b *LocalBackend) SetControlClientGetterForTesting(newControlClient func(co
|
|||||||
b.ccGen = newControlClient
|
b.ccGen = newControlClient
|
||||||
}
|
}
|
||||||
|
|
||||||
// DisablePortMapperForTest disables the portmapper for tests.
|
// DisablePortPollerForTest disables the port list poller for tests.
|
||||||
// It must be called before Start.
|
// It must be called before Start.
|
||||||
func (b *LocalBackend) DisablePortMapperForTest() {
|
func (b *LocalBackend) DisablePortPollerForTest() {
|
||||||
|
testenv.AssertInTest()
|
||||||
b.mu.Lock()
|
b.mu.Lock()
|
||||||
defer b.mu.Unlock()
|
defer b.mu.Unlock()
|
||||||
b.portpoll = nil
|
b.portpoll = nil
|
||||||
@@ -2315,6 +2314,7 @@ func (b *LocalBackend) DisablePortMapperForTest() {
|
|||||||
// PeersForTest returns all the current peers, sorted by Node.ID,
|
// PeersForTest returns all the current peers, sorted by Node.ID,
|
||||||
// for integration tests in another repo.
|
// for integration tests in another repo.
|
||||||
func (b *LocalBackend) PeersForTest() []tailcfg.NodeView {
|
func (b *LocalBackend) PeersForTest() []tailcfg.NodeView {
|
||||||
|
testenv.AssertInTest()
|
||||||
return b.currentNode().PeersForTest()
|
return b.currentNode().PeersForTest()
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -4030,6 +4030,7 @@ func (b *LocalBackend) resolveBestProfileLocked() (_ ipn.LoginProfileView, isBac
|
|||||||
// It is used for testing only, and will be removed along with the rest of the
|
// It is used for testing only, and will be removed along with the rest of the
|
||||||
// "current user" functionality as we progress on the multi-user improvements (tailscale/corp#18342).
|
// "current user" functionality as we progress on the multi-user improvements (tailscale/corp#18342).
|
||||||
func (b *LocalBackend) CurrentUserForTest() (ipn.WindowsUserID, ipnauth.Actor) {
|
func (b *LocalBackend) CurrentUserForTest() (ipn.WindowsUserID, ipnauth.Actor) {
|
||||||
|
testenv.AssertInTest()
|
||||||
b.mu.Lock()
|
b.mu.Lock()
|
||||||
defer b.mu.Unlock()
|
defer b.mu.Unlock()
|
||||||
return b.pm.CurrentUserID(), b.currentUser
|
return b.pm.CurrentUserID(), b.currentUser
|
||||||
|
|||||||
@@ -5816,7 +5816,7 @@ func newLocalBackendWithSysAndTestControl(t *testing.T, enableLogging bool, sys
|
|||||||
t.Fatalf("NewLocalBackend: %v", err)
|
t.Fatalf("NewLocalBackend: %v", err)
|
||||||
}
|
}
|
||||||
t.Cleanup(b.Shutdown)
|
t.Cleanup(b.Shutdown)
|
||||||
b.DisablePortMapperForTest()
|
b.DisablePortPollerForTest()
|
||||||
|
|
||||||
b.SetControlClientGetterForTesting(func(opts controlclient.Options) (controlclient.Client, error) {
|
b.SetControlClientGetterForTesting(func(opts controlclient.Options) (controlclient.Client, error) {
|
||||||
return newControl(t, opts), nil
|
return newControl(t, opts), nil
|
||||||
|
|||||||
@@ -358,7 +358,7 @@ func TestStateMachine(t *testing.T) {
|
|||||||
t.Fatalf("NewLocalBackend: %v", err)
|
t.Fatalf("NewLocalBackend: %v", err)
|
||||||
}
|
}
|
||||||
t.Cleanup(b.Shutdown)
|
t.Cleanup(b.Shutdown)
|
||||||
b.DisablePortMapperForTest()
|
b.DisablePortPollerForTest()
|
||||||
|
|
||||||
var cc, previousCC *mockControl
|
var cc, previousCC *mockControl
|
||||||
b.SetControlClientGetterForTesting(func(opts controlclient.Options) (controlclient.Client, error) {
|
b.SetControlClientGetterForTesting(func(opts controlclient.Options) (controlclient.Client, error) {
|
||||||
|
|||||||
@@ -45,7 +45,7 @@ func newBackend(opts *options) *ipnlocal.LocalBackend {
|
|||||||
tb.Fatalf("NewLocalBackend: %v", err)
|
tb.Fatalf("NewLocalBackend: %v", err)
|
||||||
}
|
}
|
||||||
tb.Cleanup(b.Shutdown)
|
tb.Cleanup(b.Shutdown)
|
||||||
b.DisablePortMapperForTest()
|
b.DisablePortPollerForTest()
|
||||||
b.SetControlClientGetterForTesting(opts.MakeControlClient)
|
b.SetControlClientGetterForTesting(opts.MakeControlClient)
|
||||||
return b
|
return b
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user