26 lines
532 B
Go
26 lines
532 B
Go
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)
|
|
}
|
|
}
|