mirror of
https://github.com/skidoodle/pastebin
synced 2026-04-28 11:17:41 +02:00
83 lines
1.7 KiB
Go
83 lines
1.7 KiB
Go
package store
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"go.etcd.io/bbolt"
|
|
)
|
|
|
|
func TestBoltStore(t *testing.T) {
|
|
dbPath := "test.db"
|
|
defer os.Remove(dbPath)
|
|
|
|
s, err := NewBoltStore(dbPath)
|
|
require.NoError(t, err)
|
|
defer s.Close()
|
|
|
|
t.Run("Set and Get", func(t *testing.T) {
|
|
id := "id1"
|
|
hash := "hash1"
|
|
content := "content1"
|
|
|
|
err := s.Set(id, hash, content)
|
|
assert.NoError(t, err)
|
|
|
|
p, exists, err := s.Get(id)
|
|
assert.NoError(t, err)
|
|
assert.True(t, exists)
|
|
assert.Equal(t, content, p.Content)
|
|
})
|
|
|
|
t.Run("GetIDByHash", func(t *testing.T) {
|
|
id := "id2"
|
|
hash := "hash2"
|
|
content := "content2"
|
|
|
|
s.Set(id, hash, content)
|
|
storedID, exists, err := s.GetIDByHash(hash)
|
|
assert.NoError(t, err)
|
|
assert.True(t, exists)
|
|
assert.Equal(t, id, storedID)
|
|
})
|
|
|
|
t.Run("Del", func(t *testing.T) {
|
|
id := "id3"
|
|
s.Set(id, "h3", "c3")
|
|
err := s.Del(id)
|
|
assert.NoError(t, err)
|
|
|
|
_, exists, _ := s.Get(id)
|
|
assert.False(t, exists)
|
|
})
|
|
|
|
t.Run("Cleanup", func(t *testing.T) {
|
|
s.Set("old", "oldhash", "oldcontent")
|
|
s.Cleanup(-time.Hour)
|
|
_, exists, _ := s.Get("old")
|
|
assert.False(t, exists)
|
|
})
|
|
|
|
t.Run("Cleanup Bad Data", func(t *testing.T) {
|
|
s.db.Update(func(tx *bbolt.Tx) error {
|
|
return tx.Bucket(pastesBucket).Put([]byte("bad"), []byte("invalid json"))
|
|
})
|
|
s.Cleanup(-time.Hour)
|
|
// Should skip without panic
|
|
})
|
|
}
|
|
|
|
func TestNewBoltStoreError(t *testing.T) {
|
|
// Use a directory name as file path to trigger error
|
|
err := os.Mkdir("testdir", 0755)
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll("testdir")
|
|
|
|
s, err := NewBoltStore("testdir")
|
|
assert.Error(t, err)
|
|
assert.Nil(t, s)
|
|
}
|