// Copyright (C) 2018 Storj Labs, Inc. // See LICENSE for copying information. package transport import ( "context" "testing" "github.com/stretchr/testify/assert" "storj.io/storj/pkg/pb" "storj.io/storj/pkg/provider" ) var ctx = context.Background() func TestDialNode(t *testing.T) { ca, err := provider.NewCA(ctx, 12, 4) assert.NoError(t, err) identity, err := ca.NewIdentity() assert.NoError(t, err) oc := Transport{ identity: identity, } // node.Address.Address == "" condition test node := pb.Node{ Id: "DUMMYID1", Address: &pb.NodeAddress{ Transport: pb.NodeTransport_TCP_TLS_GRPC, Address: "", }, } conn, err := oc.DialNode(ctx, &node) assert.Error(t, err) assert.Nil(t, conn) // node.Address == nil condition test node = pb.Node{ Id: "DUMMYID2", Address: nil, } conn, err = oc.DialNode(ctx, &node) assert.Error(t, err) assert.Nil(t, conn) // node is valid argument condition test node = pb.Node{ Id: "DUMMYID3", Address: &pb.NodeAddress{ Transport: pb.NodeTransport_TCP_TLS_GRPC, Address: "127.0.0.0:9000", }, } conn, err = oc.DialNode(ctx, &node) assert.NoError(t, err) assert.NotNil(t, conn) }