mautrix-telegram/pkg/gotd/telegram/peers/manager_test.go

package peers

import (
	"testing"
	"time"

	"go.uber.org/zap/zaptest"

	"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
	"go.mau.fi/mautrix-telegram/pkg/gotd/tgerr"
	"go.mau.fi/mautrix-telegram/pkg/gotd/tgmock"
)

func testManager(t *testing.T) (*tgmock.Mock, *Manager) {
	mock := tgmock.New(t)
	return mock, Options{
		Logger: zaptest.NewLogger(t),
		Cache:  &InmemoryCache{},
	}.Build(tg.NewClient(mock))
}

func getTestSelf() *tg.User {
	u := &tg.User{
		Self:       true,
		Bot:        true,
		ID:         10,
		AccessHash: 10,
		FirstName:  "Lana",
		LastName:   "Rhoades",
		Username:   "thebot",
	}
	u.SetFlags()
	return u
}

func getTestUser() *tg.User {
	u := &tg.User{
		Self:       false,
		Bot:        false,
		ID:         11,
		AccessHash: 10,
		FirstName:  "Julia",
		LastName:   "Ann",
		Username:   "aboba",
	}
	u.SetFlags()
	return u
}

func getTestUserFull() tg.UserFull {
	u := tg.UserFull{
		PhoneCallsAvailable: true,
		ID:                  11,
		About:               "hot mommy",
	}
	u.SetFlags()
	return u
}

func getTestChat() *tg.Chat {
	u := &tg.Chat{
		Noforwards:        true,
		ID:                10,
		Title:             "I hate mondays",
		ParticipantsCount: 1,
		Date:              int(time.Now().Unix()),
		Version:           1,
		Photo:             &tg.ChatPhotoEmpty{},
	}
	u.SetFlags()
	return u
}

func getTestChatFull() *tg.ChatFull {
	u := &tg.ChatFull{
		CanSetUsername: false,
		HasScheduled:   true,
		ID:             10,
		About:          "garfield blog",
		Participants: &tg.ChatParticipants{
			ChatID: 10,
			Participants: []tg.ChatParticipantClass{
				&tg.ChatParticipant{
					UserID:    10,
					InviterID: 10,
					Date:      10,
				},
			},
			Version: 1,
		},
	}
	u.SetFlags()
	return u
}

func getTestChannel() *tg.Channel {
	u := &tg.Channel{
		Broadcast:           true,
		Noforwards:          true,
		ID:                  11,
		AccessHash:          0,
		Title:               "I hate mondays",
		Username:            "",
		Photo:               &tg.ChatPhotoEmpty{},
		Date:                int(time.Now().Unix()),
		RestrictionReason:   nil,
		AdminRights:         tg.ChatAdminRights{},
		BannedRights:        tg.ChatBannedRights{},
		DefaultBannedRights: tg.ChatBannedRights{},
		ParticipantsCount:   1,
	}
	u.SetFlags()
	return u
}

func getTestChannelFull() *tg.ChannelFull {
	u := &tg.ChannelFull{
		HasScheduled:      true,
		ID:                11,
		About:             "garfield blog",
		ParticipantsCount: 1,
		ChatPhoto:         &tg.PhotoEmpty{},
	}
	u.SetFlags()
	return u
}

func getTestError() *tgerr.Error {
	return &tgerr.Error{
		Code:    1337,
		Message: "TEST_ERROR",
		Type:    "TEST_ERROR",
	}
}