mirror of
https://github.com/nagisa77/OpenIsle.git
synced 2026-02-09 16:41:04 +08:00
97 lines
4.7 KiB
Java
97 lines
4.7 KiB
Java
package com.openisle.service;
|
|
|
|
import com.openisle.model.*;
|
|
import com.openisle.repository.*;
|
|
import com.openisle.exception.RateLimitException;
|
|
import org.junit.jupiter.api.Test;
|
|
import org.springframework.scheduling.TaskScheduler;
|
|
import org.springframework.context.ApplicationContext;
|
|
|
|
import static org.junit.jupiter.api.Assertions.*;
|
|
|
|
import java.util.Optional;
|
|
import java.util.List;
|
|
|
|
import static org.mockito.Mockito.*;
|
|
|
|
class PostServiceTest {
|
|
@Test
|
|
void deletePostRemovesReads() {
|
|
PostRepository postRepo = mock(PostRepository.class);
|
|
UserRepository userRepo = mock(UserRepository.class);
|
|
CategoryRepository catRepo = mock(CategoryRepository.class);
|
|
TagRepository tagRepo = mock(TagRepository.class);
|
|
LotteryPostRepository lotteryRepo = mock(LotteryPostRepository.class);
|
|
NotificationService notifService = mock(NotificationService.class);
|
|
SubscriptionService subService = mock(SubscriptionService.class);
|
|
CommentService commentService = mock(CommentService.class);
|
|
CommentRepository commentRepo = mock(CommentRepository.class);
|
|
ReactionRepository reactionRepo = mock(ReactionRepository.class);
|
|
PostSubscriptionRepository subRepo = mock(PostSubscriptionRepository.class);
|
|
NotificationRepository notificationRepo = mock(NotificationRepository.class);
|
|
PostReadService postReadService = mock(PostReadService.class);
|
|
ImageUploader imageUploader = mock(ImageUploader.class);
|
|
TaskScheduler taskScheduler = mock(TaskScheduler.class);
|
|
EmailSender emailSender = mock(EmailSender.class);
|
|
ApplicationContext context = mock(ApplicationContext.class);
|
|
|
|
PostService service = new PostService(postRepo, userRepo, catRepo, tagRepo, lotteryRepo,
|
|
notifService, subService, commentService, commentRepo,
|
|
reactionRepo, subRepo, notificationRepo, postReadService,
|
|
imageUploader, taskScheduler, emailSender, context, PublishMode.DIRECT);
|
|
when(context.getBean(PostService.class)).thenReturn(service);
|
|
|
|
Post post = new Post();
|
|
post.setId(1L);
|
|
User author = new User();
|
|
author.setId(1L);
|
|
author.setRole(Role.USER);
|
|
post.setAuthor(author);
|
|
|
|
when(postRepo.findById(1L)).thenReturn(Optional.of(post));
|
|
when(userRepo.findByUsername("alice")).thenReturn(Optional.of(author));
|
|
when(commentRepo.findByPostAndParentIsNullOrderByCreatedAtAsc(post)).thenReturn(List.of());
|
|
when(reactionRepo.findByPost(post)).thenReturn(List.of());
|
|
when(subRepo.findByPost(post)).thenReturn(List.of());
|
|
when(notificationRepo.findByPost(post)).thenReturn(List.of());
|
|
|
|
service.deletePost(1L, "alice");
|
|
|
|
verify(postReadService).deleteByPost(post);
|
|
verify(postRepo).delete(post);
|
|
}
|
|
|
|
@Test
|
|
void createPostRespectsRateLimit() {
|
|
PostRepository postRepo = mock(PostRepository.class);
|
|
UserRepository userRepo = mock(UserRepository.class);
|
|
CategoryRepository catRepo = mock(CategoryRepository.class);
|
|
TagRepository tagRepo = mock(TagRepository.class);
|
|
LotteryPostRepository lotteryRepo = mock(LotteryPostRepository.class);
|
|
NotificationService notifService = mock(NotificationService.class);
|
|
SubscriptionService subService = mock(SubscriptionService.class);
|
|
CommentService commentService = mock(CommentService.class);
|
|
CommentRepository commentRepo = mock(CommentRepository.class);
|
|
ReactionRepository reactionRepo = mock(ReactionRepository.class);
|
|
PostSubscriptionRepository subRepo = mock(PostSubscriptionRepository.class);
|
|
NotificationRepository notificationRepo = mock(NotificationRepository.class);
|
|
PostReadService postReadService = mock(PostReadService.class);
|
|
ImageUploader imageUploader = mock(ImageUploader.class);
|
|
TaskScheduler taskScheduler = mock(TaskScheduler.class);
|
|
EmailSender emailSender = mock(EmailSender.class);
|
|
ApplicationContext context = mock(ApplicationContext.class);
|
|
|
|
PostService service = new PostService(postRepo, userRepo, catRepo, tagRepo, lotteryRepo,
|
|
notifService, subService, commentService, commentRepo,
|
|
reactionRepo, subRepo, notificationRepo, postReadService,
|
|
imageUploader, taskScheduler, emailSender, context, PublishMode.DIRECT);
|
|
when(context.getBean(PostService.class)).thenReturn(service);
|
|
|
|
when(postRepo.countByAuthorAfter(eq("alice"), any())).thenReturn(1L);
|
|
|
|
assertThrows(RateLimitException.class,
|
|
() -> service.createPost("alice", 1L, "t", "c", List.of(1L),
|
|
null, null, null, null, null, null));
|
|
}
|
|
}
|