1 // Copyright (c) 2013-2015 The btcsuite developers
2 // Use of this source code is governed by an ISC
3 // license that can be found in the LICENSE file.
12 "github.com/btcsuite/btcd/addrmgr"
13 "github.com/btcsuite/btcd/wire"
16 func TestChance(t *testing.T) {
17 now := time.Unix(time.Now().Unix(), 0)
18 var tests = []struct {
19 addr *addrmgr.KnownAddress
24 addrmgr.TstNewKnownAddress(&wire.NetAddress{Timestamp: now.Add(-35 * time.Second)},
25 0, time.Now().Add(-30*time.Minute), time.Now(), false, 0),
28 //Test case in which lastseen < 0
29 addrmgr.TstNewKnownAddress(&wire.NetAddress{Timestamp: now.Add(20 * time.Second)},
30 0, time.Now().Add(-30*time.Minute), time.Now(), false, 0),
33 //Test case in which lastattempt < 0
34 addrmgr.TstNewKnownAddress(&wire.NetAddress{Timestamp: now.Add(-35 * time.Second)},
35 0, time.Now().Add(30*time.Minute), time.Now(), false, 0),
38 //Test case in which lastattempt < ten minutes
39 addrmgr.TstNewKnownAddress(&wire.NetAddress{Timestamp: now.Add(-35 * time.Second)},
40 0, time.Now().Add(-5*time.Minute), time.Now(), false, 0),
43 //Test case with several failed attempts.
44 addrmgr.TstNewKnownAddress(&wire.NetAddress{Timestamp: now.Add(-35 * time.Second)},
45 2, time.Now().Add(-30*time.Minute), time.Now(), false, 0),
51 for i, test := range tests {
52 chance := addrmgr.TstKnownAddressChance(test.addr)
53 if math.Abs(test.expected-chance) >= err {
54 t.Errorf("case %d: got %f, expected %f", i, chance, test.expected)
59 func TestIsBad(t *testing.T) {
60 now := time.Unix(time.Now().Unix(), 0)
61 future := now.Add(35 * time.Minute)
62 monthOld := now.Add(-43 * time.Hour * 24)
63 secondsOld := now.Add(-2 * time.Second)
64 minutesOld := now.Add(-27 * time.Minute)
65 hoursOld := now.Add(-5 * time.Hour)
66 zeroTime := time.Time{}
68 futureNa := &wire.NetAddress{Timestamp: future}
69 minutesOldNa := &wire.NetAddress{Timestamp: minutesOld}
70 monthOldNa := &wire.NetAddress{Timestamp: monthOld}
71 currentNa := &wire.NetAddress{Timestamp: secondsOld}
73 //Test addresses that have been tried in the last minute.
74 if addrmgr.TstKnownAddressIsBad(addrmgr.TstNewKnownAddress(futureNa, 3, secondsOld, zeroTime, false, 0)) {
75 t.Errorf("test case 1: addresses that have been tried in the last minute are not bad.")
77 if addrmgr.TstKnownAddressIsBad(addrmgr.TstNewKnownAddress(monthOldNa, 3, secondsOld, zeroTime, false, 0)) {
78 t.Errorf("test case 2: addresses that have been tried in the last minute are not bad.")
80 if addrmgr.TstKnownAddressIsBad(addrmgr.TstNewKnownAddress(currentNa, 3, secondsOld, zeroTime, false, 0)) {
81 t.Errorf("test case 3: addresses that have been tried in the last minute are not bad.")
83 if addrmgr.TstKnownAddressIsBad(addrmgr.TstNewKnownAddress(currentNa, 3, secondsOld, monthOld, true, 0)) {
84 t.Errorf("test case 4: addresses that have been tried in the last minute are not bad.")
86 if addrmgr.TstKnownAddressIsBad(addrmgr.TstNewKnownAddress(currentNa, 2, secondsOld, secondsOld, true, 0)) {
87 t.Errorf("test case 5: addresses that have been tried in the last minute are not bad.")
90 //Test address that claims to be from the future.
91 if !addrmgr.TstKnownAddressIsBad(addrmgr.TstNewKnownAddress(futureNa, 0, minutesOld, hoursOld, true, 0)) {
92 t.Errorf("test case 6: addresses that claim to be from the future are bad.")
95 //Test address that has not been seen in over a month.
96 if !addrmgr.TstKnownAddressIsBad(addrmgr.TstNewKnownAddress(monthOldNa, 0, minutesOld, hoursOld, true, 0)) {
97 t.Errorf("test case 7: addresses more than a month old are bad.")
100 //It has failed at least three times and never succeeded.
101 if !addrmgr.TstKnownAddressIsBad(addrmgr.TstNewKnownAddress(minutesOldNa, 3, minutesOld, zeroTime, true, 0)) {
102 t.Errorf("test case 8: addresses that have never succeeded are bad.")
105 //It has failed ten times in the last week
106 if !addrmgr.TstKnownAddressIsBad(addrmgr.TstNewKnownAddress(minutesOldNa, 10, minutesOld, monthOld, true, 0)) {
107 t.Errorf("test case 9: addresses that have not succeeded in too long are bad.")
110 //Test an address that should work.
111 if addrmgr.TstKnownAddressIsBad(addrmgr.TstNewKnownAddress(minutesOldNa, 2, minutesOld, hoursOld, true, 0)) {
112 t.Errorf("test case 10: This should be a valid address.")