notifications-api/tests/app/v2/inbound_sms/test_get_inbound_sms.py

import datetime
import pytest
from flask import json, url_for

from app import DATETIME_FORMAT
from tests import create_authorization_header
from tests.app.db import create_inbound_sms


def test_get_all_inbound_sms_returns_200(
        client, sample_service
):
    all_inbound_sms = [
        create_inbound_sms(service=sample_service, user_number='447700900111', content='Hi'),
        create_inbound_sms(service=sample_service, user_number='447700900112'),
        create_inbound_sms(service=sample_service, user_number='447700900111', content='Bye'),
        create_inbound_sms(service=sample_service, user_number='07700900113')
    ]

    auth_header = create_authorization_header(service_id=sample_service.id)
    response = client.get(
        path='/v2/received-text-messages',
        headers=[('Content-Type', 'application/json'), auth_header])

    assert response.status_code == 200
    assert response.headers['Content-type'] == 'application/json'

    json_response = json.loads(response.get_data(as_text=True))['inbound_sms_list']

    reversed_all_inbound_sms = sorted(all_inbound_sms, key=lambda sms: sms.created_at, reverse=True)

    expected_response = [i.serialize() for i in reversed_all_inbound_sms]

    assert json_response == expected_response


@pytest.mark.parametrize('inbound_sms_path,user_number', [
    ('v2_inbound_sms.get_all_inbound_sms', None),
    ('v2_inbound_sms.get_inbound_sms_by_number', '447700900111')
])
def test_get_all_inbound_sms_generate_page_links(
        client, sample_service, mocker, inbound_sms_path, user_number
):
    mocker.patch.dict("app.v2.inbound_sms.get_inbound_sms.current_app.config", {"API_PAGE_SIZE": 1})
    all_inbound_sms = [
        create_inbound_sms(service=sample_service, user_number='447700900111', content='Hi'),
        create_inbound_sms(service=sample_service, user_number='447700900111'),
    ]

    auth_header = create_authorization_header(service_id=sample_service.id)
    response = client.get(
        path=url_for(inbound_sms_path, user_number=user_number),
        headers=[('Content-Type', 'application/json'), auth_header])

    assert response.status_code == 200

    json_response = json.loads(response.get_data(as_text=True))
    expected_inbound_sms_list = [all_inbound_sms[-1].serialize()]

    assert json_response['inbound_sms_list'] == expected_inbound_sms_list
    assert url_for(
        inbound_sms_path,
        user_number=user_number,
        _external=True) == json_response['links']['current']
    assert url_for(
        inbound_sms_path,
        user_number=user_number,
        older_than=all_inbound_sms[-1].id,
        _external=True) == json_response['links']['next']


def test_get_all_inbound_sms_for_no_inbound_sms_returns_200(
        client, sample_service
):
    auth_header = create_authorization_header(service_id=sample_service.id)
    response = client.get(
        path='/v2/received-text-messages',
        headers=[('Content-Type', 'application/json'), auth_header])

    assert response.status_code == 200
    assert response.headers['Content-type'] == 'application/json'

    json_response = json.loads(response.get_data(as_text=True))['inbound_sms_list']

    expected_response = []

    assert json_response == expected_response


def test_get_inbound_sms_by_number_returns_200(
        client, sample_service
):
    sample_inbound_sms1 = create_inbound_sms(service=sample_service, user_number='447700900111')
    create_inbound_sms(service=sample_service, user_number='447700900112')
    sample_inbound_sms2 = create_inbound_sms(service=sample_service, user_number='447700900111')

    auth_header = create_authorization_header(service_id=sample_service.id)
    response = client.get(
        path='/v2/received-text-messages/{}'.format('07700900111'),
        headers=[('Content-Type', 'application/json'), auth_header])

    assert response.status_code == 200
    assert response.headers['Content-type'] == 'application/json'

    json_response = json.loads(response.get_data(as_text=True))['inbound_sms_list']

    expected_response = [sample_inbound_sms2.serialize(), sample_inbound_sms1.serialize()]

    assert json_response == expected_response


def test_get_inbound_sms_for_no_inbound_sms_returns_200(
        client, sample_service
):
    auth_header = create_authorization_header(service_id=sample_service.id)
    response = client.get(
        path='/v2/received-text-messages',
        headers=[('Content-Type', 'application/json'), auth_header])

    assert response.status_code == 200
    assert response.headers['Content-type'] == 'application/json'

    json_response = json.loads(response.get_data(as_text=True))['inbound_sms_list']

    expected_response = []

    assert json_response == expected_response


def test_get_inbound_sms_by_nonexistent_number(client, sample_service):
    auth_header = create_authorization_header(service_id=sample_service.id)
    response = client.get(
        path='/v2/received-text-messages/447700900000',
        headers=[('Content-Type', 'application/json'), auth_header])

    assert response.status_code == 200
    assert response.headers['Content-type'] == 'application/json'

    json_response = json.loads(response.get_data(as_text=True))['inbound_sms_list']
    expected_response = []

    assert json_response == expected_response


@pytest.mark.parametrize('invalid_number,expected_message', [
    ('0077700', 'Not enough digits'),
    ('123456789012', 'Not a UK mobile number'),
    ('invalid_number', 'Must not contain letters or symbols')
])
def test_get_inbound_sms_by_invalid_number(
        client, sample_service, invalid_number, expected_message):
    auth_header = create_authorization_header(service_id=sample_service.id)
    response = client.get(
        path='/v2/received-text-messages/{}'.format(invalid_number),
        headers=[('Content-Type', 'application/json'), auth_header])

    assert response.status_code == 400
    assert response.headers['Content-type'] == 'application/json'

    json_response = json.loads(response.get_data(as_text=True))
    assert json_response == {
        "errors": [
            {
                "error": "BadRequestError",
                "message": expected_message
            }
        ],
        "status_code": 400
    }