mirror of
https://github.com/derfenix/webarchive.git
synced 2026-03-11 21:35:34 +03:00
61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
package badger
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap/zaptest"
|
|
|
|
"github.com/derfenix/webarchive/entity"
|
|
)
|
|
|
|
func TestSite(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
if testing.Short() {
|
|
t.Skip("skip db test")
|
|
}
|
|
|
|
ctx := context.Background()
|
|
|
|
tempDir, err := os.MkdirTemp(os.TempDir(), "badger_test")
|
|
require.NoError(t, err)
|
|
|
|
t.Cleanup(func() {
|
|
assert.NoError(t, os.RemoveAll(tempDir))
|
|
})
|
|
|
|
log := zaptest.NewLogger(t)
|
|
|
|
db, err := NewBadger(tempDir, log.Named("db"))
|
|
require.NoError(t, err)
|
|
|
|
siteRepo, err := NewPage(db, nil)
|
|
require.NoError(t, err)
|
|
|
|
t.Run("base path", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
site := entity.NewPage("https://google.com", "Save all google", entity.FormatPDF, entity.FormatSingleFile)
|
|
site.Created = site.Created.Truncate(time.Microsecond)
|
|
|
|
err := siteRepo.Save(ctx, site)
|
|
require.NoError(t, err)
|
|
|
|
storedSite, err := siteRepo.Get(ctx, site.ID)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, site, storedSite)
|
|
|
|
all, err := siteRepo.ListAll(ctx)
|
|
require.NoError(t, err)
|
|
require.Len(t, all, 1)
|
|
|
|
assert.Equal(t, site, all[0])
|
|
})
|
|
}
|