fix uploader tests by adding mock host to store

SiaFoundation · Dec 3, 2024 · 21479c1 · 21479c1
1 parent 55b3b6e
commit 21479c1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/internal/upload/uploader/uploader_test.go b/internal/upload/uploader/uploader_test.go
@@ -21,6 +21,7 @@ func TestUploaderStopped(t *testing.T) {
 	c := mocks.NewContract(types.PublicKey{1}, types.FileContractID{1})
 	cl := mocks.NewContractLocker()
 
+	hs.AddHost()
 	ul, err := New(context.Background(), cl, cs, hm, hs, c.Metadata(), zap.NewNop().Sugar())
 	if err != nil {
 		t.Fatal(err)
@@ -114,6 +115,7 @@ func TestRefreshUploader(t *testing.T) {
 	cl := mocks.NewContractLocker()
 
 	// create uploader
+	hs.AddHost()
 	hk := types.PublicKey{1}
 	c1 := cs.AddContract(hk)
 	ul, err := New(context.Background(), cl, cs, hm, hs, c1.Metadata(), zap.NewNop().Sugar())