Review feedback

Author: ciarams87

internal/mode/static/nginx/runtime/verify.go

@@ -93,14 +93,7 @@ func (c *verifyClient) waitForCorrectVersion(
 		return fmt.Errorf(noNewWorkersErrFmt, expectedVersion, err)
 	}
 
-	if err := wait.PollUntilContextCancel(
-		ctx,
-		25*time.Millisecond,
-		true, /* poll immediately */
-		func(ctx context.Context) (bool, error) {
-			version, err := c.getConfigVersion()
-			return version == expectedVersion, err
-		}); err != nil {
+	if err := c.ensureConfigVersion(ctx, expectedVersion); err != nil {
 		if errors.Is(err, context.DeadlineExceeded) {
 			err = fmt.Errorf(
 				"config version check didn't return expected version %d within the deadline",
@@ -112,6 +105,18 @@ func (c *verifyClient) waitForCorrectVersion(
 	return nil
 }
 
+func (c *verifyClient) ensureConfigVersion(ctx context.Context, expectedVersion int) error {
+	return wait.PollUntilContextCancel(
+		ctx,
+		25*time.Millisecond,
+		true, /* poll immediately */
+		func(ctx context.Context) (bool, error) {
+			version, err := c.getConfigVersion()
+			return version == expectedVersion, err
+		},
+	)
+}
+
 func ensureNewNginxWorkers(
 	ctx context.Context,
 	childProcFile string,

internal/mode/static/nginx/runtime/verify_test.go

@@ -119,7 +119,7 @@ func TestEnsureNewNginxWorkers(t *testing.T) {
 	ctx := context.Background()
 
 	cancellingCtx, cancel := context.WithCancel(ctx)
-	time.AfterFunc(10000*time.Millisecond, cancel)
+	time.AfterFunc(100*time.Millisecond, cancel)
 
 	cancellingCtx2, cancel2 := context.WithCancel(ctx)
 	time.AfterFunc(1*time.Millisecond, cancel2)
@@ -150,7 +150,7 @@ func TestEnsureNewNginxWorkers(t *testing.T) {
 			readFile:         readFilePrevious,
 			previousContents: previousContents,
 			expectError:      true,
-			name:             "no new workers",
+			name:             "timed out waiting for new workers",
 		},
 		{
 			ctx:              cancellingCtx2,