notifications-api/tests/app/inbound_number/test_rest.py

import pytest

from flask import url_for
import json

from app.models import InboundNumber
from app.dao.inbound_numbers_dao import (
    dao_get_inbound_numbers,
    dao_get_available_inbound_numbers,
    dao_get_inbound_number_for_service,
    dao_set_inbound_number_to_service
)

from tests.app.db import create_service, create_inbound_number


def test_rest_get_inbound_numbers(admin_request, sample_inbound_numbers):
    result = admin_request.get('inbound_number.get_inbound_numbers')

    assert len(result['data']) == len(sample_inbound_numbers)
    assert result['data'] == [i.serialize() for i in sample_inbound_numbers]


def test_rest_get_inbound_number(admin_request, notify_db_session, sample_service):
    inbound_number = create_inbound_number(number='1', provider='mmg', active=False, service_id=sample_service.id)

    result = admin_request.get(
        'inbound_number.get_inbound_number_for_service',
        service_id=sample_service.id
    )
    assert result['data'] == inbound_number.serialize()


def test_rest_set_number_to_several_services_returns_400(
        admin_request, notify_db_session, sample_service):
    service_1 = create_service(service_name='test service 1')
    inbound_number = create_inbound_number(number='1', provider='mmg', active=True, service_id=sample_service.id)
    create_inbound_number(number='2', provider='mmg', active=True, service_id=None)
    service_2 = create_service(service_name='test service 2')

    result = admin_request.post(
        'inbound_number.post_set_inbound_number_for_service',
        _expected_status=400,
        inbound_number_id=inbound_number.id,
        service_id=service_2.id
    )
    assert result['message'] == 'Inbound number already assigned'


def test_rest_set_multiple_numbers_to_a_service_returns_400(
        admin_request, notify_db_session, sample_service):
    create_inbound_number(number='1', provider='mmg', active=True, service_id=sample_service.id)
    inbound_number = create_inbound_number(number='2', provider='mmg', active=True, service_id=None)

    result = admin_request.post(
        'inbound_number.post_set_inbound_number_for_service',
        _expected_status=400,
        inbound_number_id=inbound_number.id,
        service_id=sample_service.id
    )
    assert result['message'] == 'Service already has an inbound number'


@pytest.mark.parametrize("active_flag,expected_flag_state", [("on", True), ("off", False)])
def test_rest_set_inbound_number_active_flag(
        admin_request, notify_db_session, active_flag, expected_flag_state):
    service = create_service(service_name='test service 1')
    inbound_number = create_inbound_number(
        number='1', provider='mmg', active=not expected_flag_state, service_id=service.id)

    admin_request.post(
        'inbound_number.post_set_inbound_number_{}'.format(active_flag),
        _expected_status=204,
        inbound_number_id=inbound_number.id
    )

    inbound_number_from_db = dao_get_inbound_number_for_service(service.id)
    assert inbound_number_from_db.active == expected_flag_state