import pytest

from auth.management_api import Auth0ManagementAPI


@pytest.mark.parametrize("user_id", (("5d9330e029d2ef0e0e3ccdaa"), ("auth0|5d9330e029d2ef0e0e3ccdaa")))
def test_user_metadata_retrieving(mocker, app_fixture, user_id):
    mocker.patch.object(Auth0ManagementAPI, "get_management_token")
    mocker.patch.object(Auth0ManagementAPI, "get_user_metadata")

    Auth0ManagementAPI.get_management_token.return_value = "token"
    Auth0ManagementAPI.get_user_metadata.return_value = {"apollo": {"homeMarketID": 14}}

    auth_management = Auth0ManagementAPI()
    meta_data = auth_management.get_user_metadata(user_id)

    data_to_compare = {"homeMarketID": 14}

    assert meta_data["apollo"] == data_to_compare


@pytest.mark.parametrize("user_id", (("5d9330e029d2ef0e0e3ccdaa"), ("auth0|5d9330e029d2ef0e0e3ccdaa")))
def test_user_metadata_updating(mocker, app_fixture, user_id):
    mocker.patch.object(Auth0ManagementAPI, "get_management_token")
    mocker.patch.object(Auth0ManagementAPI, "update_user_metadata")

    auth_management = Auth0ManagementAPI()
    new_meta_data = {"apollo_go_job_category": {"id": 1, "name": "Executive"}}

    Auth0ManagementAPI.get_management_token.return_value = "token"
    Auth0ManagementAPI.update_user_metadata.return_value = {
        "apollo": {"homeMarketID": 14},
        "apollo_go_job_category": {"id": 1, "name": "Executive"},
    }

    meta_data = auth_management.update_user_metadata(user_id, new_meta_data)
    data_to_compare = {"apollo": {"homeMarketID": 14}, "apollo_go_job_category": {"id": 1, "name": "Executive"}}

    assert meta_data == data_to_compare