internal/app/welcomemessages/service_test.go

Rama main · réplica de lectura del repositorio original.

package welcomemessages

import (
	"context"
	"errors"
	"testing"
	"time"

	"telesrv/internal/domain"
)

type welcomeStoreSpy struct {
	creates int
}

func (s *welcomeStoreSpy) CreateWelcomeMessage(_ context.Context, req domain.CreateWelcomeMessageRequest) (domain.WelcomeMessage, bool, error) {
	s.creates++
	return domain.WelcomeMessage{
		ID: 1, Peer: req.Peer, CreatorUserID: req.CreatorUserID, Date: req.Date,
		RandomID: req.RandomID, Content: req.Content, CreateFingerprint: req.CreateFingerprint, Version: 1,
	}, true, nil
}
func (*welcomeStoreSpy) EditWelcomeMessage(context.Context, domain.EditWelcomeMessageRequest) (domain.WelcomeMessage, error) {
	return domain.WelcomeMessage{}, nil
}
func (*welcomeStoreSpy) ListWelcomeMessages(context.Context, domain.Peer, int64) (domain.WelcomeMessageList, error) {
	return domain.WelcomeMessageList{Hash: 1}, nil
}
func (*welcomeStoreSpy) DeleteWelcomeMessage(context.Context, domain.Peer, int) (bool, error) {
	return true, nil
}
func (*welcomeStoreSpy) DeleteAllWelcomeMessages(context.Context, domain.Peer) (bool, error) {
	return true, nil
}
func (*welcomeStoreSpy) HasWelcomeMessages(context.Context, domain.Peer) (bool, error) {
	return true, nil
}

type welcomeChannelAccess struct {
	view domain.ChannelView
	err  error
}

func (a *welcomeChannelAccess) ResolveChannel(context.Context, int64, int64) (domain.ChannelView, error) {
	return a.view, a.err
}

func TestServiceRechecksManageWelcomeMessagesPermission(t *testing.T) {
	peer := domain.Peer{Type: domain.PeerTypeChannel, ID: 77}
	store := &welcomeStoreSpy{}
	channels := &welcomeChannelAccess{view: domain.ChannelView{
		Channel: domain.Channel{ID: peer.ID, Megagroup: true},
		Self: domain.ChannelMember{
			ChannelID: peer.ID, UserID: 9, Role: domain.ChannelRoleAdmin,
			Status:      domain.ChannelMemberActive,
			AdminRights: domain.ChannelAdminRights{ManageWelcomeMessages: true},
		},
	}}
	service := NewService(store, channels, WithClock(func() time.Time { return time.Unix(1700000000, 0) }))
	message, created, err := service.Create(context.Background(), 9, peer, 1001, domain.WelcomeMessageContent{Message: "hello"})
	if err != nil || !created || message.Date != 1700000000 || store.creates != 1 {
		t.Fatalf("authorized create = %+v created=%v calls=%d err=%v", message, created, store.creates, err)
	}

	channels.view.Self.Status = domain.ChannelMemberLeft
	if _, _, err := service.Create(context.Background(), 9, peer, 1002, domain.WelcomeMessageContent{Message: "blocked"}); !errors.Is(err, domain.ErrWelcomeMessageForbidden) || store.creates != 1 {
		t.Fatalf("inactive admin create err=%v calls=%d", err, store.creates)
	}

	channels.view.Self = domain.ChannelMember{UserID: 9, Role: domain.ChannelRoleCreator, Status: domain.ChannelMemberActive}
	if _, err := service.List(context.Background(), 9, peer, 0); err != nil {
		t.Fatalf("creator list: %v", err)
	}
	channels.view.Channel.Monoforum = true
	if _, err := service.List(context.Background(), 9, peer, 0); !errors.Is(err, domain.ErrWelcomeMessagePeerInvalid) {
		t.Fatalf("monoforum list err=%v", err)
	}
}

func TestServiceRejectsOrdinaryMemberAndAllowsJoinedBroadcastAdmin(t *testing.T) {
	peer := domain.Peer{Type: domain.PeerTypeChannel, ID: 88}
	store := &welcomeStoreSpy{}
	channels := &welcomeChannelAccess{view: domain.ChannelView{
		Channel: domain.Channel{ID: peer.ID, Broadcast: true},
		Self:    domain.ChannelMember{UserID: 10, Role: domain.ChannelRoleMember, Status: domain.ChannelMemberActive},
	}}
	service := NewService(store, channels)
	if _, err := service.DeleteAll(context.Background(), 10, peer); !errors.Is(err, domain.ErrWelcomeMessageForbidden) {
		t.Fatalf("ordinary member delete-all err=%v", err)
	}
	channels.view.Self.Role = domain.ChannelRoleAdmin
	channels.view.Self.AdminRights.ManageWelcomeMessages = true
	if ok, err := service.DeleteAll(context.Background(), 10, peer); err != nil || !ok {
		t.Fatalf("joined broadcast admin delete-all=%v,%v", ok, err)
	}
}