import demistomock as demisto
from SymantecManagementCenter import LOCAL_CATEGORY_DB_TYPE, main


def test_add_policy_content_local_category_db(mocker, requests_mock):
    """
    Given:
     - URL to add to as content to category of type LOCAL_CATEGORY_DB

    When:
     - Running the add content command

    Then:
     - Ensure the expected request body is sent
     - Verify expected human readable output
    """
    content = {
        "content": {
            "categories": [
                {"type": "inline", "name": "category1", "entries": [{"type": "url", "url": "www.demisto.com", "comment": None}]},
                {
                    "type": "inline",
                    "name": "category2",
                    "entries": [
                        {"type": "url", "url": "www.google.com", "comment": "comment"},
                        {"type": "url", "url": "www.apple.com", "comment": "comment"},
                    ],
                },
                {
                    "type": "inline",
                    "name": "category3",
                    "entries": [{"type": "url", "url": "www.demisto.com", "comment": "comment"}],
                },
                {
                    "type": "inline",
                    "name": "category4",
                    "entries": [{"type": "url", "url": "www.google.com", "comment": "comment"}],
                },
                {"type": "inline", "name": "category5", "entries": [{"type": "url", "url": "8.8.8.8", "comment": "comment"}]},
                {
                    "type": "inline",
                    "name": "category6",
                    "entries": [
                        {"type": "url", "url": "www.demisto.com", "comment": "comment"},
                        {"type": "url", "url": "www.paloaltonetworks.net", "comment": "comment"},
                    ],
                },
            ]
        },
        "schemaVersion": "1.0",
        "revisionInfo": {
            "revisionNumber": "1.12",
            "revisionDescription": "desc",
            "author": "admin",
            "revisionDate": "2020-01-25T14:58:06",
        },
    }
    url_to_add = "www.cnn.com"
    comment = "test comment"
    mocker.patch.object(demisto, "command", return_value="symantec-mc-add-policy-content")
    mocker.patch.object(
        demisto, "params", return_value={"url": "https://server", "credentials": {"identifier": "user", "password": "password"}}
    )
    mocker.patch.object(
        demisto,
        "args",
        return_value={
            "uuid": "uuid",
            "content_type": LOCAL_CATEGORY_DB_TYPE,
            "change_description": "desc",
            "category": "category2",
            "url": url_to_add,
            "description": comment,
        },
    )
    mocker.patch.object(demisto, "results")
    requests_mock.get("https://server/api/policies/uuid/content", json=content)
    requests_mock.post("https://server/api/policies/uuid/content", json={})
    main()
    expected_request_body = dict(content)
    expected_request_body["content"]["categories"][1]["entries"].append({"type": "url", "url": url_to_add, "comment": comment})
    assert requests_mock.request_history[1].json()["content"] == expected_request_body["content"]
    assert demisto.results.call_count == 1
    results = demisto.results.call_args[0][0]
    assert (
        results["HumanReadable"]
        == """### Successfully added content to the policy
|category|change_description|content_type|description|url|uuid|
|---|---|---|---|---|---|
| category2 | desc | LOCAL_CATEGORY_DB | test comment | www.cnn.com | uuid |
"""
    )


def test_delete_policy_content_local_category_db(mocker, requests_mock):
    """
    Given:
     - IP to delete from content in category of type LOCAL_CATEGORY_DB

    When:
     - Running the delete content command

    Then:
     - Ensure the expected request body is sent
     - Verify expected human readable output
    """
    content = {
        "content": {
            "categories": [
                {"type": "inline", "name": "category1", "entries": [{"type": "url", "url": "www.demisto.com", "comment": None}]},
                {
                    "type": "inline",
                    "name": "category2",
                    "entries": [
                        {"type": "url", "url": "www.google.com", "comment": "comment"},
                        {"type": "url", "url": "www.apple.com", "comment": "comment"},
                    ],
                },
                {
                    "type": "inline",
                    "name": "category3",
                    "entries": [{"type": "url", "url": "www.demisto.com", "comment": "comment"}],
                },
                {
                    "type": "inline",
                    "name": "category4",
                    "entries": [{"type": "url", "url": "www.google.com", "comment": "comment"}],
                },
                {"type": "inline", "name": "category5", "entries": [{"type": "url", "url": "8.8.8.8", "comment": "comment"}]},
                {
                    "type": "inline",
                    "name": "category6",
                    "entries": [
                        {"type": "url", "url": "www.demisto.com", "comment": "comment"},
                        {"type": "url", "url": "www.paloaltonetworks.net", "comment": "comment"},
                    ],
                },
            ]
        },
        "schemaVersion": "1.0",
        "revisionInfo": {
            "revisionNumber": "1.12",
            "revisionDescription": "desc",
            "author": "admin",
            "revisionDate": "2020-01-25T14:58:06",
        },
    }
    ip_to_delete = "8.8.8.8"
    mocker.patch.object(demisto, "command", return_value="symantec-mc-delete-policy-content")
    mocker.patch.object(
        demisto, "params", return_value={"url": "https://server", "credentials": {"identifier": "user", "password": "password"}}
    )
    mocker.patch.object(
        demisto,
        "args",
        return_value={
            "uuid": "uuid",
            "content_type": LOCAL_CATEGORY_DB_TYPE,
            "change_description": "desc",
            "category": "category5",
            "ip": ip_to_delete,
        },
    )
    mocker.patch.object(demisto, "results")
    requests_mock.get("https://server/api/policies/uuid/content", json=content)
    requests_mock.post("https://server/api/policies/uuid/content", json={})
    main()
    expected_request_body = dict(content)
    del expected_request_body["content"]["categories"][4]
    assert requests_mock.request_history[1].json()["content"] == expected_request_body["content"]
    assert demisto.results.call_count == 1
    results = demisto.results.call_args[0][0]
    assert (
        results["HumanReadable"]
        == """### Successfully deleted content from the policy
|CategoryName|ObjectDeleted|
|---|---|
| category5 | 8.8.8.8 |
| category5 | The category was deleted. Category cannot be empty. |
"""
    )


