tests for send_sms through mmg/firetext

This commit is contained in:
Leo Hemsted
2016-06-01 15:59:44 +01:00
parent 0a8cb679d7
commit a3b847bf64
4 changed files with 151 additions and 5 deletions

View File

@@ -1,5 +1,8 @@
import pytest
import uuid
from datetime import (datetime, date)
import pytest
from app import db
from app.models import (
User,
@@ -20,7 +23,8 @@ from app.dao.api_key_dao import save_model_api_key
from app.dao.jobs_dao import dao_create_job
from app.dao.notifications_dao import dao_create_notification
from app.dao.invited_user_dao import save_invited_user
import uuid
from app.clients.sms.firetext import FiretextClient
from app.clients.sms.mmg import MMGClient
@pytest.fixture(scope='function')
@@ -527,3 +531,27 @@ def sample_notification_statistics(notify_db,
notify_db.session.add(stats)
notify_db.session.commit()
return stats
@pytest.fixture(scope='function')
def mock_firetext_client(mocker, statsd_client=None):
client = FiretextClient()
statsd_client = statsd_client or mocker.Mock()
current_app = mocker.Mock(config={
'FIRETEXT_API_KEY': 'foo',
'FIRETEXT_NUMBER': 'bar'
})
client.init_app(current_app, statsd_client)
return client
@pytest.fixture(scope='function')
def mock_mmg_client(mocker, statsd_client=None):
client = MMGClient()
statsd_client = statsd_client or mocker.Mock()()
current_app = mocker.Mock(config={
'MMG_API_KEY': 'foo',
'MMG_FROM_NUMBER': 'bar'
})
client.init_app(current_app, statsd_client)
return client