diff --git a/notifications/core/src/main/kotlin/org/opensearch/notifications/core/client/DestinationHttpClient.kt b/notifications/core/src/main/kotlin/org/opensearch/notifications/core/client/DestinationHttpClient.kt index 594bfa9f..5764f852 100644 --- a/notifications/core/src/main/kotlin/org/opensearch/notifications/core/client/DestinationHttpClient.kt +++ b/notifications/core/src/main/kotlin/org/opensearch/notifications/core/client/DestinationHttpClient.kt @@ -141,7 +141,9 @@ class DestinationHttpClient { @Throws(IOException::class) fun getResponseString(response: CloseableHttpResponse): String { val entity: HttpEntity = response.entity ?: return "{}" - return EntityUtils.toString(entity) + val responseString = EntityUtils.toString(entity) + // DeliveryStatus need statusText must not be empty, convert empty response to {} + return if (responseString.isNullOrEmpty()) "{}" else responseString } @Throws(IOException::class) diff --git a/notifications/core/src/test/kotlin/org/opensearch/notifications/core/destinations/ChimeDestinationTests.kt b/notifications/core/src/test/kotlin/org/opensearch/notifications/core/destinations/ChimeDestinationTests.kt index 4944b823..46bf3a23 100644 --- a/notifications/core/src/test/kotlin/org/opensearch/notifications/core/destinations/ChimeDestinationTests.kt +++ b/notifications/core/src/test/kotlin/org/opensearch/notifications/core/destinations/ChimeDestinationTests.kt @@ -91,7 +91,7 @@ internal class ChimeDestinationTests { @Test fun `test chime message empty entity response`() { val mockHttpClient: CloseableHttpClient = EasyMock.createMock(CloseableHttpClient::class.java) - val expectedWebhookResponse = DestinationMessageResponse(RestStatus.OK.status, "") + val expectedWebhookResponse = DestinationMessageResponse(RestStatus.OK.status, "{}") val httpResponse: CloseableHttpResponse = EasyMock.createMock(CloseableHttpResponse::class.java) EasyMock.expect(mockHttpClient.execute(EasyMock.anyObject(HttpPost::class.java))).andReturn(httpResponse) diff --git a/notifications/core/src/test/kotlin/org/opensearch/notifications/core/destinations/CustomWebhookDestinationTests.kt b/notifications/core/src/test/kotlin/org/opensearch/notifications/core/destinations/CustomWebhookDestinationTests.kt index 0325b0a8..a8f3fa68 100644 --- a/notifications/core/src/test/kotlin/org/opensearch/notifications/core/destinations/CustomWebhookDestinationTests.kt +++ b/notifications/core/src/test/kotlin/org/opensearch/notifications/core/destinations/CustomWebhookDestinationTests.kt @@ -6,6 +6,7 @@ package org.opensearch.notifications.core.destinations import io.mockk.every +import io.mockk.mockk import io.mockk.mockkStatic import org.apache.http.client.methods.CloseableHttpResponse import org.apache.http.client.methods.HttpPatch @@ -108,7 +109,7 @@ internal class CustomWebhookDestinationTests { @MethodSource("methodToHttpRequestType") fun `test custom webhook message empty entity response`(method: String, expectedHttpClass: Class) { val mockHttpClient: CloseableHttpClient = EasyMock.createMock(CloseableHttpClient::class.java) - val expectedWebhookResponse = DestinationMessageResponse(RestStatus.OK.status, "") + val expectedWebhookResponse = DestinationMessageResponse(RestStatus.OK.status, "{}") val httpResponse: CloseableHttpResponse = EasyMock.createMock(CloseableHttpResponse::class.java) EasyMock.expect(mockHttpClient.execute(EasyMock.anyObject(HttpPost::class.java))).andReturn(httpResponse) @@ -230,4 +231,17 @@ internal class CustomWebhookDestinationTests { val actualRequestBody = httpClient.buildRequestBody(destination, message) assertEquals(messageText, actualRequestBody) } + + @Test + fun `test get response string`() { + val httpClient = DestinationHttpClient() + val response = mockk() + every { response.entity } returns null + var responseString = httpClient.getResponseString(response) + assertEquals(responseString, "{}") + + every { response.entity } returns StringEntity("") + responseString = httpClient.getResponseString(response) + assertEquals(responseString, "{}") + } } diff --git a/notifications/core/src/test/kotlin/org/opensearch/notifications/core/destinations/SlackDestinationTests.kt b/notifications/core/src/test/kotlin/org/opensearch/notifications/core/destinations/SlackDestinationTests.kt index 65345014..bea8a0f5 100644 --- a/notifications/core/src/test/kotlin/org/opensearch/notifications/core/destinations/SlackDestinationTests.kt +++ b/notifications/core/src/test/kotlin/org/opensearch/notifications/core/destinations/SlackDestinationTests.kt @@ -94,7 +94,7 @@ internal class SlackDestinationTests { @Test fun `test Slack message empty entity response`() { val mockHttpClient: CloseableHttpClient = EasyMock.createMock(CloseableHttpClient::class.java) - val expectedWebhookResponse = DestinationMessageResponse(RestStatus.OK.status, "") + val expectedWebhookResponse = DestinationMessageResponse(RestStatus.OK.status, "{}") val httpResponse: CloseableHttpResponse = EasyMock.createMock(CloseableHttpResponse::class.java) EasyMock.expect(mockHttpClient.execute(EasyMock.anyObject(HttpPost::class.java))).andReturn(httpResponse) diff --git a/notifications/notifications/build.gradle b/notifications/notifications/build.gradle index 8d851294..1c403b49 100644 --- a/notifications/notifications/build.gradle +++ b/notifications/notifications/build.gradle @@ -219,6 +219,12 @@ integTest { excludeTestsMatching "org.opensearch.integtest.bwc.*IT" } } + + if (usingRemoteCluster) { + filter { + excludeTestsMatching "org.opensearch.integtest.send.SendTestMessageWithMockServerIT" + } + } } project.getTasks().getByName("bundlePlugin").dependsOn(findProject(":${rootProject.name}-core").tasks.getByPath(":${rootProject.name}-core:build")) diff --git a/notifications/notifications/src/test/kotlin/org/opensearch/integtest/send/SendTestMessageWithMockServerIT.kt b/notifications/notifications/src/test/kotlin/org/opensearch/integtest/send/SendTestMessageWithMockServerIT.kt new file mode 100644 index 00000000..4cbb03bd --- /dev/null +++ b/notifications/notifications/src/test/kotlin/org/opensearch/integtest/send/SendTestMessageWithMockServerIT.kt @@ -0,0 +1,81 @@ +/* + * Copyright OpenSearch Contributors + * SPDX-License-Identifier: Apache-2.0 + */ + +package org.opensearch.integtest.send + +import com.google.gson.JsonArray +import com.google.gson.JsonObject +import com.sun.net.httpserver.HttpServer +import org.junit.AfterClass +import org.junit.Assert +import org.junit.BeforeClass +import org.opensearch.integtest.PluginRestTestCase +import org.opensearch.notifications.NotificationPlugin.Companion.PLUGIN_BASE_URI +import org.opensearch.rest.RestRequest +import org.opensearch.rest.RestStatus +import java.net.InetAddress +import java.net.InetSocketAddress + +internal class SendTestMessageWithMockServerIT : PluginRestTestCase() { + + fun `test webhook return with empty entity`() { + val url = "http://${server.address.hostString}:${server.address.port}/webhook" + logger.info("webhook url = {}", url) + // Create webhook notification config + val createRequestJsonString = """ + { + "config":{ + "name":"this is a sample config name", + "description":"this is a sample config description", + "config_type":"webhook", + "is_enabled":true, + "webhook":{ + "url":"$url", + "header_params": { + "Content-type": "text/plain" + } + } + } + } + """.trimIndent() + val configId = createConfigWithRequestJsonString(createRequestJsonString) + Assert.assertNotNull(configId) + Thread.sleep(1000) + + // send test message + val sendResponse = executeRequest( + RestRequest.Method.POST.name, "$PLUGIN_BASE_URI/feature/test/$configId", "", RestStatus.OK.status + ) + + logger.info(sendResponse) + + // verify failure response is with message + val deliveryStatus = (sendResponse.get("status_list") as JsonArray).get(0).asJsonObject + .get("delivery_status") as JsonObject + Assert.assertEquals(deliveryStatus.get("status_code").asString, "200") + Assert.assertEquals(deliveryStatus.get("status_text").asString, "{}") + } + + companion object { + private lateinit var server: HttpServer + + @JvmStatic + @BeforeClass + fun setupWebhook() { + server = HttpServer.create(InetSocketAddress(InetAddress.getLoopbackAddress(), 0), 0) + + server.createContext("/webhook") { + it.sendResponseHeaders(200, -1) + } + server.start() + } + + @JvmStatic + @AfterClass + fun stopMockServer() { + server.stop(1) + } + } +}