import json

import demistomock as demisto


def executeCommand(name, args=None):
    if name == "core-api-get" and args and "uri" in args and args["uri"] == "/settings/integration-commands":
        file_name = "TestData/integration_commands.json"
    elif name == "core-api-post" and args and "uri" in args and args["uri"] == "/settings/integration/search":
        file_name = "TestData/integration_search.json"
    else:
        raise ValueError(f"Unimplemented command called: {name}")

    with open(file_name) as f:
        raw_data = f.read()
        data = json.loads(raw_data)
        return data


def test_main(mocker):
    from ProvidesCommand import main

    mocker.patch.object(demisto, "executeCommand", side_effect=executeCommand)

    mocker.patch.object(demisto, "args", return_value={"command": "send-mail"})
    mocker.patch.object(demisto, "results")
    main()
    assert demisto.results.call_count == 1
    results = demisto.results.call_args
    assert results[0][0] == "EWS Mail Sender,Gmail,mail-sender,Mail Sender (New)"

    mocker.patch.object(demisto, "args", return_value={"command": "send-mail", "enabled": "true"})
    mocker.patch.object(demisto, "results")
    main()
    assert demisto.results.call_count == 1
    results = demisto.results.call_args
    assert results[0][0] == "Mail Sender (New)"

    mocker.patch.object(demisto, "args", return_value={"command": "send-mail", "enabled": "false"})
    mocker.patch.object(demisto, "results")
    main()
    assert demisto.results.call_count == 1
    results = demisto.results.call_args
    assert results[0][0] == "EWS Mail Sender,Gmail,mail-sender"