forked from External/grumble
105 lines
1.8 KiB
Go
105 lines
1.8 KiB
Go
// Copyright (c) 2011 The Grumble Authors
|
|
// The use of this source code is goverened by a BSD-style
|
|
// license that can be found in the LICENSE-file.
|
|
|
|
package blobstore
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/sha1"
|
|
"encoding/hex"
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestStoreRetrieve(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "blobstore")
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
bs := OpenBlobStore(dir)
|
|
|
|
data := []byte{0xde, 0xad, 0xca, 0xfe, 0xba, 0xbe, 0xbe, 0xef}
|
|
|
|
key, err := bs.Put(data)
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
|
|
recv, err := bs.Get(key)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !bytes.Equal(recv, data) {
|
|
t.Errorf("stored data and retrieved data does not match: %v vs. %v", recv, data)
|
|
}
|
|
}
|
|
|
|
func TestReadNonExistantKey(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "blobstore")
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
bs := OpenBlobStore(dir)
|
|
|
|
h := sha1.New()
|
|
h.Write([]byte{0x42})
|
|
key := hex.EncodeToString(h.Sum(nil))
|
|
buf, err := bs.Get(key)
|
|
if err != ErrNoSuchKey {
|
|
t.Error("Expected no such key %v, found it anyway. (buf=%v, err=%v)", key, buf, err)
|
|
return
|
|
}
|
|
}
|
|
|
|
func TestReadInvalidKeyLength(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "blobstore")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
bs := OpenBlobStore(dir)
|
|
|
|
key := ""
|
|
for i := 0; i < 5; i++ {
|
|
key += "0"
|
|
}
|
|
|
|
_, err = bs.Get(key)
|
|
if err != ErrBadKey {
|
|
t.Error("Expected invalid key for %v, got %v", key, err)
|
|
return
|
|
}
|
|
}
|
|
|
|
func TestReadBadKeyNonHex(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "blobstore")
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
bs := OpenBlobStore(dir)
|
|
|
|
key := ""
|
|
for i := 0; i < 40; i++ {
|
|
key += "i"
|
|
}
|
|
|
|
_, err = bs.Get(key)
|
|
if err != ErrBadKey {
|
|
t.Errorf("Expected bad key for %v, got %v", key, err)
|
|
return
|
|
}
|
|
}
|