darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-07-13 09:50:08 -04:00
Code Issues Packages Projects Releases Wiki Activity

Test add service completed.

Browse Source
This commit is contained in:
Nicholas Staples
2016-01-15 16:10:24 +00:00
parent 3b1d521c10
commit 262bbbac45
6 changed files with 55 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/__init__.py
View File

@@ -12,7 +12,7 @@ class TestClient(FlaskClient):
verify_codes_dao.add_code(user_id=user.id, code='12345', code_type='sms')
response = self.post(
url_for('main.two_factor'), data={'sms_code': '12345'})
assert response.status_code == 200
assert response.status_code == 302
def logout(self, user):
self.get(url_for("main.logout"))

2
tests/app/main/dao/test_verify_codes_dao.py
View File

@@ -3,7 +3,7 @@ from pytest import fail
from app.main.dao import verify_codes_dao
from app.main.encryption import check_hash
from tests.app.main import create_test_user, create_another_test_user
from tests import create_test_user, create_another_test_user
def test_insert_new_code_and_get_it_back(app_, db_, db_session):

48
tests/app/main/views/test_add_service.py
View File

@@ -1,5 +1,7 @@
from flask import url_for
from app.main.dao import verify_codes_dao, services_dao
from tests.app.main import create_test_user
from tests import create_test_user
from app.models import User
def test_get_should_render_add_service_template(app_, db_, db_session):
@@ -8,35 +10,51 @@ def test_get_should_render_add_service_template(app_, db_, db_session):
user = create_test_user('active')
client.login(user)
verify_codes_dao.add_code(user_id=user.id, code='12345', code_type='sms')
client.post('/two-factor', data={'sms_code': '12345'})
client.post(url_for('main.two_factor'), data={'sms_code': '12345'})
response = client.get('/add-service')
assert response.status_code == 200
assert 'Set up notifications for your service' in response.get_data(as_text=True)
def test_should_add_service_and_redirect_to_next_page(app_, db_, db_session):
def test_should_add_service_and_redirect_to_next_page(app_,
db_,
db_session,
mock_create_service,
mock_get_services):
with app_.test_request_context():
with app_.test_client() as client:
user = create_test_user('active')
user = User.query.first()
client.login(user)
verify_codes_dao.add_code(user_id=user.id, code='12345', code_type='sms')
client.post('/two-factor', data={'sms_code': '12345'})
response = client.post('/add-service', data={'service_name': 'testing the post'})
response = client.post(
url_for('main.add_service'),
data={'service_name': 'testing the post'})
assert response.status_code == 302
assert response.location == 'http://localhost/services/123/dashboard'
saved_service = services_dao.find_service_by_service_name('testing the post')
assert saved_service is not None
assert response.location == url_for('main.dashboard', service_id=123, _external=True)
assert mock_create_service.called
assert mock_get_services.called
def test_should_return_form_errors_when_service_name_is_empty(app_,
db_,
db_session):
db_session,
active_user):
with app_.test_request_context():
with app_.test_client() as client:
user = create_test_user('active')
client.login(user)
verify_codes_dao.add_code(user_id=user.id, code='12345', code_type='sms')
client.post('/two-factor', data={'sms_code': '12345'})
client.login(active_user)
response = client.post('/add-service', data={})
assert response.status_code == 200
assert 'Service name can not be empty' in response.get_data(as_text=True)
def test_should_return_form_errors_with_duplicate_service_name(app_,
db_,
db_session,
mock_get_services):
with app_.test_request_context():
with app_.test_client() as client:
user = User.query.first()
client.login(user)
response = client.post('/add-service', data={'service_name': 'service_one'})
assert response.status_code == 200
assert 'Service name already exists' in response.get_data(as_text=True)
assert mock_get_services.called