v2

store/boltdb_test.go

@@ -0,0 +1,97 @@
+package store
+
+import (
+	"encoding/json"
+	"log/slog"
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"go.etcd.io/bbolt"
+)
+
+// setupTestDB creates a temporary BoltDB instance for testing and returns a cleanup function.
+func setupTestDB(t *testing.T) (*BoltStore, func()) {
+	dir := t.TempDir()
+	dbPath := filepath.Join(dir, "test.db")
+	store, err := NewBoltStore(dbPath)
+	assert.NoError(t, err)
+
+	cleanup := func() {
+		if err := store.Close(); err != nil {
+			slog.Error("failed to close store", "error", err)
+		}
+		if err := os.RemoveAll(dir); err != nil {
+			slog.Error("failed to remove temp directory", "dir", dir, "error", err)
+		}
+	}
+
+	return store, cleanup
+}
+
+// TestBoltStore_SetGetDel tests the Set, Get, and Del methods of the BoltStore.
+func TestBoltStore_SetGetDel(t *testing.T) {
+	store, cleanup := setupTestDB(t)
+	defer cleanup()
+
+	// Test Set and Get
+	err := store.Set("key1", "value1")
+	assert.NoError(t, err)
+
+	val, ok, err := store.Get("key1")
+	assert.NoError(t, err)
+	assert.True(t, ok)
+	assert.Equal(t, "value1", val)
+
+	// Test Get non-existent key
+	_, ok, err = store.Get("non-existent")
+	assert.NoError(t, err)
+	assert.False(t, ok)
+
+	// Test Delete
+	err = store.Del("key1")
+	assert.NoError(t, err)
+
+	_, ok, err = store.Get("key1")
+	assert.NoError(t, err)
+	assert.False(t, ok)
+}
+
+// TestBoltStore_Cleanup tests the Cleanup method of the BoltStore.
+func TestBoltStore_Cleanup(t *testing.T) {
+	store, cleanup := setupTestDB(t)
+	defer cleanup()
+
+	// Set one fresh and one stale paste
+	err := store.Set("fresh", "content")
+	assert.NoError(t, err)
+
+	err = store.db.Update(func(tx *bbolt.Tx) error {
+		b := tx.Bucket(pastesBucket)
+		stalePaste := Paste{
+			Content:   "stale",
+			CreatedAt: time.Now().Add(-2 * time.Hour),
+		}
+		encoded, err := json.Marshal(stalePaste)
+		if err != nil {
+			return err
+		}
+		return b.Put([]byte("stale"), encoded)
+	})
+	assert.NoError(t, err)
+
+	// Cleanup pastes older than 1 hour
+	store.Cleanup(1 * time.Hour)
+
+	// Check that fresh paste still exists
+	_, ok, err := store.Get("fresh")
+	assert.NoError(t, err)
+	assert.True(t, ok)
+
+	// Check that stale paste was deleted
+	_, ok, err = store.Get("stale")
+	assert.NoError(t, err)
+	assert.False(t, ok)
+}