From 86b0c02eace8205160e2b12c1b5bee8fcdda3596 Mon Sep 17 00:00:00 2001
From: Matt Joiner <anacrolix@gmail.com>
Date: Wed, 16 Dec 2015 15:06:23 +1100
Subject: [PATCH] Use testify for some tests

---
 dht/dht_test.go | 18 +++++-------------
 1 file changed, 5 insertions(+), 13 deletions(-)

diff --git a/dht/dht_test.go b/dht/dht_test.go
index 53457c1a..7dd5ef58 100644
--- a/dht/dht_test.go
+++ b/dht/dht_test.go
@@ -22,21 +22,15 @@ func TestMarshalCompactNodeInfo(t *testing.T) {
 		ID: [20]byte{'a', 'b', 'c'},
 	}
 	addr, err := net.ResolveUDPAddr("udp4", "1.2.3.4:5")
-	if err != nil {
-		t.Fatal(err)
-	}
+	require.NoError(t, err)
 	cni.Addr = newDHTAddr(addr)
 	var b [CompactIPv4NodeInfoLen]byte
-	cni.PutCompact(b[:])
-	if err != nil {
-		t.Fatal(err)
-	}
+	err = cni.PutCompact(b[:])
+	require.NoError(t, err)
 	var bb [26]byte
 	copy(bb[:], []byte("abc"))
 	copy(bb[20:], []byte("\x01\x02\x03\x04\x00\x05"))
-	if b != bb {
-		t.FailNow()
-	}
+	assert.EqualValues(t, bb, b)
 }
 
 func recoverPanicOrDie(t *testing.T, f func()) {
@@ -108,9 +102,7 @@ func TestClosestNodes(t *testing.T) {
 
 func TestDHTDefaultConfig(t *testing.T) {
 	s, err := NewServer(nil)
-	if err != nil {
-		t.Fatal(err)
-	}
+	assert.NoError(t, err)
 	s.Close()
 }
 
-- 
2.48.1