/*
 * Copyright OpenSearch Contributors
 * SPDX-License-Identifier: Apache-2.0
 */

package org.opensearch.indexmanagement.controlcenter.notification.resthandler

import org.junit.Assert
import org.opensearch.client.ResponseException
import org.opensearch.indexmanagement.controlcenter.notification.getResourceURI
import org.opensearch.indexmanagement.controlcenter.notification.nodeIdsInRestIT
import org.opensearch.indexmanagement.controlcenter.notification.randomLRONConfig
import org.opensearch.indexmanagement.controlcenter.notification.randomTaskId
import org.opensearch.indexmanagement.controlcenter.notification.util.getDocID
import org.opensearch.indexmanagement.makeRequest
import org.opensearch.core.rest.RestStatus

@Suppress("UNCHECKED_CAST")
class RestDeleteLRONConfigActionIT : LRONConfigRestTestCase() {
    fun `test delete LRONConfig`() {
        val lronConfig = randomLRONConfig(taskId = randomTaskId(nodeId = nodeIdsInRestIT.random()))
        createLRONConfig(lronConfig)

        val response = client().makeRequest("DELETE", getResourceURI(lronConfig.taskId, lronConfig.actionName))
        assertEquals("delete LRONConfig failed", RestStatus.OK, response.restStatus())
        val responseBody = response.asMap()
        val deletedId = responseBody["_id"] as String
        val deletedResult = responseBody["result"] as String
        Assert.assertEquals("not same doc id", getDocID(lronConfig.taskId, lronConfig.actionName), deletedId)
        Assert.assertEquals("wrong delete result", "deleted", deletedResult)

        try {
            client().makeRequest("GET", getResourceURI(lronConfig.taskId, lronConfig.actionName))
            fail("Expected 404 NOT_FOUND")
        } catch (e: ResponseException) {
            logger.info(e.response.asMap())
            assertEquals("Unexpected status", RestStatus.NOT_FOUND, e.response.restStatus())
        }
    }

    fun `test delete nonexist LRONConfig response`() {
        /* index a random doc to create .opensearch-control-center index */
        createLRONConfig(randomLRONConfig(taskId = randomTaskId(nodeId = nodeIdsInRestIT.random())))
        val lronConfig = randomLRONConfig(taskId = randomTaskId(nodeId = nodeIdsInRestIT.random()))
        val response = client().makeRequest("DELETE", getResourceURI(lronConfig.taskId, lronConfig.actionName))
        assertEquals("delete LRONConfig failed", RestStatus.OK, response.restStatus())
        val responseBody = response.asMap()
        val deletedId = responseBody["_id"] as String
        val deletedResult = responseBody["result"] as String
        Assert.assertEquals("not same doc id", getDocID(lronConfig.taskId, lronConfig.actionName), deletedId)
        Assert.assertEquals("wrong delete result", "not_found", deletedResult)
    }
}