45a3c2e974
* pkg/provider: with pkg/provider merged, make a single heavy client binary and deprecate old services * add setup to gw binary too * captplanet: output what addresses everything is listening on * revert peertls/io_util changes * define config flag across all commands * use trimsuffix
156 lines
3.6 KiB
Go
156 lines
3.6 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package overlay
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/grpc"
|
|
|
|
"storj.io/storj/pkg/peertls"
|
|
proto "storj.io/storj/protos/overlay" // naming proto to avoid confusion with this package
|
|
)
|
|
|
|
func TestNewServer(t *testing.T) {
|
|
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", 0))
|
|
assert.NoError(t, err)
|
|
|
|
srv := newMockServer()
|
|
assert.NotNil(t, srv)
|
|
|
|
go srv.Serve(lis)
|
|
srv.Stop()
|
|
}
|
|
|
|
func TestNewClient_CreateTLS(t *testing.T) {
|
|
var err error
|
|
|
|
tmpPath, err := ioutil.TempDir("", "TestNewClient")
|
|
assert.NoError(t, err)
|
|
defer os.RemoveAll(tmpPath)
|
|
|
|
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", 0))
|
|
assert.NoError(t, err)
|
|
|
|
basePath := filepath.Join(tmpPath, "TestNewClient_CreateTLS")
|
|
srv, tlsOpts := newMockTLSServer(t, basePath, true)
|
|
go srv.Serve(lis)
|
|
defer srv.Stop()
|
|
|
|
address := lis.Addr().String()
|
|
c, err := NewClient(address, tlsOpts.DialOption())
|
|
assert.NoError(t, err)
|
|
|
|
r, err := c.Lookup(context.Background(), &proto.LookupRequest{})
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, r)
|
|
}
|
|
|
|
func TestNewClient_LoadTLS(t *testing.T) {
|
|
var err error
|
|
|
|
tmpPath, err := ioutil.TempDir("", "TestNewClient")
|
|
assert.NoError(t, err)
|
|
defer os.RemoveAll(tmpPath)
|
|
|
|
basePath := filepath.Join(tmpPath, "TestNewClient_LoadTLS")
|
|
_, err = peertls.NewTLSFileOptions(
|
|
basePath,
|
|
basePath,
|
|
true,
|
|
false,
|
|
)
|
|
|
|
assert.NoError(t, err)
|
|
|
|
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", 0))
|
|
assert.NoError(t, err)
|
|
// NB: do NOT create a cert, it should be loaded from disk
|
|
srv, tlsOpts := newMockTLSServer(t, basePath, false)
|
|
|
|
go srv.Serve(lis)
|
|
defer srv.Stop()
|
|
|
|
address := lis.Addr().String()
|
|
c, err := NewClient(address, tlsOpts.DialOption())
|
|
assert.NoError(t, err)
|
|
|
|
r, err := c.Lookup(context.Background(), &proto.LookupRequest{})
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, r)
|
|
}
|
|
|
|
func TestNewClient_IndependentTLS(t *testing.T) {
|
|
var err error
|
|
|
|
tmpPath, err := ioutil.TempDir("", "TestNewClient_IndependentTLS")
|
|
assert.NoError(t, err)
|
|
defer os.RemoveAll(tmpPath)
|
|
|
|
clientBasePath := filepath.Join(tmpPath, "client")
|
|
serverBasePath := filepath.Join(tmpPath, "server")
|
|
|
|
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", 0))
|
|
assert.NoError(t, err)
|
|
srv, _ := newMockTLSServer(t, serverBasePath, true)
|
|
|
|
go srv.Serve(lis)
|
|
defer srv.Stop()
|
|
|
|
clientTLSOps, err := peertls.NewTLSFileOptions(
|
|
clientBasePath,
|
|
clientBasePath,
|
|
true,
|
|
false,
|
|
)
|
|
|
|
assert.NoError(t, err)
|
|
|
|
address := lis.Addr().String()
|
|
c, err := NewClient(address, clientTLSOps.DialOption())
|
|
assert.NoError(t, err)
|
|
|
|
r, err := c.Lookup(context.Background(), &proto.LookupRequest{})
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, r)
|
|
}
|
|
|
|
func newMockServer(opts ...grpc.ServerOption) *grpc.Server {
|
|
grpcServer := grpc.NewServer(opts...)
|
|
proto.RegisterOverlayServer(grpcServer, &MockOverlay{})
|
|
|
|
return grpcServer
|
|
}
|
|
|
|
func newMockTLSServer(t *testing.T, tlsBasePath string, create bool) (*grpc.Server, *peertls.TLSFileOptions) {
|
|
tlsOpts, err := peertls.NewTLSFileOptions(
|
|
tlsBasePath,
|
|
tlsBasePath,
|
|
create,
|
|
false,
|
|
)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, tlsOpts)
|
|
|
|
grpcServer := newMockServer(tlsOpts.ServerOption())
|
|
return grpcServer, tlsOpts
|
|
}
|
|
|
|
type MockOverlay struct{}
|
|
|
|
func (o *MockOverlay) FindStorageNodes(ctx context.Context, req *proto.FindStorageNodesRequest) (*proto.FindStorageNodesResponse, error) {
|
|
return &proto.FindStorageNodesResponse{}, nil
|
|
}
|
|
|
|
func (o *MockOverlay) Lookup(ctx context.Context, req *proto.LookupRequest) (*proto.LookupResponse, error) {
|
|
return &proto.LookupResponse{}, nil
|
|
}
|