Move RunImmediatelyScheduledJobs to faketime

Use it everywhere.

PiperOrigin-RevId: 375539262

3 files changed, 8 insertions, 23 deletions

diff --git a/pkg/tcpip/network/internal/ip/duplicate_address_detection_test.go b/pkg/tcpip/network/internal/ip/duplicate_address_detection_test.go
index a22b712c6..24687cf06 100644
--- a/pkg/tcpip/network/internal/ip/duplicate_address_detection_test.go
+++ b/pkg/tcpip/network/internal/ip/duplicate_address_detection_test.go
@@ -133,7 +133,7 @@ func TestDADCheckDuplicateAddress(t *testing.T) {
 		t.Errorf("got dad.checkDuplicateAddress(%s, _) = %d, want = %d", addr1, res, stack.DADDisabled)
 	}
 	// Wait for any initially fired timers to complete.
-	clock.Advance(0)
+	clock.RunImmediatelyScheduledJobs()
 	if diff := dad.check(nil); diff != "" {
 		t.Errorf("dad check mismatch (-want +got):\n%s", diff)
 	}
@@ -147,7 +147,7 @@ func TestDADCheckDuplicateAddress(t *testing.T) {
 	if res := dad.checkDuplicateAddress(addr1, handler(ch, addr1)); res != stack.DADStarting {
 		t.Errorf("got dad.checkDuplicateAddress(%s, _) = %d, want = %d", addr1, res, stack.DADStarting)
 	}
-	clock.Advance(0)
+	clock.RunImmediatelyScheduledJobs()
 	if diff := dad.check([]tcpip.Address{addr1}); diff != "" {
 		t.Errorf("dad check mismatch (-want +got):\n%s", diff)
 	}
@@ -156,7 +156,7 @@ func TestDADCheckDuplicateAddress(t *testing.T) {
 	if res := dad.checkDuplicateAddress(addr1, handler(ch, addr1)); res != stack.DADAlreadyRunning {
 		t.Errorf("got dad.checkDuplicateAddress(%s, _) = %d, want = %d", addr1, res, stack.DADAlreadyRunning)
 	}
-	clock.Advance(0)
+	clock.RunImmediatelyScheduledJobs()
 	if diff := dad.check(nil); diff != "" {
 		t.Errorf("dad check mismatch (-want +got):\n%s", diff)
 	}
@@ -170,7 +170,7 @@ func TestDADCheckDuplicateAddress(t *testing.T) {
 	if res := dad.checkDuplicateAddress(addr2, handler(ch, addr2)); res != stack.DADStarting {
 		t.Errorf("got dad.checkDuplicateAddress(%s, _) = %d, want = %d", addr2, res, stack.DADStarting)
 	}
-	clock.Advance(0)
+	clock.RunImmediatelyScheduledJobs()
 	if diff := dad.check([]tcpip.Address{addr2}); diff != "" {
 		t.Errorf("dad check mismatch (-want +got):\n%s", diff)
 	}
@@ -208,7 +208,7 @@ func TestDADCheckDuplicateAddress(t *testing.T) {
 	if res := dad.checkDuplicateAddress(addr2, handler(ch, addr2)); res != stack.DADStarting {
 		t.Errorf("got dad.checkDuplicateAddress(%s, _) = %d, want = %d", addr2, res, stack.DADStarting)
 	}
-	clock.Advance(0)
+	clock.RunImmediatelyScheduledJobs()
 	if diff := dad.check([]tcpip.Address{addr2, addr2}); diff != "" {
 		t.Errorf("dad check mismatch (-want +got):\n%s", diff)
 	}
@@ -247,7 +247,7 @@ func TestDADStop(t *testing.T) {
 	if res := dad.checkDuplicateAddress(addr3, handler(ch, addr3)); res != stack.DADStarting {
 		t.Errorf("got dad.checkDuplicateAddress(%s, _) = %d, want = %d", addr2, res, stack.DADStarting)
 	}
-	clock.Advance(0)
+	clock.RunImmediatelyScheduledJobs()
 	if diff := dad.check([]tcpip.Address{addr1, addr2, addr3}); diff != "" {
 		t.Errorf("dad check mismatch (-want +got):\n%s", diff)
 	}
@@ -272,7 +272,7 @@ func TestDADStop(t *testing.T) {
 	if res := dad.checkDuplicateAddress(addr1, handler(ch, addr1)); res != stack.DADStarting {
 		t.Errorf("got dad.checkDuplicateAddress(%s, _) = %d, want = %d", addr1, res, stack.DADStarting)
 	}
-	clock.Advance(0)
+	clock.RunImmediatelyScheduledJobs()
 	if diff := dad.check([]tcpip.Address{addr1}); diff != "" {
 		t.Errorf("dad check mismatch (-want +got):\n%s", diff)
 	}
@@ -347,7 +347,7 @@ func TestNonce(t *testing.T) {
 				t.Errorf("got dad.checkDuplicateAddress(%s, _) = %d, want = %d", addr1, res, stack.DADStarting)
 			}
 
-			clock.Advance(0)
+			clock.RunImmediatelyScheduledJobs()
 			for i, want := range test.expectedResults {
 				if got := dad.extendIfNonceEqual(addr1, test.mockedReceivedNonce); got != want {
 					t.Errorf("(i=%d) got dad.extendIfNonceEqual(%s, _) = %d, want = %d", i, addr1, got, want)
diff --git a/pkg/tcpip/network/internal/testutil/BUILD b/pkg/tcpip/network/internal/testutil/BUILD
index b36134ddd..a180e5c75 100644
--- a/pkg/tcpip/network/internal/testutil/BUILD
+++ b/pkg/tcpip/network/internal/testutil/BUILD
@@ -15,7 +15,6 @@ go_library(
     deps = [
         "//pkg/tcpip",
         "//pkg/tcpip/buffer",
-        "//pkg/tcpip/faketime",
         "//pkg/tcpip/header",
         "//pkg/tcpip/stack",
     ],
diff --git a/pkg/tcpip/network/internal/testutil/testutil.go b/pkg/tcpip/network/internal/testutil/testutil.go
index 328d5efee..605e9ef8d 100644
--- a/pkg/tcpip/network/internal/testutil/testutil.go
+++ b/pkg/tcpip/network/internal/testutil/testutil.go
@@ -19,27 +19,13 @@ package testutil
 import (
 	"fmt"
 	"math/rand"
-	"time"
 
 	"gvisor.dev/gvisor/pkg/tcpip"
 	"gvisor.dev/gvisor/pkg/tcpip/buffer"
-	"gvisor.dev/gvisor/pkg/tcpip/faketime"
 	"gvisor.dev/gvisor/pkg/tcpip/header"
 	"gvisor.dev/gvisor/pkg/tcpip/stack"
 )
 
-const (
-	// immediateDuration is a duration of zero for scheduling work that needs to
-	// be done immediately but asynchronously to avoid deadlock.
-	immediateDuration time.Duration = 0
-)
-
-// RunImmediatelyScheduledJobs runs all jobs scheduled to run at the current
-// time.
-func RunImmediatelyScheduledJobs(clock *faketime.ManualClock) {
-	clock.Advance(immediateDuration)
-}
-
 // MockLinkEndpoint is an endpoint used for testing, it stores packets written
 // to it and can mock errors.
 type MockLinkEndpoint struct {