diff options
author | ethersphere <thesw@rm.eth> | 2018-06-20 20:06:27 +0800 |
---|---|---|
committer | ethersphere <thesw@rm.eth> | 2018-06-22 03:10:31 +0800 |
commit | e187711c6545487d4cac3701f0f506bb536234e2 (patch) | |
tree | d2f6150f70b84b36e49a449082aeda267b4b9046 /swarm/storage/localstore_test.go | |
parent | 574378edb50c907b532946a1d4654dbd6701b20a (diff) | |
download | dexon-e187711c6545487d4cac3701f0f506bb536234e2.tar.gz dexon-e187711c6545487d4cac3701f0f506bb536234e2.tar.zst dexon-e187711c6545487d4cac3701f0f506bb536234e2.zip |
swarm: network rewrite merge
Diffstat (limited to 'swarm/storage/localstore_test.go')
-rw-r--r-- | swarm/storage/localstore_test.go | 118 |
1 files changed, 118 insertions, 0 deletions
diff --git a/swarm/storage/localstore_test.go b/swarm/storage/localstore_test.go new file mode 100644 index 000000000..2bb81efa3 --- /dev/null +++ b/swarm/storage/localstore_test.go @@ -0,0 +1,118 @@ +// Copyright 2018 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 ( + "io/ioutil" + "os" + "testing" +) + +var ( + hashfunc = MakeHashFunc(DefaultHash) +) + +// tests that the content address validator correctly checks the data +// tests that resource update chunks are passed through content address validator +// the test checking the resouce update validator internal correctness is found in resource_test.go +func TestValidator(t *testing.T) { + // set up localstore + datadir, err := ioutil.TempDir("", "storage-testvalidator") + if err != nil { + t.Fatal(err) + } + defer os.RemoveAll(datadir) + + params := NewDefaultLocalStoreParams() + params.Init(datadir) + store, err := NewLocalStore(params, nil) + if err != nil { + t.Fatal(err) + } + + // check puts with no validators, both succeed + chunks := GenerateRandomChunks(259, 2) + goodChunk := chunks[0] + badChunk := chunks[1] + copy(badChunk.SData, goodChunk.SData) + + PutChunks(store, goodChunk, badChunk) + if err := goodChunk.GetErrored(); err != nil { + t.Fatalf("expected no error on good content address chunk in spite of no validation, but got: %s", err) + } + if err := badChunk.GetErrored(); err != nil { + t.Fatalf("expected no error on bad content address chunk in spite of no validation, but got: %s", err) + } + + // add content address validator and check puts + // bad should fail, good should pass + store.Validators = append(store.Validators, NewContentAddressValidator(hashfunc)) + chunks = GenerateRandomChunks(DefaultChunkSize, 2) + goodChunk = chunks[0] + badChunk = chunks[1] + copy(badChunk.SData, goodChunk.SData) + + PutChunks(store, goodChunk, badChunk) + if err := goodChunk.GetErrored(); err != nil { + t.Fatalf("expected no error on good content address chunk with content address validator only, but got: %s", err) + } + if err := badChunk.GetErrored(); err == nil { + t.Fatal("expected error on bad content address chunk with content address validator only, but got nil") + } + + // append a validator that always denies + // bad should fail, good should pass, + var negV boolTestValidator + store.Validators = append(store.Validators, negV) + + chunks = GenerateRandomChunks(DefaultChunkSize, 2) + goodChunk = chunks[0] + badChunk = chunks[1] + copy(badChunk.SData, goodChunk.SData) + + PutChunks(store, goodChunk, badChunk) + if err := goodChunk.GetErrored(); err != nil { + t.Fatalf("expected no error on good content address chunk with content address validator only, but got: %s", err) + } + if err := badChunk.GetErrored(); err == nil { + t.Fatal("expected error on bad content address chunk with content address validator only, but got nil") + } + + // append a validator that always approves + // all shall pass + var posV boolTestValidator = true + store.Validators = append(store.Validators, posV) + + chunks = GenerateRandomChunks(DefaultChunkSize, 2) + goodChunk = chunks[0] + badChunk = chunks[1] + copy(badChunk.SData, goodChunk.SData) + + PutChunks(store, goodChunk, badChunk) + if err := goodChunk.GetErrored(); err != nil { + t.Fatalf("expected no error on good content address chunk with content address validator only, but got: %s", err) + } + if err := badChunk.GetErrored(); err != nil { + t.Fatalf("expected no error on bad content address chunk with content address validator only, but got: %s", err) + } +} + +type boolTestValidator bool + +func (self boolTestValidator) Validate(addr Address, data []byte) bool { + return bool(self) +} |