import demistomock as demisto  # noqa: F401
from CommonServerPython import *  # noqa: F401

"""Tests the script functions of the Json2HtmlTable script"""

import pytest
import Json2HtmlTable


@pytest.mark.parametrize(
    "value,exptected", [(None, None), ('{"test": "value"}', {"test": "value"}), (['{"test": "value"}'], {"test": "value"})]
)
def test_get_json_from_string(value, exptected):
    """Tests the get_json_from_string function"""

    result: dict | str | None = Json2HtmlTable.get_json_from_string(value=value)
    assert result == exptected


@pytest.mark.parametrize(
    "value,expected",
    [
        ('{"test": "value"}', '<table border="1"><tr><th>test</th><td>value</td></tr></table>'),
        (['{"test": "value"}'], '<table border="1"><tr><th>test</th><td>value</td></tr></table>'),
    ],
)
def test_json_to_html_command(value, expected):
    demisto_args: dict = {"value": value}
    result: CommandResults = Json2HtmlTable.json_to_html_command(args=demisto_args)
    assert result.outputs == expected


@pytest.mark.parametrize(
    "value,attributes,styling,expected",
    [
        (
            '{"test": "value"}',
            'style="table-layout: fixed; width: 100%;"',
            None,
            '<table style="table-layout: fixed; width: 100%;"><tr><th>test</th><td>value</td></tr></table>',
        ),
        (
            ['{"test": "value"}'],
            'style="table-layout: fixed; width: 100%;"',
            None,
            '<table style="table-layout: fixed; width: 100%;"><tr><th>test</th><td>value</td></tr></table>',
        ),
        (
            '{"test": "value"}',
            'style="table-layout: fixed; width: 100%;" id="my-custom-table"',
            "#my-custom-table: {background-color: red;}",
            (
                '<style>#my-custom-table: {background-color: red;}</style><table style="table-layout: fixed; width: 100%;" id="my'
                '-custom-table"><tr><th>test</th><td>value</td></tr></table>'
            ),
        ),
    ],
)
def test_json_to_html_with_attributes(value, attributes, styling, expected):
    demisto_args: dict = {"value": value, "table_attributes": attributes, "custom_styling": styling}
    result: CommandResults = Json2HtmlTable.json_to_html_command(args=demisto_args)
    assert result.outputs == expected