mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-09 23:02:13 -05:00
79 lines
3.0 KiB
Python
79 lines
3.0 KiB
Python
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
|