arnaucube
arnaucube/go-dhtPublic Self-hosted repository
go-dht/id_test.go
id_test.goRaw
 1package main
 2
 3import (
 4	"testing"
 5
 6	"github.com/stretchr/testify/assert"
 7)
 8
 9func TestNewID(t *testing.T) {
10	// get some IDs
11	// for i := 0; i < 10; i++ {
12	//         id, err := NewID()
13	//         assert.Nil(t, err)
14	//         fmt.Println(id)
15	// }
16
17	idA, err := IDFromString("0fd85ddddf15aeec2d5d8b01b013dbca030a18d7")
18	assert.Nil(t, err)
19	assert.Equal(t, "0fd85ddddf15aeec2d5d8b01b013dbca030a18d7", idA.String())
20}
21
22func TestIDCmp(t *testing.T) {
23	idA, err := IDFromString("0fd85ddddf15aeec2d5d8b01b013dbca030a18d7")
24	assert.Nil(t, err)
25	idB, err := IDFromString("c48d8b53dbefb609ed4e94d386dd5b22efcb2c5b")
26	assert.Nil(t, err)
27	assert.True(t, !idA.Cmp(idB))
28}
29
30func TestIDDistance(t *testing.T) {
31	idA, err := IDFromString("0fd85ddddf15aeec2d5d8b01b013dbca030a18d7")
32	assert.Nil(t, err)
33	idB, err := IDFromString("c48d8b53dbefb609ed4e94d386dd5b22efcb2c5b")
34	assert.Nil(t, err)
35	assert.Equal(t, "cb55d68e04fa18e5c0131fd236ce80e8ecc1348c", idA.Distance(idB).String())
36}