Add actions test (#33)

* Add pytest and fix import

* Add actions test

hackr/object_generator.py

@@ -1,5 +1,5 @@
 import json
-import generator
+from . import generator
 
 
 def generate_json(count, **kwargs):

requirements.txt

@@ -4,4 +4,5 @@ qrcode
 pillow
 xmljson
 sphinx_rtd_theme
-matplotlib 2.0.2
+matplotlib==2.0.2
+pytest

tests/test_actions.py

@@ -0,0 +1,40 @@
+from io import StringIO
+from unittest import mock
+
+import smtplib
+import pytest
+
+from hackr.actions import email
+
+
+@mock.patch('smtplib.SMTP')
+def test_email(mock_smtp):
+    message = 'Mail message'
+    params = {
+        'email': '[email protected]',
+        'password': 'password',
+        'to': '[email protected]',
+        'subject': 'Test',
+    }
+    mock_smtp = mock.Mock()
+
+    response = email(message, **params)
+
+    assert response == 'Mail Sent Successfully'
+
+
+@mock.patch('smtplib.SMTP')
+def test_email_exception(mock_smtp):
+    message = 'Mail message'
+    params = {
+        'email': '[email protected]',
+        'password': 'password',
+        'to': '[email protected]',
+        'subject': 'Test',
+    }
+
+    instance = mock_smtp.return_value
+    instance.ehlo.side_effect = smtplib.SMTPRecipientsRefused('Failed to sent mail.')
+    response = email(message, **params)
+
+    assert not response