add another test

Signed-off-by: lance6716 <[email protected]>
pingcap · Dec 28, 2023 · 0306cec · 0306cec
1 parent 638070f
commit 0306cec
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/br/pkg/lightning/backend/local/local_test.go b/br/pkg/lightning/backend/local/local_test.go
@@ -2386,4 +2386,9 @@ func TestCheckDiskAvail(t *testing.T) {
 	ctx := context.Background()
 	err := checkDiskAvail(ctx, store)
 	require.NoError(t, err)
+
+	// pd may return this StoreInfo before the store reports heartbeat
+	store = &http.StoreInfo{Status: http.StoreStatus{LeaderWeight: 1.0, RegionWeight: 1.0}}
+	err = checkDiskAvail(ctx, store)
+	require.NoError(t, err)
 }