def test_delete_policy_content_local_category_db_negative(mocker, requests_mock):
    """
    Given:
     - IP, which does not exist in the content, to delete from content in category of type LOCAL_CATEGORY_DB

    When:
     - Running the delete content command

    Then:
     - Ensure an exception is raised
     - Verify error message is raised as expected
    """
    content = {
        "content": {
            "categories": [
                {"type": "inline", "name": "category1", "entries": [{"type": "url", "url": "www.demisto.com", "comment": None}]},
                {
                    "type": "inline",
                    "name": "category2",
                    "entries": [
                        {"type": "url", "url": "www.google.com", "comment": "comment"},
                        {"type": "url", "url": "www.apple.com", "comment": "comment"},
                    ],
                },
                {
                    "type": "inline",
                    "name": "category3",
                    "entries": [{"type": "url", "url": "www.demisto.com", "comment": "comment"}],
                },
                {
                    "type": "inline",
                    "name": "category4",
                    "entries": [{"type": "url", "url": "www.google.com", "comment": "comment"}],
                },
                {"type": "inline", "name": "category5", "entries": [{"type": "url", "url": "8.8.8.8", "comment": "comment"}]},
                {
                    "type": "inline",
                    "name": "category6",
                    "entries": [
                        {"type": "url", "url": "www.demisto.com", "comment": "comment"},
                        {"type": "url", "url": "www.paloaltonetworks.net", "comment": "comment"},
                    ],
                },
            ]
        },
        "schemaVersion": "1.0",
        "revisionInfo": {
            "revisionNumber": "1.12",
            "revisionDescription": "desc",
            "author": "admin",
            "revisionDate": "2020-01-25T14:58:06",
        },
    }
    ip_which_does_not_exist = "4.4.4.4"
    mocker.patch.object(demisto, "command", return_value="symantec-mc-delete-policy-content")
    mocker.patch.object(
        demisto, "params", return_value={"url": "https://server", "credentials": {"identifier": "user", "password": "password"}}
    )
    mocker.patch.object(
        demisto,
        "args",
        return_value={
            "uuid": "uuid",
            "content_type": LOCAL_CATEGORY_DB_TYPE,
            "change_description": "desc",
            "category": "category5",
            "ip": ip_which_does_not_exist,
        },
    )
    requests_mock.get("https://server/api/policies/uuid/content", json=content)
    return_error_mock = mocker.patch("SymantecManagementCenter.return_error")
    main()
    err_msg = return_error_mock.call_args[0][0]
    assert "Deletion failed - Could not find object to delete." in err_msg


