From a1baf2881b6bf44086552484ab329bdf49c94f0e Mon Sep 17 00:00:00 2001
From: SmithaManjunath <smanjunath@microsoft.com>
Date: Fri, 1 Apr 2022 11:48:23 +0530
Subject: [PATCH] increase code coverage of notification-core to 80%

---
 .../osdu/notification/di/CredentialHeadersProviderTest.java     | 2 +-
 .../osdu/notification/utils/NotificationFilterTest.java         | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/notification-core/src/test/java/org/opengroup/osdu/notification/di/CredentialHeadersProviderTest.java b/notification-core/src/test/java/org/opengroup/osdu/notification/di/CredentialHeadersProviderTest.java
index e7788440a..2d604ac94 100644
--- a/notification-core/src/test/java/org/opengroup/osdu/notification/di/CredentialHeadersProviderTest.java
+++ b/notification-core/src/test/java/org/opengroup/osdu/notification/di/CredentialHeadersProviderTest.java
@@ -54,7 +54,7 @@ public class CredentialHeadersProviderTest {
         when(serviceAccountJwtClient.getIdToken("opendes")).thenReturn("sampleAuthToken");
 
         DpsHeaders dpsHeaders = headersProvider.getObject();
-        assertNotNull(headersProvider.getObject());
+        assertNotNull(dpsHeaders);
         assertEquals(dpsHeaders.getHeaders().get("content-type"), "application/json");
         assertEquals(dpsHeaders.getHeaders().get("authorization"), "sampleAuthToken");
         assertEquals(dpsHeaders.getHeaders().get("data-partition-id"), "opendes");
diff --git a/notification-core/src/test/java/org/opengroup/osdu/notification/utils/NotificationFilterTest.java b/notification-core/src/test/java/org/opengroup/osdu/notification/utils/NotificationFilterTest.java
index 46ce614d8..ba2c6125b 100644
--- a/notification-core/src/test/java/org/opengroup/osdu/notification/utils/NotificationFilterTest.java
+++ b/notification-core/src/test/java/org/opengroup/osdu/notification/utils/NotificationFilterTest.java
@@ -18,6 +18,7 @@ import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
 
+import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotNull;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.Mockito.doNothing;
@@ -52,6 +53,7 @@ public class NotificationFilterTest {
 
         notificationFilter.doFilter(servletRequest, servletResponse, filterChain);
         assertNotNull(dpsHeaders.getHeaders().get("correlation-id"));
+        assertEquals(dpsHeaders.getHeaders().get("content-type"),"application/json");
 
     }
 }
\ No newline at end of file
-- 
GitLab