mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-10 23:32:27 -05:00
104 lines
4.0 KiB
Python
104 lines
4.0 KiB
Python
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_service_to_several_inbound_numbers_returns_400(
|
|
admin_request, notify_db_session, sample_service):
|
|
service_1 = create_service(service_name='test service 1')
|
|
create_inbound_number(number='1', provider='mmg', active=False, service_id=sample_service.id)
|
|
inbound_number = create_inbound_number(number='2', provider='mmg', active=True, service_id=service_1.id)
|
|
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'] == 'No inbound numbers available'
|
|
|
|
|
|
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_number_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'
|
|
|
|
|
|
def test_rest_set_inbound_number_active_flag_off(admin_request, notify_db_session):
|
|
service = create_service(service_name='test service 1')
|
|
inbound_number = create_inbound_number(number='1', provider='mmg', active=True, service_id=service.id)
|
|
|
|
admin_request.post(
|
|
'inbound_number.post_set_inbound_number_off',
|
|
_expected_status=204,
|
|
inbound_number_id=inbound_number.id
|
|
)
|
|
|
|
inbound_number_off = dao_get_inbound_number_for_service(service.id)
|
|
assert not inbound_number_off.active
|
|
|
|
|
|
def test_rest_set_inbound_number_active_flag_on(admin_request, notify_db_session):
|
|
service = create_service(service_name='test service 1')
|
|
inbound_number = create_inbound_number(number='1', provider='mmg', active=False, service_id=service.id)
|
|
|
|
admin_request.post(
|
|
'inbound_number.post_set_inbound_number_on',
|
|
_expected_status=204,
|
|
inbound_number_id=inbound_number.id
|
|
)
|
|
|
|
inbound_number_on = dao_get_inbound_number_for_service(service.id)
|
|
assert inbound_number_on.active
|