eatery-service/src/test/java/com/drinkool/services/ShiftServiceTest.java

@@ -102,7 +102,9 @@ public class ShiftServiceTest {
 
 		given()
 			.contentType("application/json")
-			.header(InternalValue.userId, staffId.toString()) // Mock user ID
+			.header(InternalValue.role, Role.Staff)
+			.header(InternalValue.userId, staffId.toString()) 
+			.header(InternalValue.managerId, ownerId.toString())
 			.body("{ \"query\": \"" + mutation + "\" }")
 			.when()
 			.post("/graphql")