mirror of
https://github.com/GSA/notifications-api.git
synced 2026-08-24 16:23:44 -04:00
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
import pytest
|
|
from sqlalchemy.exc import IntegrityError
|
|
|
|
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 app.models import InboundNumber
|
|
|
|
from tests.app.db import create_service
|
|
|
|
|
|
def test_get_inbound_numbers(notify_db, notify_db_session, sample_inbound_numbers):
|
|
res = dao_get_inbound_numbers()
|
|
|
|
assert len(res) == 3
|
|
assert res == sample_inbound_numbers
|
|
|
|
|
|
def test_get_available_inbound_numbers(notify_db, notify_db_session, sample_inbound_numbers):
|
|
res = dao_get_available_inbound_numbers()
|
|
|
|
assert len(res) == 1
|
|
assert res[0] == sample_inbound_numbers[0]
|
|
|
|
|
|
def test_allocate_inbound_number_to_service(notify_db, notify_db_session, sample_inbound_numbers):
|
|
service = create_service(service_name='test service')
|
|
numbers = dao_get_available_inbound_numbers()
|
|
|
|
dao_set_inbound_number_to_service(service.id, numbers[0])
|
|
|
|
res = InboundNumber.query.filter(InboundNumber.service_id == service.id).all()
|
|
|
|
assert len(res) == 1
|
|
assert res[0].service_id == service.id
|
|
|
|
|
|
def test_allocating_a_service_twice_will_raise_an_error(notify_db, notify_db_session, sample_inbound_numbers):
|
|
from tests.app.db import create_inbound_number
|
|
create_inbound_number(number='4', provider='mmg')
|
|
service = create_service(service_name='test service')
|
|
numbers = dao_get_available_inbound_numbers()
|
|
|
|
dao_set_inbound_number_to_service(service.id, numbers[0])
|
|
|
|
with pytest.raises(IntegrityError) as e:
|
|
dao_set_inbound_number_to_service(service.id, numbers[1])
|
|
|
|
res = InboundNumber.query.filter(InboundNumber.service_id == service.id).all()
|
|
|
|
assert len(res) == 1
|
|
assert res[0].service_id == service.id
|
|
assert 'duplicate key value violates unique constraint' in str(e.value)
|
|
|
|
|
|
def test_get_inbound_number_for_service(notify_db, notify_db_session, sample_inbound_numbers, sample_service):
|
|
res = dao_get_inbound_number_for_service(sample_service.id)
|
|
|
|
assert len(res) == 1
|
|
assert res[0].service_id == sample_service.id
|