test a few more methods IQSS#7492

ErykKul · May 16, 2022 · 6427be1 · 6427be1
1 parent 51521ca
commit 6427be1
Showing 1 changed file with 41 additions and 0 deletions.
diff --git a/src/test/java/edu/harvard/iq/dataverse/authorization/users/AuthenticatedUserTest.java b/src/test/java/edu/harvard/iq/dataverse/authorization/users/AuthenticatedUserTest.java
@@ -5,10 +5,13 @@
  */
 package edu.harvard.iq.dataverse.authorization.users;
 
+import edu.harvard.iq.dataverse.UserNotification;
 import edu.harvard.iq.dataverse.UserNotification.Type;
+import edu.harvard.iq.dataverse.UserNotificationServiceBean;
 import edu.harvard.iq.dataverse.authorization.AuthenticatedUserDisplayInfo;
 import edu.harvard.iq.dataverse.authorization.AuthenticatedUserLookup;
 import edu.harvard.iq.dataverse.mocks.MocksFactory;
+import edu.harvard.iq.dataverse.settings.SettingsServiceBean;
 import java.sql.Timestamp;
 import java.util.Date;
 import java.util.EnumSet;
@@ -21,14 +24,24 @@
 
 import javax.json.JsonObject;
 import javax.json.JsonString;
+import org.junit.runner.RunWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.junit.MockitoJUnitRunner;
 
 /**
  * Tested class: AuthenticatedUser.java
  *
  * @author bsilverstein
  */
+@RunWith(MockitoJUnitRunner.class)
 public class AuthenticatedUserTest {
 
+    @Mock
+    private SettingsServiceBean settingsServiceBean;
+    @InjectMocks
+    private UserNotificationServiceBean userNotificationService;
+
     public AuthenticatedUserTest() {
     }
 
@@ -393,6 +406,34 @@ public void testTypeTokenizer() {
         assertTrue("typeSet contains REVOKEROLE", typeSet.contains(Type.REVOKEROLE));
     }
 
+    @Test
+    public void testIsEmailMuted() {
+        testUser.setMutedEmails(mutedTypes);
+        UserNotification userNotification = new UserNotification();
+        userNotification.setUser(testUser);
+        userNotification.setSendDate(null);
+        userNotification.setObjectId(null);
+        userNotification.setRequestor(null);
+        userNotification.setType(Type.ASSIGNROLE); // muted
+        assertTrue(userNotificationService.isEmailMuted(userNotification));
+        userNotification.setType(Type.APIGENERATED); // not muted
+        assertFalse(userNotificationService.isEmailMuted(userNotification));
+    }
+
+    @Test
+    public void isNotificationMuted() {
+        testUser.setMutedNotifications(mutedTypes);
+        UserNotification userNotification = new UserNotification();
+        userNotification.setUser(testUser);
+        userNotification.setSendDate(null);
+        userNotification.setObjectId(null);
+        userNotification.setRequestor(null);
+        userNotification.setType(Type.ASSIGNROLE); // muted
+        assertTrue(userNotificationService.isNotificationMuted(userNotification));
+        userNotification.setType(Type.APIGENERATED); // not muted
+        assertFalse(userNotificationService.isNotificationMuted(userNotification));
+    }
+
     /**
      * All commented tests below have only been generated / are not complete for
      * AuthenticatedUser.java The tests above should all run fine, due to time