Add dao methods to retrieve providers, switch, toggle along with tests

This commit is contained in:
Imdad Ahad
2017-01-17 15:38:54 +00:00
parent 56f1ed11d6
commit 2896ed27f2
3 changed files with 190 additions and 10 deletions

View File

@@ -1,9 +1,10 @@
import requests_mock
import pytest
import uuid
from datetime import (datetime, date, timedelta)
from sqlalchemy import asc
from sqlalchemy.orm.session import make_transient
import requests_mock
import pytest
from flask import current_app
from app import db
@@ -34,7 +35,11 @@ from app.dao.notifications_dao import dao_create_notification
from app.dao.invited_user_dao import save_invited_user
from app.dao.provider_rates_dao import create_provider_rates
from app.clients.sms.firetext import FiretextClient
from app.dao.provider_details_dao import (
dao_update_provider_details,
get_provider_details_by_identifier,
get_alternative_sms_provider
)
from tests.app.db import create_user
@@ -628,6 +633,27 @@ def fake_uuid():
return "6ce466d0-fd6a-11e5-82f5-e0accb9d11a6"
@pytest.fixture(scope='function')
def current_sms_provider():
return ProviderDetails.query.filter_by(
notification_type='sms'
).order_by(
asc(ProviderDetails.priority)
).first()
@pytest.fixture(scope='function')
def set_primary_sms_provider(identifier='mmg'):
primary_provider = get_provider_details_by_identifier(identifier)
secondary_provider = get_alternative_sms_provider(identifier)
primary_provider.priority = 10
secondary_provider.priority = 20
dao_update_provider_details(primary_provider)
dao_update_provider_details(secondary_provider)
@pytest.fixture(scope='function')
def ses_provider():
return ProviderDetails.query.filter_by(identifier='ses').one()
@@ -635,7 +661,7 @@ def ses_provider():
@pytest.fixture(scope='function')
def firetext_provider():
return ProviderDetails.query.filter_by(identifier='mmg').one()
return ProviderDetails.query.filter_by(identifier='firetext').one()
@pytest.fixture(scope='function')