SYNCT-71: Added test for new logic in SyncAuditService

openmrs · Dec 15, 2017 · 8b809a5 · 8b809a5
1 parent 41ceb48
commit 8b809a5
Showing 1 changed file with 58 additions and 0 deletions.
diff --git a/api/src/test/java/org/openmrs/module/sync2/api/impl/SyncAuditServiceImplTest.java b/api/src/test/java/org/openmrs/module/sync2/api/impl/SyncAuditServiceImplTest.java
@@ -138,6 +138,64 @@ public void saveSuccessfulAudit_shouldNotSave() {
 
         Assert.assertNull(fetched);
     }
+
+    @Test
+    public void saveAuditMessage_shouldSaveSuccessfulAudit() {
+        AuditMessage auditMessage = new AuditMessage();
+        auditMessage.setSuccess(true);
+
+        when(dao.saveItem(any())).thenReturn(auditMessage);
+        when(configurationService.getSyncConfiguration().getGeneral().isPersistSuccessAudit()).thenReturn(true);
+        when(configurationService.getSyncConfiguration().getGeneral().isPersistFailureAudit()).thenReturn(false);
+
+        AuditMessage fetched = auditService.saveAuditMessage(auditMessage);
+
+        Assert.assertEquals(auditMessage, fetched);
+    }
+
+    @Test
+    public void saveAuditMessage_shouldNotSaveSuccessfulAudit() {
+        AuditMessage auditMessage = new AuditMessage();
+        auditMessage.setSuccess(true);
+
+
+        when(dao.saveItem(any())).thenReturn(auditMessage);
+        when(configurationService.getSyncConfiguration().getGeneral().isPersistSuccessAudit()).thenReturn(false);
+        when(configurationService.getSyncConfiguration().getGeneral().isPersistFailureAudit()).thenReturn(false);
+
+        AuditMessage fetched = auditService.saveAuditMessage(auditMessage);
+
+        Assert.assertNull(fetched);
+    }
+
+    @Test
+    public void saveAuditMessage_shouldSaveFailureAudit() {
+        AuditMessage auditMessage = new AuditMessage();
+        auditMessage.setSuccess(false);
+
+        when(dao.saveItem(any())).thenReturn(auditMessage);
+        when(configurationService.getSyncConfiguration().getGeneral().isPersistSuccessAudit()).thenReturn(false);
+        when(configurationService.getSyncConfiguration().getGeneral().isPersistFailureAudit()).thenReturn(true);
+
+        AuditMessage fetched = auditService.saveAuditMessage(auditMessage);
+
+        Assert.assertEquals(auditMessage, fetched);
+    }
+
+    @Test
+    public void saveAuditMessage_shouldNotSaveFailureAudit() {
+        AuditMessage auditMessage = new AuditMessage();
+        auditMessage.setSuccess(false);
+
+
+        when(dao.saveItem(any())).thenReturn(auditMessage);
+        when(configurationService.getSyncConfiguration().getGeneral().isPersistSuccessAudit()).thenReturn(false);
+        when(configurationService.getSyncConfiguration().getGeneral().isPersistFailureAudit()).thenReturn(false);
+
+        AuditMessage fetched = auditService.saveAuditMessage(auditMessage);
+
+        Assert.assertNull(fetched);
+    }
 
     private AuditMessage prepareAuditMessage(Boolean success) throws ParseException {
         AuditMessage newMessage = new AuditMessage();