@@ -0,0 +1,25 @@
|
||||
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)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user