#87 fix: tokenGenerator 반영

TeamSynergyy · Jun 10, 2024 · 9b3be20 · 9b3be20
1 parent 6db3df7
commit 9b3be20
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/src/test/java/com/seoultech/synergybe/domain/post/PostGeneratorTest.java b/src/test/java/com/seoultech/synergybe/domain/post/PostGeneratorTest.java
@@ -3,6 +3,7 @@
 import com.seoultech.synergybe.domain.common.CustomPasswordEncoder;
 import com.seoultech.synergybe.domain.common.generator.IdGenerator;
 import com.seoultech.synergybe.domain.common.generator.IdPrefix;
+import com.seoultech.synergybe.domain.common.generator.TokenGenerator;
 import com.seoultech.synergybe.domain.post.implement.PostManager;
 import com.seoultech.synergybe.domain.user.User;
 import com.seoultech.synergybe.domain.user.repository.UserRepository;
@@ -25,6 +26,9 @@ public class PostGeneratorTest {
     @Autowired
     IdGenerator idGenerator;
 
+    @Autowired
+    TokenGenerator tokenGenerator;
+
     @Autowired
     PostManager postManager;
 
@@ -34,8 +38,11 @@ public class PostGeneratorTest {
 
     @Test
     void generatePost() {
+        Long userId = idGenerator.generateId();
+        String userToken = tokenGenerator.generateToken(IdPrefix.USER);
+
         User user = User.builder()
-                .id("user-id")
+                .id(userId)
                 .password("3e4r5t6y6y7u")
                 .passwordEncoder(passwordEncoder)
                 .email("email@email.com")
@@ -49,10 +56,13 @@ void generatePost() {
         List<Post> postList = new ArrayList<>();
 
 
-        for (int i = 0; i < 10000; i++) {
-            String postId = idGenerator.generateId(IdPrefix.POST);
+        for (int i = 0; i < 10; i++) {
+            Long postId = idGenerator.generateId();
+            String postToken = tokenGenerator.generateToken(IdPrefix.POST);
+
             Post post = Post.builder()
                     .id(postId)
+                    .postToken(postToken)
                     .user(user)
                     .title("title")
                     .content("content")