Minor changes from self review

Signed-off-by: Matt Lord <mattalord@gmail.com>
vitessio · Feb 20, 2025 · 26055a5 · 26055a5
1 parent ca229ff
commit 26055a5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/go/vt/discovery/tablet_picker_test.go b/go/vt/discovery/tablet_picker_test.go
@@ -720,6 +720,7 @@ func TestPickNonLaggingTablets(t *testing.T) {
 	require.NoError(t, err)
 	ctx, cancel := context.WithTimeout(ctx, contextTimeout)
 	defer cancel()
+
 	var pickedPrimary, pickedLaggingReplica, pickedNonLaggingReplica int
 	for i := 0; i < numTestIterations; i++ {
 		tablet, err := tp.PickForStreaming(ctx)
@@ -734,9 +735,9 @@ func TestPickNonLaggingTablets(t *testing.T) {
 			pickedNonLaggingReplica++
 		}
 	}
-	assert.Zero(t, pickedPrimary)
-	assert.Zero(t, pickedLaggingReplica)
-	assert.Equal(t, numTestIterations, pickedNonLaggingReplica)
+	require.Zero(t, pickedPrimary)
+	require.Zero(t, pickedLaggingReplica)
+	require.Equal(t, numTestIterations, pickedNonLaggingReplica)
 }
 
 type pickerTestEnv struct {