f932c3ced8
* wip overlay cache cmd * overlay cache add and list commands * use zap * fixes * simplify * add restrictions * cleanup * remove defaults for overlay management commands
87 lines
1.9 KiB
Go
87 lines
1.9 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
package overlay
|
|
|
|
import (
|
|
"context"
|
|
"net/url"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"storj.io/storj/pkg/kademlia"
|
|
)
|
|
|
|
func TestRun(t *testing.T) {
|
|
config := Config{}
|
|
bctx := context.Background()
|
|
kad := &kademlia.Kademlia{}
|
|
var key kademlia.CtxKey
|
|
|
|
cases := []struct {
|
|
testName string
|
|
testFunc func()
|
|
}{
|
|
{
|
|
testName: "Run with nil",
|
|
testFunc: func() {
|
|
err := config.Run(bctx, nil)
|
|
|
|
assert.Error(t, err)
|
|
assert.Equal(t, err.Error(), "overlay error: programmer error: kademlia responsibility unstarted")
|
|
},
|
|
},
|
|
{
|
|
testName: "Run with nil, pass pointer to Kademlia in context",
|
|
testFunc: func() {
|
|
ctx := context.WithValue(bctx, key, kad)
|
|
err := config.Run(ctx, nil)
|
|
|
|
assert.Error(t, err)
|
|
assert.Equal(t, err.Error(), "overlay error: database scheme not supported: ")
|
|
},
|
|
},
|
|
{
|
|
testName: "db scheme redis conn fail",
|
|
testFunc: func() {
|
|
ctx := context.WithValue(bctx, key, kad)
|
|
var config = Config{DatabaseURL: "redis://somedir/overlay.db/?db=1"}
|
|
err := config.Run(ctx, nil)
|
|
|
|
assert.Error(t, err)
|
|
assert.Equal(t, err.Error(), "redis error: ping failed%!(EXTRA *net.OpError=dial tcp: address somedir: missing port in address)")
|
|
},
|
|
},
|
|
{
|
|
testName: "db scheme bolt conn fail",
|
|
testFunc: func() {
|
|
ctx := context.WithValue(bctx, key, kad)
|
|
var config = Config{DatabaseURL: "bolt://somedir/overlay.db"}
|
|
err := config.Run(ctx, nil)
|
|
|
|
assert.Error(t, err)
|
|
assert.Equal(t, err.Error(), "open /overlay.db: permission denied")
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
t.Run(c.testName, func(t *testing.T) {
|
|
c.testFunc()
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestUrlPwd(t *testing.T) {
|
|
res := UrlPwd(nil)
|
|
|
|
assert.Equal(t, res, "")
|
|
|
|
uinfo := url.UserPassword("testUser", "testPassword")
|
|
|
|
uri := url.URL{User: uinfo}
|
|
|
|
res = UrlPwd(&uri)
|
|
|
|
assert.Equal(t, res, "testPassword")
|
|
}
|