Update notification builder

TEAMMATES · Feb 13, 2023 · 897f971 · 897f971
1 parent dd4b597
commit 897f971
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/teammates/storage/sqlentity/Notification.java b/src/main/java/teammates/storage/sqlentity/Notification.java
@@ -219,6 +219,7 @@ public boolean equals(Object other) {
      */
     public static class NotificationBuilder {
 
+        private UUID notificationId;
         private Instant startTime;
         private Instant endTime;
         private NotificationStyle style;
@@ -227,6 +228,11 @@ public static class NotificationBuilder {
         private String message;
         private boolean shown;
 
+        public NotificationBuilder withNotificationId(UUID notificationId) {
+            this.notificationId = notificationId;
+            return this;
+        }
+
         public NotificationBuilder withStartTime(Instant startTime) {
             this.startTime = startTime;
             return this;