mrleo1nid
f1528de948
ci / contract / contract (сгенерированный код совпадает с openapi.yaml) (push) Successful in 1m17s
ci / ts / ts (packages + web) (push) Successful in 3m13s
ci / api / api (push) Successful in 59s
ci / mobile / mobile (Metro-бандл Android) (push) Successful in 1m31s
ci / db / db (push) Successful in 2m50s
59 lines
2.5 KiB
Go
59 lines
2.5 KiB
Go
//go:build integration
|
|
|
|
package dbtest
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"pipu/api/internal/library"
|
|
"pipu/api/internal/platform/db"
|
|
)
|
|
|
|
func TestSourceCoverStats(t *testing.T) {
|
|
pool := connect(t)
|
|
ctx := context.Background()
|
|
require.NoError(t, db.Migrate(ctx, pool))
|
|
store := library.NewStore(pool)
|
|
src, err := store.SaveSource(ctx, library.Source{ID: uuid.New(), Name: "cover stats", Type: library.SourceLocal, Status: "mounted", TagsFirst: true, NFSOptions: "ro"})
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() {
|
|
require.NoError(t, store.DeleteSource(ctx, src.ID))
|
|
require.NoError(t, store.CleanupOrphans(ctx))
|
|
})
|
|
root := t.TempDir()
|
|
scanner := library.NewScanner(store, fakeProber{}, library.NewProgressBoard(), slog.New(slog.DiscardHandler))
|
|
for _, rel := range []string{"covered/StatsArtist - First.flac", "covered/StatsArtist - Second.flac", "external/StatsArtist - Third.flac", "pending/StatsArtist - Fourth.flac"} {
|
|
file := filepath.Join(root, filepath.FromSlash(rel))
|
|
require.NoError(t, os.MkdirAll(filepath.Dir(file), 0o755))
|
|
require.NoError(t, os.WriteFile(file, []byte(rel), 0o644))
|
|
_, err := scanner.ScanFile(ctx, src, root, rel)
|
|
require.NoError(t, err)
|
|
}
|
|
_, err = pool.Exec(ctx, `UPDATE albums SET cover_path = 'test-key', cover_checked_at = now() WHERE id IN (
|
|
SELECT t.album_id FROM tracks t JOIN media_files f ON f.track_id=t.id WHERE f.source_id=$1 AND f.rel_path LIKE 'covered/%')`, src.ID)
|
|
require.NoError(t, err)
|
|
_, err = pool.Exec(ctx, `UPDATE albums SET genius_cover_url = 'https://example.org/cover.jpg', cover_checked_at = now() WHERE id IN (
|
|
SELECT t.album_id FROM tracks t JOIN media_files f ON f.track_id=t.id WHERE f.source_id=$1 AND f.rel_path LIKE 'external/%')`, src.ID)
|
|
require.NoError(t, err)
|
|
stats, err := store.SourceCoverStats(ctx, src.ID)
|
|
require.NoError(t, err)
|
|
require.Equal(t, library.CoverStats{FilesTotal: 4, FilesWithCover: 2, AlbumsPending: 1}, stats)
|
|
// Removed and excluded files leave the denominator as well as the numerator.
|
|
_, err = pool.Exec(ctx, `UPDATE media_files SET missing_since=$2 WHERE source_id=$1 AND rel_path LIKE '%Second.flac'`, src.ID, time.Now())
|
|
require.NoError(t, err)
|
|
src.ExcludedDirs = []string{"external"}
|
|
_, err = store.SaveSource(ctx, src)
|
|
require.NoError(t, err)
|
|
stats, err = store.SourceCoverStats(ctx, src.ID)
|
|
require.NoError(t, err)
|
|
require.Equal(t, library.CoverStats{FilesTotal: 2, FilesWithCover: 1, AlbumsPending: 1}, stats)
|
|
}
|