mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-09-08 08:28:15 -04:00
Added choose_service using services_dao.
This commit is contained in:
@@ -6,7 +6,8 @@ from flask.ext.migrate import Migrate, MigrateCommand
|
||||
from flask.ext.script import Manager
|
||||
from sqlalchemy.schema import MetaData
|
||||
from . import (
|
||||
create_test_user, create_another_test_user, service_json, TestClient)
|
||||
create_test_user, create_another_test_user, service_json, TestClient,
|
||||
get_test_user)
|
||||
from app import create_app, db
|
||||
|
||||
|
||||
@@ -66,6 +67,9 @@ def service_one(request):
|
||||
|
||||
@pytest.fixture(scope='function')
|
||||
def active_user(request, db_, db_session):
|
||||
usr = get_test_user()
|
||||
if usr:
|
||||
return usr
|
||||
return create_test_user('active')
|
||||
|
||||
|
||||
@@ -80,8 +84,13 @@ def mock_send_email(request, mocker):
|
||||
|
||||
|
||||
@pytest.fixture(scope='function')
|
||||
def mock_get_service(mocker):
|
||||
return mocker.patch('app.notifications_api_client.get_service')
|
||||
def mock_get_service(mocker, active_user):
|
||||
def _create(service_id):
|
||||
service = service_json(
|
||||
service_id, "Test Service", [active_user.id], limit=1000,
|
||||
active=False, restricted=True)
|
||||
return {'data': service, 'token': 1}
|
||||
return mocker.patch('app.notifications_api_client.get_service', side_effect=_create)
|
||||
|
||||
|
||||
@pytest.fixture(scope='function')
|
||||
@@ -125,3 +134,12 @@ def mock_get_services(mocker, active_user):
|
||||
mock_class = mocker.patch(
|
||||
'app.notifications_api_client.get_services', side_effect=_create)
|
||||
return mock_class
|
||||
|
||||
|
||||
@pytest.fixture(scope='function')
|
||||
def mock_delete_service(mocker, mock_get_service):
|
||||
def _delete(service_id):
|
||||
return mock_get_service.side_effect(service_id)
|
||||
mock_class = mocker.patch(
|
||||
'app.notifications_api_client.delete_service', side_effect=_delete)
|
||||
return mock_class
|
||||
|
||||
Reference in New Issue
Block a user