More tests for rest and dao.

2026-09-11 10:28:55 -04:00 · 2016-01-13 12:14:21 +00:00
parent dad0fff4ba
commit 99820b3e85
5 changed files with 187 additions and 6 deletions
@@ -0,0 +1,52 @@
+import pytest
+from app.dao.templates_dao import (
+    save_model_template, get_model_templates, delete_model_template)
+from tests.app.conftest import sample_template as create_sample_template
+from app.models import Template
+
+
+def test_create_template(notify_api, notify_db, notify_db_session, sample_service):
+    assert Template.query.count() == 0
+    template_name = 'Sample Template'
+    data = {
+        'name': template_name,
+        'template_type': "sms",
+        'content': "Template content",
+        'service': sample_service}
+    template = Template(**data)
+    save_model_template(template)
+    assert Template.query.count() == 1
+    assert Template.query.first().name == template_name
+    assert Template.query.first().id == template.id
+
+
+def test_get_Templates(notify_api, notify_db, notify_db_session, sample_service):
+    sample_template = create_sample_template(notify_db,
+                                             notify_db_session,
+                                             service=sample_service)
+    assert Template.query.count() == 1
+    assert len(get_model_templates()) == 1
+    template_name = "Another Template"
+    sample_template = create_sample_template(notify_db,
+                                             notify_db_session,
+                                             template_name=template_name,
+                                             service=sample_service)
+    assert Template.query.count() == 2
+    assert len(get_model_templates()) == 2
+
+
+def test_get_user_Template(notify_api, notify_db, notify_db_session, sample_service):
+    assert Template.query.count() == 0
+    template_name = "Random Template"
+    sample_template = create_sample_template(notify_db,
+                                             notify_db_session,
+                                             template_name=template_name,
+                                             service=sample_service)
+    assert get_model_templates(template_id=sample_template.id).name == template_name
+    assert Template.query.count() == 1
+
+
+def test_delete_template(notify_api, notify_db, notify_db_session, sample_template):
+    assert Template.query.count() == 1
+    delete_model_template(sample_template)
+    assert Template.query.count() == 0