import uuid

import pytest

from app.notify_client.billing_api_client import BillingAPIClient


def test_get_free_sms_fragment_limit_for_year_correct_endpoint(mocker, api_user_active):
    service_id = uuid.uuid4()
    expected_url = '/service/{}/billing/free-sms-fragment-limit'.format(service_id)
    client = BillingAPIClient()

    mock_get = mocker.patch('app.notify_client.billing_api_client.BillingAPIClient.get')

    client.get_free_sms_fragment_limit_for_year(service_id, year=1999)
    mock_get.assert_called_once_with(expected_url, params={'financial_year_start': 1999})


def test_post_free_sms_fragment_limit_for_current_year_endpoint(mocker, api_user_active):
    service_id = uuid.uuid4()
    sms_limit_data = {'free_sms_fragment_limit': 1111, 'financial_year_start': None}
    mock_post = mocker.patch('app.notify_client.billing_api_client.BillingAPIClient.post')
    client = BillingAPIClient()

    client.create_or_update_free_sms_fragment_limit(service_id=service_id, free_sms_fragment_limit=1111)

    mock_post.assert_called_once_with(
        url='/service/{}/billing/free-sms-fragment-limit'.format(service_id),
        data=sms_limit_data
    )


def test_post_free_sms_fragment_limit_for_year_endpoint(mocker, api_user_active):
    service_id = uuid.uuid4()
    sms_limit_data = {'free_sms_fragment_limit': 1111, 'financial_year_start': 2017}
    mock_post = mocker.patch('app.notify_client.billing_api_client.BillingAPIClient.post')
    client = BillingAPIClient()

    client.create_or_update_free_sms_fragment_limit(service_id=service_id,
                                                    free_sms_fragment_limit=1111,
                                                    year=2017)
    mock_post.assert_called_once_with(
        url='/service/{}/billing/free-sms-fragment-limit'.format(service_id),
        data=sms_limit_data
    )


@pytest.mark.parametrize('func, expected_url', [
    (BillingAPIClient.get_data_for_volumes_by_service_report, '/platform-stats/volumes-by-service'),
    (BillingAPIClient.get_data_for_daily_volumes_report, '/platform-stats/daily-volumes-report'),
    (
        BillingAPIClient.get_data_for_daily_sms_provider_volumes_report,
        '/platform-stats/daily-sms-provider-volumes-report'
    ),
])
def test_get_data_for_volume_reports(mocker, api_user_active, func, expected_url):
    mock_get = mocker.patch('app.notify_client.billing_api_client.BillingAPIClient.get')
    client = BillingAPIClient()

    func(client, '2022-03-01', '2022-03-31')

    mock_get.assert_called_once_with(url=expected_url, params={'start_date': '2022-03-01', 'end_date': '2022-03-31'})