36 lines
819 B
Go
36 lines
819 B
Go
|
package sections
|
||
|
|
|
||
|
|
import (
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/Silo-Server/silo-server/internal/models"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestLimitUserCollectionSectionItemsKeepsFilteredTotal(t *testing.T) {
|
||
|
|
filtered := []*models.MediaItem{
|
||
|
|
{ContentID: "visible-a"},
|
||
|
|
{ContentID: "visible-b"},
|
||
|
|
{ContentID: "visible-c"},
|
||
|
|
}
|
||
|
|
|
||
|
|
got, total := limitUserCollectionSectionItems(filtered, 2)
|
||
|
|
|
||
|
|
if total != 3 {
|
||
|
|
t.Fatalf("total = %d, want full filtered count", total)
|
||
|
|
}
|
||
|
|
if ids := sectionMediaItemIDs(got); !reflect.DeepEqual(ids, []string{"visible-a", "visible-b"}) {
|
||
|
|
t.Fatalf("limited IDs = %#v, want first visible items", ids)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func sectionMediaItemIDs(items []*models.MediaItem) []string {
|
||
|
|
ids := make([]string, 0, len(items))
|
||
|
|
for _, item := range items {
|
||
|
|
if item != nil {
|
||
|
|
ids = append(ids, item.ContentID)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return ids
|
||
|
|
}
|