test: Unit test for logs helper file (fossasia#5603)

mrsaicharan1 · abhinavk96 · commit c345817d06d5 · 2019-02-18T11:24:23.000+08:00
diff --git a/tests/all/integration/api/helpers/test_log.py b/tests/all/integration/api/helpers/test_log.py
@@ -0,0 +1,38 @@
+from tests.all.integration.setup_database import Setup
+from app import current_app as app
+from tests.all.integration.utils import OpenEventTestCase
+from app.api.helpers.log import record_activity
+from app.models.activity import Activity
+from tests.all.integration.auth_helper import create_user
+from app.models import db
+
+import unittest
+
+
+class TestLogging(OpenEventTestCase):
+    def setUp(self):
+        self.app = Setup.create_app()
+
+    def test_record_activity_valid_template(self):
+        """Test to record activity for valid template"""
+        with app.test_request_context():
+            test_user = create_user(email="logging@test.com", password="logpass")
+            record_activity('create_user', login_user=test_user, user=test_user)
+            user_id_format = ' (' + str(test_user.id) + ')'
+            test_actor = test_user.email + user_id_format
+            self.assertTrue('User logging@test.com' + user_id_format + ' created',
+                            db.session.query(Activity).filter_by(actor=test_actor).first().action)
+
+    def test_record_activity_invalid_template(self):
+        """Test to record activity for invalid template"""
+        with app.test_request_context():
+            test_user = create_user(email="logging@test.com", password="logpass")
+            record_activity('invalid_template', login_user=test_user, user=test_user)
+            user_id_format = ' (' + str(test_user.id) + ')'
+            test_actor = test_user.email + user_id_format
+            self.assertTrue('[ERROR LOGGING] invalid_template',
+                            db.session.query(Activity).filter_by(actor=test_actor).first().action)
+
+
+if __name__ == '__main__':
+    unittest.main()
