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

package org.opensearch.commons.notifications.model

import com.fasterxml.jackson.core.JsonParseException
import org.junit.jupiter.api.Assertions
import org.junit.jupiter.api.Assertions.assertThrows
import org.junit.jupiter.api.Assertions.fail
import org.junit.jupiter.api.Test
import org.opensearch.commons.utils.createObjectFromJsonString
import org.opensearch.commons.utils.getJsonString
import org.opensearch.commons.utils.recreateObject

internal class SnsTests {

    @Test
    fun `SNS should throw exception if empty topic`() {
        assertThrows(IllegalArgumentException::class.java) {
            Sns("", null)
        }
        val jsonString = "{\"topic_arn\":\"\"}"
        assertThrows(IllegalArgumentException::class.java) {
            createObjectFromJsonString(jsonString) { Sns.parse(it) }
        }
    }

    @Test
    fun `SNS should throw exception if invalid topic ARN`() {
        assertThrows(IllegalArgumentException::class.java) {
            Sns("arn:aws:es:us-east-1:012345678989:test", null)
        }
        val jsonString = "{\"topic_arn\":\"arn:aws:es:us-east-1:012345678989:test\"}"
        assertThrows(IllegalArgumentException::class.java) {
            createObjectFromJsonString(jsonString) { Sns.parse(it) }
        }
    }

    @Test
    fun `SNS should throw exception if invalid role ARN`() {
        assertThrows(IllegalArgumentException::class.java) {
            Sns("arn:aws:sns:us-east-1:012345678912:topic-test", "arn:aws:iam:us-east-1:0123456789:role-test")
        }
        val jsonString =
            "{\"topic_arn\":\"arn:aws:sns:us-east-1:012345678912:topic-test\",\"role_arn\":\"arn:aws:iam:us-east-1:0123456789:role-test\"}"
        assertThrows(IllegalArgumentException::class.java) {
            createObjectFromJsonString(jsonString) { Sns.parse(it) }
        }
    }

    @Test
    fun `test SNS correctly validates SNS FIFO topic ARN`() {
        try {
            Sns(
                "arn:aws:sns:ap-southeast-2:333654771707:sns-fifo-alerting.fifo",
                "arn:aws:iam::012345678912:role/iam-test"
            )
        } catch (e: Exception) {
            fail("Expected fifo sns topic ARN to be validated successfully", e)
        }
    }

    @Test
    fun `SNS serialize and deserialize transport object should be equal`() {
        val sampleSns = Sns("arn:aws:sns:us-east-1:012345678912:topic-test", "arn:aws:iam::012345678912:role/iam-test")
        val recreatedObject = recreateObject(sampleSns) { Sns(it) }
        Assertions.assertEquals(sampleSns, recreatedObject)
    }

    @Test
    fun `SNS serialize and deserialize using json object should be equal`() {
        val sampleSns = Sns("arn:aws:sns:us-east-1:012345678912:topic-test", "arn:aws:iam::012345678912:role/iam-test")
        val jsonString = getJsonString(sampleSns)
        val recreatedObject = createObjectFromJsonString(jsonString) { Sns.parse(it) }
        Assertions.assertEquals(sampleSns, recreatedObject)
    }

    @Test
    fun `SNS should deserialize json object using parser`() {
        val sampleSns = Sns("arn:aws:sns:us-east-1:012345678912:topic-test", "arn:aws:iam::012345678912:role/iam-test")
        val jsonString = "{\"topic_arn\":\"${sampleSns.topicArn}\",\"role_arn\":\"${sampleSns.roleArn}\"}"
        val recreatedObject = createObjectFromJsonString(jsonString) { Sns.parse(it) }
        Assertions.assertEquals(sampleSns, recreatedObject)
    }

    @Test
    fun `SNS should throw exception when invalid json object is passed`() {
        val jsonString = "sample message"
        assertThrows(JsonParseException::class.java) {
            createObjectFromJsonString(jsonString) { Sns.parse(it) }
        }
    }

    @Test
    fun `SNS should throw exception when arn is replace with arn2 in json object`() {
        val sampleSns = Sns("arn:aws:sns:us-east-1:012345678912:topic-test", "arn:aws:iam::012345678912:role/iam-test")
        val jsonString = "{\"topic_arn2\":\"${sampleSns.topicArn}\",\"role_arn\":\"${sampleSns.roleArn}\"}"
        assertThrows(IllegalArgumentException::class.java) {
            createObjectFromJsonString(jsonString) { Sns.parse(it) }
        }
    }

    @Test
    fun `SNS should safely ignore extra field in json object`() {
        val sampleSns = Sns("arn:aws:sns:us-east-1:012345678912:topic-test", null)
        val jsonString = "{\"topic_arn\":\"${sampleSns.topicArn}\", \"another\":\"field\"}"
        val recreatedObject = createObjectFromJsonString(jsonString) { Sns.parse(it) }
        Assertions.assertEquals(sampleSns, recreatedObject)
    }
}