Fix incorrect EOF when decoding some peer protocol message types

[btrtrc.git] / main_test.go

diff --git a/main_test.go b/main_test.go
index 9bb622e206609cdcacf21420b464a1e200b48b4d..578d992178d744a4d892bd692f7114a13a8477c4 100644 (file)
--- a/main_test.go
+++ b/main_test.go
@@ -4,16 +4,18 @@ import (
        "log"
        "os"
        "testing"
+
+       _ "github.com/anacrolix/envpprof"
+       analog "github.com/anacrolix/log"
 )
 
 func init() {
        log.SetFlags(log.LstdFlags | log.Lshortfile)
+       analog.DefaultTimeFormatter = analog.TimeFormatSecondsSinceInit
 }
 
 func TestMain(m *testing.M) {
-       TestingTempDir.Init("torrent.test")
        code := m.Run()
-       TestingTempDir.RemoveAll()
        // select {}
        os.Exit(code)
 }