From 2f6f22941d2063aae299a8f04991a37c0ae5a040 Mon Sep 17 00:00:00 2001
From: TakahashiNg <83152264+TakahashiNguyen@users.noreply.github.com>
Date: Tue, 12 May 2026 01:41:09 +0000
Subject: [PATCH] chore: update

---
 .../src/test/java/com/drinkool/services/ShiftServiceTest.java | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/eatery-service/src/test/java/com/drinkool/services/ShiftServiceTest.java b/eatery-service/src/test/java/com/drinkool/services/ShiftServiceTest.java
index 854b33c..88f4033 100644
--- a/eatery-service/src/test/java/com/drinkool/services/ShiftServiceTest.java
+++ b/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")