80727ae90b
* adds netstate rpc server pagination, mocks pagination in test/util.go * updates ns client example, combines ns client and server test to netstate_test, adds pagination to bolt client * better organizes netstate test calls * wip breaking netstate test into smaller tests * wip modularizing netstate tests * adds some test panics * wip netstate test attempts * testing bug in netstate TestDeleteAuth * wip fixes global variable problem, still issues with list * wip fixes get request params and args * fixes bug in path when using MakePointers helper fn * updates mockdb list func, adds test, changes Limit to int * fixes merge conflicts * fixes broken tests from merge * remove unnecessary PointerEntry struct * removes error when Get returns nil value from boltdb * breaks boltdb client tests into smaller tests * renames AssertNoErr test helper to HandleErr * adds StartingKey and Limit parameters to redis list func, adds beginning of redis tests * adds helper func for mockdb List function * if no starting key provided for netstate List, the first value in storage will be used * adds basic pagination for redis List function, adds tests * adds list limit to call in overlay/server.go * streamlines/fixes some nits from review * removes use of obsolete EncryptedUnencryptedSize * uses MockKeyValueStore instead of redis instance in redis client test * changes test to expect nil returned for getting missing key * remove error from `KeyValueStore#Get` * fix bolt test * Merge pull request #1 from bryanchriswhite/nat-pagination remove error from `KeyValueStore#Get` * adds Get returning error back to KeyValueStore interface and affected clients * trying to appease travis: returns errors in Get calls in overlay/cache and cache_test * handles redis get error when no key found
84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package redis
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"storj.io/storj/internal/test"
|
|
"storj.io/storj/storage"
|
|
)
|
|
|
|
type RedisClientTest struct {
|
|
*testing.T
|
|
c storage.KeyValueStore
|
|
}
|
|
|
|
func NewRedisClientTest(t *testing.T) *RedisClientTest {
|
|
kv := make(test.KvStore)
|
|
c := test.NewMockKeyValueStore(kv)
|
|
return &RedisClientTest{
|
|
T: t,
|
|
c: c,
|
|
}
|
|
}
|
|
|
|
func (rt *RedisClientTest) Close() {
|
|
rt.c.Close()
|
|
}
|
|
|
|
func (rt *RedisClientTest) HandleErr(err error, msg string) {
|
|
rt.Error(msg)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
panic(msg)
|
|
}
|
|
|
|
func TestListWithoutStartKey(t *testing.T) {
|
|
rt := NewRedisClientTest(t)
|
|
defer rt.Close()
|
|
|
|
if err := rt.c.Put(storage.Key([]byte("path/1")), []byte("pointer1")); err != nil {
|
|
rt.HandleErr(err, "Failed to put")
|
|
}
|
|
if err := rt.c.Put(storage.Key([]byte("path/2")), []byte("pointer2")); err != nil {
|
|
rt.HandleErr(err, "Failed to put")
|
|
}
|
|
if err := rt.c.Put(storage.Key([]byte("path/3")), []byte("pointer3")); err != nil {
|
|
rt.HandleErr(err, "Failed to put")
|
|
}
|
|
|
|
_, err := rt.c.List(nil, storage.Limit(3))
|
|
if err != nil {
|
|
rt.HandleErr(err, "Failed to list")
|
|
}
|
|
}
|
|
|
|
func TestListWithStartKey(t *testing.T) {
|
|
rt := NewRedisClientTest(t)
|
|
defer rt.Close()
|
|
|
|
if err := rt.c.Put(storage.Key([]byte("path/1")), []byte("pointer1")); err != nil {
|
|
rt.HandleErr(err, "Failed to put")
|
|
}
|
|
if err := rt.c.Put(storage.Key([]byte("path/2")), []byte("pointer2")); err != nil {
|
|
rt.HandleErr(err, "Failed to put")
|
|
}
|
|
if err := rt.c.Put(storage.Key([]byte("path/3")), []byte("pointer3")); err != nil {
|
|
rt.HandleErr(err, "Failed to put")
|
|
}
|
|
if err := rt.c.Put(storage.Key([]byte("path/4")), []byte("pointer4")); err != nil {
|
|
rt.HandleErr(err, "Failed to put")
|
|
}
|
|
if err := rt.c.Put(storage.Key([]byte("path/5")), []byte("pointer5")); err != nil {
|
|
rt.HandleErr(err, "Failed to put")
|
|
}
|
|
|
|
_, err := rt.c.List([]byte("path/2"), storage.Limit(2))
|
|
if err != nil {
|
|
rt.HandleErr(err, "Failed to list")
|
|
}
|
|
}
|