110067722: Added endpoints for add-service

Post is not complete as of yet.

tests/app/main/dao/test_service_dao.py

@@ -1,10 +1,12 @@
+import pytest
+import sqlalchemy
+
 from app.main.dao import services_dao
 from tests.app.main import create_test_user
 
 
 def test_can_insert_and_retrieve_new_service(notifications_admin, notifications_admin_db, notify_db_session):
     user = create_test_user()
-
     id = services_dao.insert_new_service('testing service', user)
     saved_service = services_dao.get_service_by_id(id)
     assert id == saved_service.id
@@ -37,3 +39,24 @@ def test_get_service_returns_none_if_service_does_not_exist(notifications_admin,
                                                             notify_db_session):
     service = services_dao.get_service_by_id(1)
     assert service is None
+
+
+def test_find_by_service_name_returns_right_service(notifications_admin,
+                                                    notifications_admin_db,
+                                                    notify_db_session):
+    user = create_test_user()
+    id = services_dao.insert_new_service('testing service', user)
+    another = services_dao.insert_new_service('Testing the Service', user)
+    found = services_dao.find_service_by_service_name('testing service')
+    assert found.id == id
+    assert found.name == 'testing service'
+    found_another = services_dao.find_service_by_service_name('Testing the Service')
+    assert found_another == services_dao.get_service_by_id(another)
+
+
+def test_should_not_allow_two_services_of_the_same_name(notifications_admin, notifications_admin_db, notify_db_session):
+    user = create_test_user()
+    services_dao.insert_new_service('duplicate service', user)
+    with pytest.raises(sqlalchemy.exc.IntegrityError) as error:
+        services_dao.insert_new_service('duplicate service', user)
+        assert 'duplicate key value violates unique constraint "services_name_key' in error.value

tests/app/main/views/test_add_service.py

@@ -0,0 +1,26 @@
+from app.main.dao import verify_codes_dao
+from tests.app.main import create_test_user
+
+
+def test_get_should_render_add_service_template(notifications_admin, notifications_admin_db, notify_db_session):
+    with notifications_admin.test_client() as client:
+        with client.session_transaction() as session:
+            user = create_test_user()
+            session['user_id'] = user.id
+        verify_codes_dao.add_code(user_id=user.id, code='12345', code_type='sms')
+        client.post('/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(notifications_admin, notifications_admin_db, notify_db_session):
+    with notifications_admin.test_client() as client:
+        with client.session_transaction() as session:
+            user = create_test_user()
+            session['user_id'] = user.id
+        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'})
+        assert response.status_code == 302
+        assert response.location == 'http://localhost/dashboard'