def test_update_policy_content_local_category_db(mocker, requests_mock):
    """
    Given:
     - URL to update its comment in a category of type LOCAL_CATEGORY_DB

    When:
     - Running the update content command

    Then:
     - Ensure the expected request body is sent
     - Verify expected human readable output
    """
    content = {
        "content": {
            "categories": [
                {"type": "inline", "name": "category1", "entries": [{"type": "url", "url": "www.demisto.com", "comment": None}]},
                {
                    "type": "inline",
                    "name": "category2",
                    "entries": [
                        {"type": "url", "url": "www.google.com", "comment": "comment"},
                        {"type": "url", "url": "www.apple.com", "comment": "comment"},
                    ],
                },
                {
                    "type": "inline",
                    "name": "category3",
                    "entries": [{"type": "url", "url": "www.demisto.com", "comment": "comment"}],
                },
                {
                    "type": "inline",
                    "name": "category4",
                    "entries": [{"type": "url", "url": "www.google.com", "comment": "comment"}],
                },
                {"type": "inline", "name": "category5", "entries": [{"type": "url", "url": "8.8.8.8", "comment": "comment"}]},
                {
                    "type": "inline",
                    "name": "category6",
                    "entries": [
                        {"type": "url", "url": "www.demisto.com", "comment": "comment"},
                        {"type": "url", "url": "www.paloaltonetworks.net", "comment": "comment"},
                    ],
                },
            ]
        },
        "schemaVersion": "1.0",
        "revisionInfo": {
            "revisionNumber": "1.12",
            "revisionDescription": "desc",
            "author": "admin",
            "revisionDate": "2020-01-25T14:58:06",
        },
    }
    url_to_update = "www.demisto.com"
    comment = "new comment"
    mocker.patch.object(demisto, "command", return_value="symantec-mc-update-policy-content")
    mocker.patch.object(
        demisto, "params", return_value={"url": "https://server", "credentials": {"identifier": "user", "password": "password"}}
    )
    mocker.patch.object(
        demisto,
        "args",
        return_value={
            "uuid": "uuid",
            "content_type": LOCAL_CATEGORY_DB_TYPE,
            "change_description": "desc",
            "category": "category1",
            "url": url_to_update,
            "description": comment,
        },
    )
    mocker.patch.object(demisto, "results")
    requests_mock.get("https://server/api/policies/uuid/content", json=content)
    requests_mock.post("https://server/api/policies/uuid/content", json={})
    main()
    expected_request_body = dict(content)
    expected_request_body["content"]["categories"][0]["entries"][0] = {"type": "url", "url": url_to_update, "comment": comment}
    assert requests_mock.request_history[1].json()["content"] == expected_request_body["content"]
    assert demisto.results.call_count == 1
    results = demisto.results.call_args[0][0]
    assert (
        results["HumanReadable"]
        == """### Successfully updated content in the policy
|category|change_description|content_type|description|url|uuid|
|---|---|---|---|---|---|
| category1 | desc | LOCAL_CATEGORY_DB | new comment | www.demisto.com | uuid |
"""
    )


def test_update_policy_content_local_category_db_negative(mocker, requests_mock):
    """
    Given:
     - IP, which does not exist in the content, to update in category of type LOCAL_CATEGORY_DB

    When:
     - Running the update content command

    Then:
     - Ensure an exception is raised
     - Verify error message is raised as expected
    """
    content = {
        "content": {
            "categories": [
                {"type": "inline", "name": "category1", "entries": [{"type": "url", "url": "www.demisto.com", "comment": None}]},
                {
                    "type": "inline",
                    "name": "category2",
                    "entries": [
                        {"type": "url", "url": "www.google.com", "comment": "comment"},
                        {"type": "url", "url": "www.apple.com", "comment": "comment"},
                    ],
                },
                {
                    "type": "inline",
                    "name": "category3",
                    "entries": [{"type": "url", "url": "www.demisto.com", "comment": "comment"}],
                },
                {
                    "type": "inline",
                    "name": "category4",
                    "entries": [{"type": "url", "url": "www.google.com", "comment": "comment"}],
                },
                {"type": "inline", "name": "category5", "entries": [{"type": "url", "url": "8.8.8.8", "comment": "comment"}]},
                {
                    "type": "inline",
                    "name": "category6",
                    "entries": [
                        {"type": "url", "url": "www.demisto.com", "comment": "comment"},
                        {"type": "url", "url": "www.paloaltonetworks.net", "comment": "comment"},
                    ],
                },
            ]
        },
        "schemaVersion": "1.0",
        "revisionInfo": {
            "revisionNumber": "1.12",
            "revisionDescription": "desc",
            "author": "admin",
            "revisionDate": "2020-01-25T14:58:06",
        },
    }
    ip_which_does_not_exist = "4.4.4.4"
    mocker.patch.object(demisto, "command", return_value="symantec-mc-update-policy-content")
    mocker.patch.object(
        demisto, "params", return_value={"url": "https://server", "credentials": {"identifier": "user", "password": "password"}}
    )
    mocker.patch.object(
        demisto,
        "args",
        return_value={
            "uuid": "uuid",
            "content_type": LOCAL_CATEGORY_DB_TYPE,
            "change_description": "desc",
            "category": "category5",
            "ip": ip_which_does_not_exist,
            "description": "some desc",
        },
    )
    requests_mock.get("https://server/api/policies/uuid/content", json=content)
    return_error_mock = mocker.patch("SymantecManagementCenter.return_error")
    main()
    err_msg = return_error_mock.call_args[0][0]
    assert "Update failed - Could not find object to update." in err_msg