from flask import url_for
from bs4 import BeautifulSoup
import json


def test_should_return_list_of_all_jobs(app_,
                                        service_one,
                                        api_user_active,
                                        mock_get_user,
                                        mock_get_user_by_email,
                                        mock_login,
                                        mock_get_jobs,
                                        mock_has_permissions):
    with app_.test_request_context():
        with app_.test_client() as client:
            client.login(api_user_active)
            response = client.get(url_for('main.view_jobs', service_id=101))

        assert response.status_code == 200
        page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser')
        assert page.h1.string == 'Notifications activity'
        jobs = page.tbody.find_all('tr')
        assert len(jobs) == 5


def test_should_show_page_for_one_job(
    app_,
    service_one,
    api_user_active,
    mock_login,
    mock_get_user,
    mock_get_user_by_email,
    mock_get_service,
    mock_get_service_template,
    job_data,
    mock_get_job,
    mock_get_notifications,
    mock_has_permissions
):
    service_id = job_data['service']
    job_id = job_data['id']
    file_name = job_data['original_file_name']

    with app_.test_request_context():
        with app_.test_client() as client:
            client.login(api_user_active)
            response = client.get(url_for('main.view_job', service_id=service_id, job_id=job_id))

        assert response.status_code == 200
        content = response.get_data(as_text=True)
        assert "Test Service: Your vehicle tax is about to expire" in content
        assert file_name in content


def test_should_show_updates_for_one_job_as_json(
    app_,
    service_one,
    api_user_active,
    mock_login,
    mock_get_user,
    mock_get_user_by_email,
    mock_get_service,
    mock_get_service_template,
    job_data,
    mock_get_job,
    mock_get_notifications,
    mock_has_permissions
):
    service_id = job_data['service']
    job_id = job_data['id']
    file_name = job_data['original_file_name']

    with app_.test_request_context():
        with app_.test_client() as client:
            client.login(api_user_active)
            response = client.get(url_for('main.view_job_updates', service_id=service_id, job_id=job_id))

        assert response.status_code == 200
        content = json.loads(response.get_data(as_text=True))
        assert 'processed' in content['counts']
        assert 'queued' in content['counts']
        assert 'Recipient' in content['notifications']
        assert 'Status' in content['notifications']
        assert 'Started' in content['status']


def test_should_show_notifications_for_a_service(app_,
                                                 service_one,
                                                 api_user_active,
                                                 mock_login,
                                                 mock_get_user,
                                                 mock_get_user_by_email,
                                                 mock_get_service,
                                                 mock_get_notifications,
                                                 mock_has_permissions):
    with app_.test_request_context():
        with app_.test_client() as client:
            client.login(api_user_active)
            response = client.get(url_for('main.view_notifications', service_id=service_one['id']))
        assert response.status_code == 200
        content = response.get_data(as_text=True)
        notifications = mock_get_notifications(service_one['id'])
        notification = notifications['notifications'][0]
        assert notification['to'] in content
        assert notification['status'] in content
        assert notification['template']['name'] in content
        assert '.csv' in content


def test_should_show_notifications_for_a_service_with_next_previous(app_,
                                                                    service_one,
                                                                    api_user_active,
                                                                    mock_login,
                                                                    mock_get_user,
                                                                    mock_get_user_by_email,
                                                                    mock_get_service,
                                                                    mock_get_notifications_with_previous_next,
                                                                    mock_has_permissions):
    with app_.test_request_context():
        with app_.test_client() as client:
            client.login(api_user_active)
            response = client.get(url_for('main.view_notifications', service_id=service_one['id'], page=2))
        assert response.status_code == 200
        content = response.get_data(as_text=True)
        assert url_for('main.view_notifications', service_id=service_one['id'], page=3) in content
        assert url_for('main.view_notifications', service_id=service_one['id'], page=1) in content
        assert 'Previous page' in content
        assert 'Next page' in content