From 743c3a9d219be5b1ab505dd595fca6f1a9c0d188 Mon Sep 17 00:00:00 2001
From: ScarletRedMan <terentev.andrey.2002@gmail.com>
Date: Tue, 26 Mar 2024 00:53:16 +0700
Subject: [PATCH] Fixed tests

---
 .../ru/dragonestia/picker/service/NodeServiceTests.java    | 3 +++
 .../ru/dragonestia/picker/service/RoomServiceTests.java    | 7 +++++++
 2 files changed, 10 insertions(+)

diff --git a/server/src/test/java/ru/dragonestia/picker/service/NodeServiceTests.java b/server/src/test/java/ru/dragonestia/picker/service/NodeServiceTests.java
index 1dff9fd..bbe18f8 100644
--- a/server/src/test/java/ru/dragonestia/picker/service/NodeServiceTests.java
+++ b/server/src/test/java/ru/dragonestia/picker/service/NodeServiceTests.java
@@ -4,6 +4,7 @@
 import org.junit.jupiter.api.Test;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.security.test.context.support.WithMockUser;
 import ru.dragonestia.picker.api.exception.NodeAlreadyExistException;
 import ru.dragonestia.picker.api.model.node.PickingMethod;
 import ru.dragonestia.picker.api.repository.type.NodeIdentifier;
@@ -17,6 +18,7 @@ public class NodeServiceTests {
     @Autowired
     private NodeService nodeService;
 
+    @WithMockUser(roles = {"NODE_MANAGEMENT"})
     @Test
     void test_nodeCreateAndRemove() {
         var node = new Node(NodeIdentifier.of("test"), PickingMethod.SEQUENTIAL_FILLING, false);
@@ -30,6 +32,7 @@ void test_nodeCreateAndRemove() {
         Assertions.assertFalse(() -> nodeService.find(node.getIdentifier()).isPresent());
     }
 
+    @WithMockUser(roles = {"NODE_MANAGEMENT"})
     @Test
     void test_allNodes() {
         nodeService.all().forEach(node -> nodeService.remove(node));
diff --git a/server/src/test/java/ru/dragonestia/picker/service/RoomServiceTests.java b/server/src/test/java/ru/dragonestia/picker/service/RoomServiceTests.java
index 0968f54..8d8b7da 100644
--- a/server/src/test/java/ru/dragonestia/picker/service/RoomServiceTests.java
+++ b/server/src/test/java/ru/dragonestia/picker/service/RoomServiceTests.java
@@ -5,6 +5,7 @@
 import org.junit.jupiter.api.Test;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.security.test.context.support.WithMockUser;
 import ru.dragonestia.picker.api.exception.NodeAlreadyExistException;
 import ru.dragonestia.picker.api.exception.NodeNotFoundException;
 import ru.dragonestia.picker.api.exception.NotPersistedNodeException;
@@ -46,6 +47,7 @@ void init() {
         } catch (NodeAlreadyExistException ignore) {}
     }
 
+    @WithMockUser(roles = {"NODE_MANAGEMENT"})
     @Test
     void test_createAndRemove() {
         var room = roomFactory.create(RoomIdentifier.of("test-room"), node, IRoom.UNLIMITED_SLOTS, "", false);
@@ -59,6 +61,7 @@ void test_createAndRemove() {
         Assertions.assertFalse(roomService.find(node, room.getIdentifier()).isPresent());
     }
 
+    @WithMockUser(roles = {"NODE_MANAGEMENT"})
     @Test
     void test_allRooms() {
         var rooms = List.of(
@@ -76,6 +79,7 @@ void test_allRooms() {
         Assertions.assertTrue(rooms.containsAll(list));
     }
 
+    @WithMockUser(roles = {"NODE_MANAGEMENT"})
     @Test
     void test_exceptNotPersistedNode() {
         Assertions.assertThrows(NotPersistedNodeException.class, () -> {
@@ -83,6 +87,7 @@ void test_exceptNotPersistedNode() {
         });
     }
 
+    @WithMockUser(roles = {"NODE_MANAGEMENT"})
     @Test
     void test_pickRoom() {
         var rooms = List.of(
@@ -107,6 +112,7 @@ void test_pickRoom() {
         Assertions.assertEquals("test-room4", roomService.pickAvailable(node, users).roomId());
     }
 
+    @WithMockUser(roles = {"NODE_MANAGEMENT"})
     @Test
     void test_removeNode() {
         nodeService.remove(node);
@@ -114,6 +120,7 @@ void test_removeNode() {
         Assertions.assertThrows(NodeNotFoundException.class, () -> roomService.all(node));
     }
 
+    @WithMockUser(roles = {"NODE_MANAGEMENT"})
     @Test
     void test_nodeDoesNotExists() {
         var node = new Node(NodeIdentifier.of("bruh"), PickingMethod.ROUND_ROBIN, false);