stores: add partial slab to TestUnhealthySlabs

SiaFoundation · Jul 31, 2024 · 6363ba7 · 6363ba7
1 parent 01ea3ff
commit 6363ba7
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/autopilot/migrator.go b/autopilot/migrator.go
@@ -297,6 +297,9 @@ OUTER:
 			case jobs <- job{slab, i, len(toMigrate), set, b}:
 			}
 		}
+
+		// all slabs migrated
+		return
 	}
 }
 

diff --git a/stores/metadata_test.go b/stores/metadata_test.go
@@ -1748,6 +1748,12 @@ func TestUnhealthySlabs(t *testing.T) {
 		t.Fatal(err)
 	}
 
+	// add a partial slab
+	_, _, err = ss.AddPartialSlab(context.Background(), []byte{1, 2, 3}, 1, 3, testContractSet)
+	if err != nil {
+		t.Fatal(err)
+	}
+
 	if err := ss.RefreshHealth(context.Background()); err != nil {
 		t.Fatal(err)
 	}
-Original file line number
+Diff line change
@@ Expand Up / @@ -297,6 +297,9 @@ OUTER: @@
     			case jobs <- job{slab, i, len(toMigrate), set, b}:
     			}
     		}
+    		// all slabs migrated
+    		return
     	}
     }
@@ Expand Down @@