tests: contact form

* Added testing suite for contact form.

Signed-off-by: Aman Jain <jainaman224@gmail.com>

tests/unit/pages/test_pages_views.py

@@ -0,0 +1,161 @@
+# -*- coding: utf-8 -*-
+#
+# This file is part of Zenodo.
+# Copyright (C) 2017 CERN.
+#
+# Zenodo is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Zenodo is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Zenodo. If not, see <http://www.gnu.org/licenses/>.
+#
+# In applying this licence, CERN does not waive the privileges and immunities
+# granted to it by virtue of its status as an Intergovernmental Organization
+# or submit itself to any jurisdiction.
+
+"""Unit tests Pages Utils."""
+
+from __future__ import absolute_import, print_function
+
+from flask import url_for
+from helpers import login_user_via_session
+from six import BytesIO
+from werkzeug import MultiDict
+
+
+def test_send_support_email(app, db, es, users):
+    """Test mail sending."""
+    with app.extensions['mail'].record_messages() as outbox:
+        with app.test_client() as client:
+            res = client.get(url_for('zenodo_pages.support'))
+            assert res.status_code == 200
+
+            res = client.get(
+                url_for('zenodo_pages.support')
+            )
+            assert res.status_code == 200
+
+            res = client.post(
+                url_for('zenodo_pages.support'),
+                data=dict()
+            )
+            assert res.status_code == 200
+            assert b'field-name has-error' in res.data
+            assert b'field-email has-error' in res.data
+            assert b'field-subject has-error' in res.data
+            assert b'field-description has-error' in res.data
+            assert b'field-attachments has-error' not in res.data
+
+            form = MultiDict(dict(
+                name='Aman',
+                email='abcxyz@gmail.com',
+                subject='hello',
+                issue_category='tech-support',
+                description='Please help us! Troubleshoot our problem.'
+            ))
+
+            res = client.post(
+                url_for('zenodo_pages.support'),
+                data=form
+            )
+            assert b'has-error' not in res.data
+            assert len(outbox) == 1
+            sent_msg = outbox[0]
+            assert sent_msg.subject == '[tech-support]: hello'
+            assert sent_msg.reply_to == 'abcxyz@gmail.com'
+            assert 'Aman <abcxyz@gmail.com>' in sent_msg.body
+
+            form = MultiDict(dict(
+                name='Foo',
+                email='example@mail.com',
+                subject='Bar',
+                issue_category='tech-support',
+                description='Please help us! Troubleshoot our problem.'
+            ))
+            test_file = BytesIO(b'My other file contents')
+            test_file2 = BytesIO(b'Another My other file contents')
+            form.add('attachments', (test_file, 'file2.txt'))
+            form.add('attachments', (test_file2, 'test3.txt'))
+            res = client.post(
+                url_for('zenodo_pages.support'),
+                data=form,
+                content_type='multipart/form-data',
+                follow_redirects=True
+            )
+            assert len(outbox) == 2
+            sent_msg = outbox[1]
+            file1 = sent_msg.attachments[0]
+            assert file1.filename == 'file2.txt'
+            assert file1.data == b'My other file contents'
+            file2 = sent_msg.attachments[1]
+            assert file2.filename == 'test3.txt'
+            assert file2.data == b'Another My other file contents'
+
+            login_user_via_session(client, email=users[1]['email'])
+            res = client.get(
+                url_for('zenodo_pages.support')
+            )
+            assert b'Zenodo staff will get back to you by your Zenodo email:' \
+                in res.data
+            assert b'field-email' not in res.data
+
+            form = MultiDict(dict(
+                name='Foo',
+                subject='Bar',
+                issue_category='tech-support',
+                description='Please help us! Troubleshoot our problem.'
+            ))
+            res = client.post(
+                url_for('zenodo_pages.support'),
+                data=form
+            )
+            assert len(outbox) == 3
+            sent_msg = outbox[2]
+            assert 'USER ID:' in sent_msg.body
+            assert users[1]['email'] in sent_msg.body
+
+            test_file = BytesIO(b'My file contents')
+            form.add('attachments', (test_file, 'file1.txt'))
+            res = client.post(
+                url_for('zenodo_pages.support'),
+                data=form,
+                content_type='multipart/form-data',
+                follow_redirects=True
+            )
+            assert len(outbox) == 4
+            sent_msg = outbox[3]
+            file1 = sent_msg.attachments[0]
+            assert file1.filename == 'file1.txt'
+            assert file1.data == b'My file contents'
+
+            form = MultiDict(dict(
+                name='Foo',
+                subject='Bar',
+                issue_category='tech-support',
+                description='Please help us! Troubleshoot our problem.'
+            ))
+            test_file = BytesIO(b'My other file contents')
+            test_file2 = BytesIO(b'Another My other file contents')
+            form.add('attachments', (test_file, 'file2.txt'))
+            form.add('attachments', (test_file2, 'test3.txt'))
+            res = client.post(
+                url_for('zenodo_pages.support'),
+                data=form,
+                content_type='multipart/form-data',
+                follow_redirects=True
+            )
+            assert len(outbox) == 5
+            sent_msg = outbox[4]
+            file1 = sent_msg.attachments[0]
+            assert file1.filename == 'file2.txt'
+            assert file1.data == b'My other file contents'
+            file2 = sent_msg.attachments[1]
+            assert file2.filename == 'test3.txt'
+            assert file2.data == b'Another My other file contents'