mirror of
https://github.com/gabehf/Koito.git
synced 2026-03-07 13:38:15 -08:00
36 lines
898 B
Go
36 lines
898 B
Go
package catalog_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/gabehf/koito/internal/catalog"
|
|
"github.com/gabehf/koito/internal/mbz"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestBackfillDuration(t *testing.T) {
|
|
setupTestDataWithMbzIDs(t)
|
|
|
|
ctx := context.Background()
|
|
mbzc := &mbz.MbzMockCaller{
|
|
Artists: mbzArtistData,
|
|
Releases: mbzReleaseData,
|
|
Tracks: mbzTrackData,
|
|
}
|
|
|
|
var err error
|
|
|
|
err = catalog.BackfillTrackDurationsFromMusicBrainz(context.Background(), store, &mbz.MbzErrorCaller{})
|
|
assert.NoError(t, err)
|
|
|
|
err = catalog.BackfillTrackDurationsFromMusicBrainz(ctx, store, mbzc)
|
|
assert.NoError(t, err)
|
|
|
|
count, err := store.Count(ctx, `
|
|
SELECT COUNT(*) FROM tracks_with_title WHERE title = $1 AND duration > 0
|
|
`, "Tokyo Calling")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 1, count, "track was not updated with duration")
|
|
}
|