mirror of
https://github.com/owncast/owncast.git
synced 2024-10-10 19:16:02 +00:00
114 lines
2.6 KiB
Go
114 lines
2.6 KiB
Go
package federationrepository
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/owncast/owncast/models"
|
|
"github.com/owncast/owncast/storage/data"
|
|
"github.com/owncast/owncast/utils"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
ds, err := data.NewStore(":memory:")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
federationRepository := New(ds)
|
|
|
|
number := 100
|
|
for i := 0; i < number; i++ {
|
|
u := createFakeFollower()
|
|
federationRepository.createFollow(u.ActorIRI, u.Inbox, "https://fake.fediverse.server/some/request", u.Name, u.Username, u.Image, nil, true)
|
|
followers = append(followers, u)
|
|
}
|
|
|
|
m.Run()
|
|
}
|
|
|
|
var followers = []models.Follower{}
|
|
|
|
func TestQueryFollowers(t *testing.T) {
|
|
federationRepository := Get()
|
|
|
|
f, total, err := federationRepository.GetFederationFollowers(10, 0)
|
|
if err != nil {
|
|
t.Errorf("Error querying followers: %s", err)
|
|
}
|
|
|
|
if len(f) != 10 {
|
|
t.Errorf("Expected 10 followers, got %d", len(f))
|
|
}
|
|
|
|
if total != 100 {
|
|
t.Errorf("Expected 100 followers, got %d", total)
|
|
}
|
|
}
|
|
|
|
func TestQueryFollowersWithOffset(t *testing.T) {
|
|
federationRepository := Get()
|
|
|
|
f, total, err := federationRepository.GetFederationFollowers(10, 10)
|
|
if err != nil {
|
|
t.Errorf("Error querying followers: %s", err)
|
|
}
|
|
|
|
if len(f) != 10 {
|
|
t.Errorf("Expected 10 followers, got %d", len(f))
|
|
}
|
|
|
|
if total != 100 {
|
|
t.Errorf("Expected 100 followers, got %d", total)
|
|
}
|
|
}
|
|
|
|
func TestQueryFollowersWithOffsetAndLimit(t *testing.T) {
|
|
federationRepository := Get()
|
|
|
|
f, total, err := federationRepository.GetFederationFollowers(10, 90)
|
|
if err != nil {
|
|
t.Errorf("Error querying followers: %s", err)
|
|
}
|
|
|
|
if len(f) != 10 {
|
|
t.Errorf("Expected 10 followers, got %d", len(f))
|
|
}
|
|
|
|
if total != 100 {
|
|
t.Errorf("Expected 100 followers, got %d", total)
|
|
}
|
|
}
|
|
|
|
func TestQueryFollowersWithPagination(t *testing.T) {
|
|
federationRepository := Get()
|
|
|
|
f, _, err := federationRepository.GetFederationFollowers(15, 10)
|
|
if err != nil {
|
|
t.Errorf("Error querying followers: %s", err)
|
|
}
|
|
|
|
comparisonFollowers := followers[10:25]
|
|
if len(f) != len(comparisonFollowers) {
|
|
t.Errorf("Expected %d followers, got %d", len(comparisonFollowers), len(f))
|
|
}
|
|
|
|
for i, follower := range f {
|
|
if follower.ActorIRI != comparisonFollowers[i].ActorIRI {
|
|
t.Errorf("Expected %s, got %s", comparisonFollowers[i].ActorIRI, follower.ActorIRI)
|
|
}
|
|
}
|
|
}
|
|
|
|
func createFakeFollower() models.Follower {
|
|
user, _ := utils.GenerateRandomString(10)
|
|
|
|
return models.Follower{
|
|
ActorIRI: "https://freedom.eagle/user/" + user,
|
|
Inbox: "https://fake.fediverse.server/user/" + user + "/inbox",
|
|
Image: "https://fake.fediverse.server/user/" + user + "/avatar.png",
|
|
Name: user,
|
|
Username: user,
|
|
Timestamp: utils.NullTime{},
|
|
}
|
|
}
|