import demistomock as demisto  # noqa: F401
import pytest
from CommonServerPython import *  # noqa: F401
from PrintContext import print_context

CONTEXT_DATA = [{"data1": 1}]


@pytest.mark.parametrize("expected_content, fmt, ctx", [(CONTEXT_DATA, "json", CONTEXT_DATA), ("Context empty.", "table", [])])
def test_print_context(mocker, expected_content, fmt, ctx):
    """
    Given:
        - The script args.
    When:
        - Running the print_context function.
    Then:
        - Validating the outputs as expected.
    """
    results_mock = mocker.patch.object(demisto, "results")
    print_context(fmt, ctx)
    res = results_mock.call_args[0][0]
    assert expected_content == res


@pytest.mark.parametrize(
    "expected_content, fmt, ctx",
    [
        ("**Context data**:\n```\n", "markdown", CONTEXT_DATA),
    ],
)
def test_print_context_markdown(mocker, expected_content, fmt, ctx):
    """
    Given:
        - The script args.
    When:
        - Running the print_context function.
    Then:
        - Validating the outputs as expected.
    """
    results_mock = mocker.patch.object(demisto, "results")
    print_context(fmt, ctx)
    res = results_mock.call_args[0][0]["Contents"]
    assert expected_content in res