c6f8144221
* wip * add separate `Process` tests for bolt and redis-backed overlay * more testing * fix gitignore * fix linter error * goimports goimports GOIMPORTS GoImPortS!!!! * fix port madness * forgot to add * add `mux` as handler and shorten context timeouts * gofreakingimports * fix comments * refactor test & add logger/monkit registry * debugging travis * add comment * Set redisAddress to empty string for bolt-test * travis experiment * refactoring tests * Merge remote-tracking branch 'upstream/master' into bolt-backed-overlay-cache
109 lines
2.4 KiB
Go
109 lines
2.4 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package overlay
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zeebo/errs"
|
|
"go.uber.org/zap"
|
|
"google.golang.org/grpc"
|
|
"gopkg.in/spacemonkeygo/monkit.v2"
|
|
|
|
"storj.io/storj/internal/test"
|
|
"storj.io/storj/pkg/process"
|
|
proto "storj.io/storj/protos/overlay" // naming proto to avoid confusion with this package
|
|
)
|
|
|
|
func newTestService(t *testing.T) Service {
|
|
return Service{
|
|
logger: zap.NewNop(),
|
|
metrics: monkit.Default,
|
|
}
|
|
}
|
|
|
|
func TestNewServer(t *testing.T) {
|
|
t.SkipNow()
|
|
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", 0))
|
|
assert.NoError(t, err)
|
|
|
|
srv := NewServer(nil, nil, nil, nil)
|
|
assert.NotNil(t, srv)
|
|
|
|
go srv.Serve(lis)
|
|
srv.Stop()
|
|
}
|
|
|
|
func TestNewClient(t *testing.T) {
|
|
// a := "35.232.202.229:8080"
|
|
// c, err := NewClient(&a, grpc.WithInsecure())
|
|
t.SkipNow()
|
|
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", 0))
|
|
assert.NoError(t, err)
|
|
srv := NewServer(nil, nil, nil, nil)
|
|
go srv.Serve(lis)
|
|
defer srv.Stop()
|
|
|
|
address := lis.Addr().String()
|
|
c, err := NewClient(&address, grpc.WithInsecure())
|
|
assert.NoError(t, err)
|
|
|
|
r, err := c.Lookup(context.Background(), &proto.LookupRequest{})
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, r)
|
|
}
|
|
|
|
func TestProcess_redis(t *testing.T) {
|
|
flag.Set("localPort", "0")
|
|
done := test.EnsureRedis(t)
|
|
defer done()
|
|
|
|
o := newTestService(t)
|
|
ctx, _ := context.WithTimeout(context.Background(), 500*time.Millisecond)
|
|
err := o.Process(ctx)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestProcess_bolt(t *testing.T) {
|
|
flag.Set("localPort", "0")
|
|
flag.Set("redisAddress", "")
|
|
boltdbPath, err := filepath.Abs("test_bolt.db")
|
|
assert.NoError(t, err)
|
|
|
|
if err != nil {
|
|
defer func() {
|
|
if err := os.Remove(boltdbPath); err != nil {
|
|
log.Println(errs.New("error while removing test bolt db: %s", err))
|
|
}
|
|
}()
|
|
}
|
|
|
|
flag.Set("boltdbPath", boltdbPath)
|
|
|
|
o := newTestService(t)
|
|
ctx, _ := context.WithTimeout(context.Background(), 500*time.Millisecond)
|
|
err = o.Process(ctx)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestProcess_error(t *testing.T) {
|
|
flag.Set("localPort", "0")
|
|
flag.Set("boltdbPath", "")
|
|
flag.Set("redisAddress", "")
|
|
|
|
o := newTestService(t)
|
|
ctx, _ := context.WithTimeout(context.Background(), 500*time.Millisecond)
|
|
err := o.Process(ctx)
|
|
assert.True(t, process.ErrUsage.Has(err))
|
|
}
|