go-ethereum/swarm/storage/memstore_test.go

159 lines
3.6 KiB
Go
Raw Normal View History

// Copyright 2016 The go-ethereum Authors
// This file is part of the go-ethereum library.
//
// The go-ethereum library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The go-ethereum library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
package storage
import (
"context"
"testing"
2018-06-20 07:06:27 -05:00
"github.com/ethereum/go-ethereum/swarm/log"
)
2018-06-20 07:06:27 -05:00
func newTestMemStore() *MemStore {
storeparams := NewDefaultStoreParams()
return NewMemStore(storeparams, nil)
}
func testMemStoreRandom(n int, t *testing.T) {
2018-06-20 07:06:27 -05:00
m := newTestMemStore()
defer m.Close()
testStoreRandom(m, n, t)
2018-06-20 07:06:27 -05:00
}
func testMemStoreCorrect(n int, t *testing.T) {
2018-06-20 07:06:27 -05:00
m := newTestMemStore()
defer m.Close()
testStoreCorrect(m, n, t)
2018-06-20 07:06:27 -05:00
}
func TestMemStoreRandom_1(t *testing.T) {
testMemStoreRandom(1, t)
}
2018-06-20 07:06:27 -05:00
func TestMemStoreCorrect_1(t *testing.T) {
testMemStoreCorrect(1, t)
}
func TestMemStoreRandom_1k(t *testing.T) {
testMemStoreRandom(1000, t)
}
func TestMemStoreCorrect_1k(t *testing.T) {
testMemStoreCorrect(100, t)
}
func TestMemStoreNotFound(t *testing.T) {
2018-06-20 07:06:27 -05:00
m := newTestMemStore()
defer m.Close()
_, err := m.Get(context.TODO(), ZeroAddr)
2018-06-20 07:06:27 -05:00
if err != ErrChunkNotFound {
t.Errorf("Expected ErrChunkNotFound, got %v", err)
}
}
func benchmarkMemStorePut(n int, b *testing.B) {
2018-06-20 07:06:27 -05:00
m := newTestMemStore()
defer m.Close()
benchmarkStorePut(m, n, b)
2018-06-20 07:06:27 -05:00
}
func benchmarkMemStoreGet(n int, b *testing.B) {
2018-06-20 07:06:27 -05:00
m := newTestMemStore()
defer m.Close()
benchmarkStoreGet(m, n, b)
2018-06-20 07:06:27 -05:00
}
func BenchmarkMemStorePut_500(b *testing.B) {
benchmarkMemStorePut(500, b)
2018-06-20 07:06:27 -05:00
}
func BenchmarkMemStoreGet_500(b *testing.B) {
benchmarkMemStoreGet(500, b)
2018-06-20 07:06:27 -05:00
}
func TestMemStoreAndLDBStore(t *testing.T) {
ldb, cleanup := newLDBStore(t)
ldb.setCapacity(4000)
defer cleanup()
cacheCap := 200
memStore := NewMemStore(NewStoreParams(4000, 200, nil, nil), nil)
2018-06-20 07:06:27 -05:00
tests := []struct {
n int // number of chunks to push to memStore
chunkSize int64 // size of chunk (by default in Swarm - 4096)
2018-06-20 07:06:27 -05:00
}{
{
n: 1,
chunkSize: 4096,
},
{
n: 101,
2018-06-20 07:06:27 -05:00
chunkSize: 4096,
},
{
n: 501,
chunkSize: 4096,
},
{
n: 1100,
2018-06-20 07:06:27 -05:00
chunkSize: 4096,
},
}
for i, tt := range tests {
log.Info("running test", "idx", i, "tt", tt)
var chunks []Chunk
2018-06-20 07:06:27 -05:00
for i := 0; i < tt.n; i++ {
c := GenerateRandomChunk(tt.chunkSize)
2018-06-20 07:06:27 -05:00
chunks = append(chunks, c)
}
for i := 0; i < tt.n; i++ {
err := ldb.Put(context.TODO(), chunks[i])
if err != nil {
t.Fatal(err)
}
err = memStore.Put(context.TODO(), chunks[i])
if err != nil {
t.Fatal(err)
}
2018-06-20 07:06:27 -05:00
if got := memStore.cache.Len(); got > cacheCap {
t.Fatalf("expected to get cache capacity less than %v, but got %v", cacheCap, got)
}
}
for i := 0; i < tt.n; i++ {
_, err := memStore.Get(context.TODO(), chunks[i].Address())
2018-06-20 07:06:27 -05:00
if err != nil {
if err == ErrChunkNotFound {
_, err := ldb.Get(context.TODO(), chunks[i].Address())
2018-06-20 07:06:27 -05:00
if err != nil {
t.Fatalf("couldn't get chunk %v from ldb, got error: %v", i, err)
}
} else {
t.Fatalf("got error from memstore: %v", err)
}
}
}
}
}