Test and fix closed sqlite storage panicking during piece hashing

[btrtrc.git] / client_test.go

diff --git a/client_test.go b/client_test.go
index 95c0face918715bcb3c48ae5129b5a4bfa34aa88..22281efd6cde9bf0eaa6a0b03d7f48150ee5a500 100644 (file)
--- a/client_test.go
+++ b/client_test.go
@@ -592,6 +592,7 @@ func TestPieceCompletedInStorageButNotClient(t *testing.T) {
        cfg.DataDir = greetingTempDir
        seeder, err := NewClient(TestingConfig(t))
        require.NoError(t, err)
+       defer seeder.Close()
        seeder.AddTorrentSpec(&TorrentSpec{
                InfoBytes: greetingMetainfo.InfoBytes,
        })
@@ -787,13 +788,16 @@ func TestObfuscatedHeaderFallbackSeederRequiresLeecherPrefersNot(t *testing.T) {
 }
 
 func TestClientAddressInUse(t *testing.T) {
-       s, _ := NewUtpSocket("udp", ":50007", nil, log.Default)
+       s, _ := NewUtpSocket("udp", "localhost:50007", nil, log.Default)
        if s != nil {
                defer s.Close()
        }
-       cfg := TestingConfig(t).SetListenAddr(":50007")
+       cfg := TestingConfig(t).SetListenAddr("localhost:50007")
        cfg.DisableUTP = false
        cl, err := NewClient(cfg)
+       if err == nil {
+               assert.Nil(t, cl.Close())
+       }
        require.Error(t, err)
        require.Nil(t, cl)
 }