mirror of https://github.com/gabehf/Koito.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
3.3 KiB
97 lines
3.3 KiB
package psql_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/gabehf/koito/internal/db"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetTopArtistsPaginated(t *testing.T) {
|
|
testDataForTopItems(t)
|
|
ctx := context.Background()
|
|
|
|
// Test valid
|
|
resp, err := store.GetTopArtistsPaginated(ctx, db.GetItemsOpts{Period: db.PeriodAllTime})
|
|
require.NoError(t, err)
|
|
require.Len(t, resp.Items, 4)
|
|
assert.Equal(t, int64(4), resp.TotalCount)
|
|
assert.Equal(t, "Artist One", resp.Items[0].Name)
|
|
assert.Equal(t, "Artist Two", resp.Items[1].Name)
|
|
assert.Equal(t, "Artist Three", resp.Items[2].Name)
|
|
assert.Equal(t, "Artist Four", resp.Items[3].Name)
|
|
|
|
// Test pagination
|
|
resp, err = store.GetTopArtistsPaginated(ctx, db.GetItemsOpts{Limit: 1, Page: 2, Period: db.PeriodAllTime})
|
|
require.NoError(t, err)
|
|
require.Len(t, resp.Items, 1)
|
|
assert.Equal(t, "Artist Two", resp.Items[0].Name)
|
|
|
|
// Test page out of range
|
|
resp, err = store.GetTopArtistsPaginated(ctx, db.GetItemsOpts{Limit: 1, Page: 10, Period: db.PeriodAllTime})
|
|
require.NoError(t, err)
|
|
assert.Empty(t, resp.Items)
|
|
assert.False(t, resp.HasNextPage)
|
|
|
|
// Test invalid inputs
|
|
_, err = store.GetTopArtistsPaginated(ctx, db.GetItemsOpts{Limit: -1, Page: 0})
|
|
assert.Error(t, err)
|
|
|
|
_, err = store.GetTopArtistsPaginated(ctx, db.GetItemsOpts{Limit: 1, Page: -1})
|
|
assert.Error(t, err)
|
|
|
|
// Test specify period
|
|
resp, err = store.GetTopArtistsPaginated(ctx, db.GetItemsOpts{Period: db.PeriodDay})
|
|
require.NoError(t, err)
|
|
require.Len(t, resp.Items, 0) // empty
|
|
assert.Equal(t, int64(0), resp.TotalCount)
|
|
// should default to PeriodDay
|
|
resp, err = store.GetTopArtistsPaginated(ctx, db.GetItemsOpts{})
|
|
require.NoError(t, err)
|
|
require.Len(t, resp.Items, 0) // empty
|
|
assert.Equal(t, int64(0), resp.TotalCount)
|
|
|
|
resp, err = store.GetTopArtistsPaginated(ctx, db.GetItemsOpts{Period: db.PeriodWeek})
|
|
require.NoError(t, err)
|
|
require.Len(t, resp.Items, 1)
|
|
assert.Equal(t, int64(1), resp.TotalCount)
|
|
assert.Equal(t, "Artist Four", resp.Items[0].Name)
|
|
|
|
resp, err = store.GetTopArtistsPaginated(ctx, db.GetItemsOpts{Period: db.PeriodMonth})
|
|
require.NoError(t, err)
|
|
require.Len(t, resp.Items, 2)
|
|
assert.Equal(t, int64(2), resp.TotalCount)
|
|
assert.Equal(t, "Artist Three", resp.Items[0].Name)
|
|
assert.Equal(t, "Artist Four", resp.Items[1].Name)
|
|
|
|
resp, err = store.GetTopArtistsPaginated(ctx, db.GetItemsOpts{Period: db.PeriodYear})
|
|
require.NoError(t, err)
|
|
require.Len(t, resp.Items, 3)
|
|
assert.Equal(t, int64(3), resp.TotalCount)
|
|
assert.Equal(t, "Artist Two", resp.Items[0].Name)
|
|
assert.Equal(t, "Artist Three", resp.Items[1].Name)
|
|
assert.Equal(t, "Artist Four", resp.Items[2].Name)
|
|
|
|
// Test specify dates
|
|
|
|
testDataAbsoluteListenTimes(t)
|
|
|
|
resp, err = store.GetTopArtistsPaginated(ctx, db.GetItemsOpts{Year: 2023})
|
|
require.NoError(t, err)
|
|
require.Len(t, resp.Items, 1)
|
|
assert.Equal(t, int64(1), resp.TotalCount)
|
|
assert.Equal(t, "Artist One", resp.Items[0].Name)
|
|
|
|
resp, err = store.GetTopArtistsPaginated(ctx, db.GetItemsOpts{Month: 6, Year: 2024})
|
|
require.NoError(t, err)
|
|
require.Len(t, resp.Items, 1)
|
|
assert.Equal(t, int64(1), resp.TotalCount)
|
|
assert.Equal(t, "Artist Two", resp.Items[0].Name)
|
|
|
|
// invalid, year required with month
|
|
_, err = store.GetTopArtistsPaginated(ctx, db.GetItemsOpts{Month: 10})
|
|
require.Error(t, err)
|
|
}
|