5d20cf8829
* peertls: don't log errors for double close understood that this part of the code is undergoing heavy change right now, but just want to make sure this fix gets incorporated somewhere * git cleanup: node-id stuff * cleanup * rename identity_util.go * wip `CertificateAuthority` refactor * refactoring * gitignore update * wip * Merge remote-tracking branch 'storj/doubleclose' into node-id3 * storj/doubleclose: peertls: don't log errors for double close * add peertls tests & gomports * wip: + refactor + style changes + cleanup + [wip] add version to CA and identity configs + [wip] heavy client setup * refactor * wip: + refactor + style changes + add `CAConfig.Load` + add `CAConfig.Save` * wip: + add `LoadOrCreate` and `Create` to CA and Identity configs + add overwrite to CA and identity configs + heavy client setup + refactor + style changes + cleanup * wip * fixing things * fixing things * wip hc setup * hc setup: + refactor + bugfixing * improvements based on reveiw feedback * goimports * improvements: + responding to review feedback + refactor * feedback-based improvements * feedback-based improvements * feedback-based improvements * feedback-based improvements * feedback-based improvements * feedback-based improvements * cleanup * refactoring CA and Identity structs * Merge branch 'master' into node-id3 * move version field to setup config structs for CA and identity * fix typo * responding to revieiw feedback * responding to revieiw feedback * responding to revieiw feedback * responding to revieiw feedback * responding to revieiw feedback * responding to revieiw feedback * Merge branch 'master' into node-id3 * fix gateway setup finally * go imports * fix `FullCertificateAuthority.GenerateIdentity` * cleanup overlay tests * bugfixing * update ca/identity setup * go imports * fix peertls test copy/paste fail * responding to review feedback * setup tweaking * update farmer setup
140 lines
2.9 KiB
Go
140 lines
2.9 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information
|
|
|
|
package kademlia
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
pb "github.com/golang/protobuf/proto"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
proto "storj.io/storj/protos/overlay"
|
|
)
|
|
|
|
func TestLocal(t *testing.T) {
|
|
rt := createRT([]byte("AA"))
|
|
local := rt.Local()
|
|
assert.Equal(t, *rt.self, local)
|
|
}
|
|
|
|
func TestK(t *testing.T) {
|
|
rt := createRT([]byte("AA"))
|
|
k := rt.K()
|
|
assert.Equal(t, rt.bucketSize, k)
|
|
|
|
}
|
|
|
|
func TestCacheSize(t *testing.T) {
|
|
//TODO
|
|
rt := createRT([]byte("AA"))
|
|
expected := 0
|
|
result := rt.CacheSize()
|
|
assert.Equal(t, expected, result)
|
|
}
|
|
|
|
func TestGetBucket(t *testing.T) {
|
|
rt := createRT([]byte("AA"))
|
|
node := mockNode("AA")
|
|
node2 := mockNode("BB")
|
|
err := rt.addNode(node2)
|
|
assert.NoError(t, err)
|
|
|
|
cases := []struct {
|
|
nodeID string
|
|
expected *KBucket
|
|
ok bool
|
|
}{
|
|
{nodeID: node.Id,
|
|
expected: &KBucket{nodes: []*proto.Node{node, node2}},
|
|
ok: true,
|
|
},
|
|
{nodeID: node2.Id,
|
|
expected: &KBucket{nodes: []*proto.Node{node, node2}},
|
|
ok: true,
|
|
},
|
|
}
|
|
for i, v := range cases {
|
|
b, e := rt.GetBucket(node2.Id)
|
|
for j, w := range v.expected.nodes {
|
|
assert.True(t, pb.Equal(w, b.Nodes()[j]))
|
|
}
|
|
assert.Equal(t, v.ok, e)
|
|
fmt.Printf("error occured at index %d", i) //what's a better way to print the index?
|
|
}
|
|
}
|
|
|
|
func TestGetBuckets(t *testing.T) {
|
|
rt := createRT([]byte("AA"))
|
|
node := mockNode("AA")
|
|
node2 := mockNode("BB")
|
|
err := rt.addNode(node2)
|
|
assert.NoError(t, err)
|
|
expected := []*proto.Node{node, node2}
|
|
buckets, err := rt.GetBuckets()
|
|
assert.NoError(t, err)
|
|
for _, v := range buckets {
|
|
for j, w := range v.Nodes() {
|
|
assert.True(t, pb.Equal(expected[j], w))
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestFindNear(t *testing.T) {
|
|
rt := createRT([]byte("AA"))
|
|
node := mockNode("AA")
|
|
node2 := mockNode("BB")
|
|
err := rt.addNode(node2)
|
|
assert.NoError(t, err)
|
|
expected := []*proto.Node{node2}
|
|
nodes, err := rt.FindNear(StringToNodeID(node.Id), 1)
|
|
assert.NoError(t, err)
|
|
for i, v := range nodes {
|
|
assert.True(t, pb.Equal(expected[i], v))
|
|
}
|
|
}
|
|
|
|
func TestConnectionSuccess(t *testing.T) {
|
|
//TODO
|
|
}
|
|
|
|
func TestConnectionFailed(t *testing.T) {
|
|
//TODO
|
|
}
|
|
|
|
func TestSetBucketTimestamp(t *testing.T) {
|
|
id := []byte("AA")
|
|
idStr := string(id)
|
|
rt := createRT(id)
|
|
now := time.Now().UTC()
|
|
|
|
err := rt.createOrUpdateKBucket(id, now)
|
|
assert.NoError(t, err)
|
|
ti, err := rt.GetBucketTimestamp(idStr, nil)
|
|
assert.Equal(t, now, ti)
|
|
assert.NoError(t, err)
|
|
now = time.Now().UTC()
|
|
err = rt.SetBucketTimestamp(idStr, now)
|
|
assert.NoError(t, err)
|
|
ti, err = rt.GetBucketTimestamp(idStr, nil)
|
|
assert.Equal(t, now, ti)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestGetBucketTimestamp(t *testing.T) {
|
|
rt := createRT([]byte("AA"))
|
|
now := time.Now().UTC()
|
|
id := "AA"
|
|
err := rt.createOrUpdateKBucket([]byte(id), now)
|
|
assert.NoError(t, err)
|
|
ti, err := rt.GetBucketTimestamp(id, nil)
|
|
assert.Equal(t, now, ti)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestGetNodeRoutingTable(t *testing.T) {
|
|
//TODO
|
|
}
|