0f5a2f4ef5
* enable more linters * Run gofmt -s * run goimports * run unconvert * fix naked return * fix misspellings * fix ineffectual assigments * fix missing declaration * don't use deprecated grpc.Errof * check errors in tests * run gofmt -w -r "assert.Nil(err) -> assert.NoError(err)" * fix directory permissions * don't use nil Context * simplify boolean expressions * use bytes.Equal instead of bytes.Compare * merge variable declarations, remove redundant returns * fix some golint errors * run goimports * handle more errors * delete empty TestMain * delete empty TestMain * ignore examples for now * fix lint errors * remove unused values * more fixes * run gofmt -w -s . * add more comments * fix naming * more lint fixes * try switching travis to go1.11 * fix unnecessary conversions * fix deprecated methods * use go1.10 and disable gofmt/goimports for now * switch to 1.10 * don't re-enable gofmt and goimports * switch covermode to atomic because of -race * gofmt
91 lines
2.0 KiB
Go
91 lines
2.0 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package node
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/grpc"
|
|
|
|
"storj.io/storj/internal/test"
|
|
"storj.io/storj/pkg/provider"
|
|
proto "storj.io/storj/protos/overlay"
|
|
)
|
|
|
|
var ctx = context.Background()
|
|
|
|
func TestLookup(t *testing.T) {
|
|
cases := []struct {
|
|
self proto.Node
|
|
to proto.Node
|
|
find proto.Node
|
|
expectedErr error
|
|
}{
|
|
{
|
|
self: proto.Node{Id: test.NewNodeID(t), Address: &proto.NodeAddress{Address: ":7070"}},
|
|
to: proto.Node{Id: test.NewNodeID(t), Address: &proto.NodeAddress{Address: ":8080"}},
|
|
find: proto.Node{Id: test.NewNodeID(t), Address: &proto.NodeAddress{Address: ":9090"}},
|
|
expectedErr: nil,
|
|
},
|
|
}
|
|
|
|
for _, v := range cases {
|
|
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", 8080))
|
|
assert.NoError(t, err)
|
|
|
|
srv, mock, err := newTestServer(ctx)
|
|
assert.NoError(t, err)
|
|
go srv.Serve(lis)
|
|
defer srv.Stop()
|
|
|
|
ca, err := provider.NewCA(ctx, 12, 4)
|
|
assert.NoError(t, err)
|
|
identity, err := ca.NewIdentity()
|
|
assert.NoError(t, err)
|
|
|
|
nc, err := NewNodeClient(identity, v.self)
|
|
assert.NoError(t, err)
|
|
|
|
_, err = nc.Lookup(ctx, v.to, v.find)
|
|
assert.Equal(t, v.expectedErr, err)
|
|
assert.Equal(t, 1, mock.queryCalled)
|
|
}
|
|
}
|
|
|
|
func newTestServer(ctx context.Context) (*grpc.Server, *mockNodeServer, error) {
|
|
ca, err := provider.NewCA(ctx, 12, 4)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
identity, err := ca.NewIdentity()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
identOpt, err := identity.ServerOption()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
grpcServer := grpc.NewServer(identOpt)
|
|
mn := &mockNodeServer{queryCalled: 0}
|
|
|
|
proto.RegisterNodesServer(grpcServer, mn)
|
|
|
|
return grpcServer, mn, nil
|
|
|
|
}
|
|
|
|
type mockNodeServer struct {
|
|
queryCalled int
|
|
}
|
|
|
|
func (mn *mockNodeServer) Query(ctx context.Context, req *proto.QueryRequest) (*proto.QueryResponse, error) {
|
|
mn.queryCalled++
|
|
return &proto.QueryResponse{}, nil
|
|
}
|