from enum import IntEnum
from flask import url_for
from bs4 import BeautifulSoup

from app import format_datetime_short

valid_letter_jobs = [
    {
        'service_name': {'name': 'test_name'},
        'id': 'test_id',
        'notification_count': 2,
        'job_status': 'ready to send',
        'created_at': '2017-04-01T12:00:00'
    },
    {
        'service_name': {'name': 'test_name 2'},
        'id': 'test_id 2',
        'notification_count': 1,
        'job_status': 'sent to dvla',
        'created_at': '2017-04-02T13:00:00'
    },
    {
        'service_name': {'name': 'test_name 3'},
        'id': 'test_id 3',
        'notification_count': 1,
        'job_status': 'in progress',
        'created_at': '2017-04-03T14:00:00'
    }
]

send_letter_jobs_response = {"response": "Task created to send files to DVLA"}


class letter_jobs_header(IntEnum):
    SERVICE_NAME = 0
    JOB_ID = 1
    NOTIFICATION_COUNT = 2
    JOB_STATUS = 3
    CREATED_AT = 4
    CHECKBOX = 5
    TEMP_STATUS = 6


def test_get_letter_jobs_returns_list_of_all_letter_jobs(logged_in_platform_admin_client, mocker):
    mock_get_letters = mocker.patch('app.letter_jobs_client.get_letter_jobs', return_value=valid_letter_jobs)

    response = logged_in_platform_admin_client.get(url_for('main.letter_jobs'))

    assert mock_get_letters.called
    assert response.status_code == 200

    page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser')
    assert page.h1.string == 'Letter jobs'

    table = page.find('table')
    table_body = table.find('tbody')
    rows = table_body.find_all('tr')

    assert len(rows) == len(valid_letter_jobs)

    for row_pos in range(len(rows)):
        cols = rows[row_pos].find_all('td')
        assert valid_letter_jobs[row_pos]['service_name']['name'] == cols[letter_jobs_header.SERVICE_NAME].text
        assert valid_letter_jobs[row_pos]['id'] == cols[letter_jobs_header.JOB_ID].text
        assert valid_letter_jobs[row_pos]['notification_count'] == int(cols[letter_jobs_header.NOTIFICATION_COUNT].text)
        assert valid_letter_jobs[row_pos]['job_status'] == cols[letter_jobs_header.JOB_STATUS].text
        assert format_datetime_short(
            valid_letter_jobs[row_pos]['created_at']) == cols[letter_jobs_header.CREATED_AT].text
        if not (valid_letter_jobs[row_pos]['job_status'] == 'ready to send' or
                valid_letter_jobs[row_pos]['job_status'] == 'sent to dvla'):
            assert 'disabled' in str(cols[letter_jobs_header.CHECKBOX])


def test_post_letter_jobs_select_1_letter_job_submits_1_job(logged_in_platform_admin_client, mocker):
    letter_jobs_first_selected = {'job_id': ['test_id']}

    mock_get_letters = mocker.patch('app.letter_jobs_client.get_letter_jobs', return_value=valid_letter_jobs)
    mock_send_letters = mocker.patch('app.letter_jobs_client.send_letter_jobs', return_value=send_letter_jobs_response)

    response = logged_in_platform_admin_client.post(url_for('main.letter_jobs'), data=letter_jobs_first_selected)

    assert mock_get_letters.called
    assert mock_send_letters.called
    assert response.status_code == 200

    page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser')

    table = page.find('table')
    table_body = table.find('tbody')
    rows = table_body.find_all('tr')

    assert len(rows) == len(valid_letter_jobs)

    colr0 = rows[0].find_all('td')
    colr1 = rows[1].find_all('td')
    colr2 = rows[2].find_all('td')

    assert colr0[letter_jobs_header.TEMP_STATUS].text == "sending"
    assert colr1[letter_jobs_header.TEMP_STATUS].text == ""
    assert colr2[letter_jobs_header.TEMP_STATUS].text == ""

    message = page.find('p', attrs={'id': 'message'}).text
    assert "Task created to send files to DVLA" in message


def test_post_letter_jobs_none_selected_shows_message(logged_in_platform_admin_client, mocker):
    mock_get_letters = mocker.patch('app.letter_jobs_client.get_letter_jobs', return_value=valid_letter_jobs)
    mock_send_letters = mocker.patch('app.letter_jobs_client.send_letter_jobs', return_value=send_letter_jobs_response)

    response = logged_in_platform_admin_client.post(url_for('main.letter_jobs'), data={})

    assert mock_get_letters.called
    assert not mock_send_letters.called
    assert response.status_code == 200

    page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser')
    message = page.find('p', attrs={'id': 'message'}).text

    assert "No jobs selected" in message

@@ -1483,37 +1483,36 @@
  jobs'%0A%0A    
-table
+rows
  = page.
 find('table'
@@ -1503,91 +1503,28 @@
 age.
-find('table')%0A    table_body = table.find('tbody')%0A    rows = 
+select('
 table
-_
+ t
 body
-.find_all('
+ 
 tr')
@@ -3138,21 +3138,20 @@
     
-table
+rows
  = page.
 find
@@ -3150,91 +3150,28 @@
 age.
-find('table')%0A    table_body = table.find('tbody')%0A    rows = 
+select('
 table
-_
+ t
 body
-.find_all('
+ 
 tr')

Refactor test_letter_jobs