package esmedia

import (
	"testing"

	"91porn-server/models/v/mediamod"

	"go.mongodb.org/mongo-driver/bson/primitive"
)

func TestBuildSourceIncludesCountCollect(t *testing.T) {
	id := primitive.NewObjectID()
	source := buildSource([]*mediamod.Media{{
		ID:           id,
		CountCollect: 12,
	}})

	media, ok := source[id.Hex()].(mediamod.ESMedia)
	if !ok {
		t.Fatalf("source type = %T, want mediamod.ESMedia", source[id.Hex()])
	}
	if media.CountCollect != 12 {
		t.Fatalf("countCollect = %d, want 12", media.CountCollect)
	}
}