Fix UTs

registration/db_test.go

@@ -13,6 +13,7 @@ func TestInsertAndGetProof(t *testing.T) {
 	tempdir := t.TempDir()
 	db, err := newDatabase(tempdir, []byte{})
 	require.NoError(t, err)
+	t.Cleanup(func() { require.NoError(t, db.Close()) })
 
 	require.NoError(t, db.SaveProof(context.Background(), shared.NIP{Epoch: 1}, [][]byte{}))
 	proof, err := db.GetProof(context.Background(), "1")

registration/registration_test.go

@@ -209,7 +209,7 @@ func TestRecoveringRoundInProgress(t *testing.T) {
 
 	roundCfg := server.RoundConfig{
 		EpochDuration: time.Hour,
-		PhaseShift:    time.Second / 2,
+		PhaseShift:    time.Second,
 	}
 
 	ctx, cancel := context.WithCancel(context.Background())

server/server_test.go

@@ -462,6 +462,7 @@ func TestLoadSubmits(t *testing.T) {
 	cfg.Round.EpochDuration = time.Minute * 2
 	cfg.Round.PhaseShift = time.Minute
 	cfg.RawRPCListener = randomHost
+	cfg.RawRESTListener = randomHost
 	cfg, err := server.SetupConfig(cfg)
 	req.NoError(err)
 

service/service_test.go

@@ -3,6 +3,8 @@ package service_test
 import (
 	"bytes"
 	"context"
+	"errors"
+	"os"
 	"path/filepath"
 	"testing"
 	"time"
@@ -217,7 +219,10 @@ func TestRecoverFinishedRound(t *testing.T) {
 	}, time.Second, time.Millisecond)
 
 	req.Equal(uint(9876), proof.Epoch)
-	req.NoDirExists(filepath.Join(datadir, "rounds", "9876"))
+	req.Eventually(func() bool {
+		_, err := os.Lstat(filepath.Join(datadir, "rounds", "9876"))
+		return errors.Is(err, os.ErrNotExist)
+	}, time.Second, 10*time.Millisecond)
 
 	cancel()
 	req.NoError(eg.Wait())

state/state.go

@@ -29,6 +29,7 @@ func Load(filename string, v any) error {
 	if err != nil {
 		return fmt.Errorf("loading file: %w", err)
 	}
+	defer f.Close()
 
 	_, err = xdr.Unmarshal(f, v)
 	if err != nil {