From 3e26952f2fbfc9eaaa547a6594723fca53ab5c57 Mon Sep 17 00:00:00 2001
From: Matt Joiner <anacrolix@gmail.com>
Date: Fri, 22 Aug 2014 03:45:31 +1000
Subject: [PATCH] Improve DHT distance tests
---
dht/dht_test.go | 17 +++++++++++++++--
1 file changed, 15 insertions(+), 2 deletions(-)
diff --git a/dht/dht_test.go b/dht/dht_test.go
index 18d2488c..48aaa1e5 100644
--- a/dht/dht_test.go
+++ b/dht/dht_test.go
@@ -45,6 +45,19 @@ var testIDs = []string{
"\x03" + zeroID[1:],
"\x03" + zeroID[1:18] + "\x55\xf0",
"\x55" + zeroID[1:17] + "\xff\x55\x0f",
+ "\x54" + zeroID[1:18] + "\x50\x0f",
+}
+
+func TestDistances(t *testing.T) {
+ if idDistance(testIDs[3], testIDs[0]) != 4+8+4+4 {
+ t.FailNow()
+ }
+ if idDistance(testIDs[3], testIDs[1]) != 4+8+4+4 {
+ t.FailNow()
+ }
+ if idDistance(testIDs[3], testIDs[2]) != 4+8+8 {
+ t.FailNow()
+ }
}
func TestBadIdStrings(t *testing.T) {
@@ -74,7 +87,7 @@ func TestBadIdStrings(t *testing.T) {
}
func TestClosestNodes(t *testing.T) {
- cn := newKClosestNodesSelector(2, zeroID)
+ cn := newKClosestNodesSelector(2, testIDs[3])
for _, i := range rand.Perm(len(testIDs)) {
cn.Push(testIDs[i])
}
@@ -85,7 +98,7 @@ func TestClosestNodes(t *testing.T) {
for _, id := range cn.IDs() {
m[id] = true
}
- if !m[zeroID] || !m[testIDs[1]] {
+ if !m[testIDs[3]] || !m[testIDs[4]] {
t.FailNow()
}
}
--
2.51.0