Refactored model and dao

tests/app/dao/test_inbound_numbers_dao.py

@@ -1,10 +1,11 @@
 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_allocate_inbound_number_to_service
+    dao_set_inbound_number_to_service
 )
 from app.models import InboundNumber
 
@@ -27,8 +28,9 @@ def test_get_available_inbound_numbers(notify_db, notify_db_session, sample_inbo
 
 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_allocate_inbound_number_to_service(service.id)
+    dao_set_inbound_number_to_service(service.id, numbers[0])
 
     res = InboundNumber.query.filter(InboundNumber.service_id == service.id).all()
 
@@ -37,18 +39,21 @@ def test_allocate_inbound_number_to_service(notify_db, notify_db_session, sample
 
 
 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_allocate_inbound_number_to_service(service.id)
+    dao_set_inbound_number_to_service(service.id, numbers[0])
 
-    with pytest.raises(IndexError) as e:
-        dao_allocate_inbound_number_to_service(service.id)
+    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 str(e.value) == 'No inbound numbers available'
+    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):

tests/app/dao/test_services_dao.py

@@ -8,7 +8,7 @@ from sqlalchemy.exc import IntegrityError
 from freezegun import freeze_time
 from app import db
 from app.dao.inbound_numbers_dao import (
-    dao_allocate_inbound_number_to_service, 
+    dao_set_inbound_number_to_service, 
     dao_get_available_inbound_numbers
 )
 from app.dao.services_dao import (
@@ -907,6 +907,6 @@ def test_dao_allocating_inbound_number_shows_on_service(notify_db_session, sampl
 
     service = create_service(service_name='test service')
 
-    dao_allocate_inbound_number_to_service(service.id)
+    dao_set_inbound_number_to_service(service.id, inbound_numbers[0])
 
     assert service.inbound_number.number == inbound_numbers[0].number