Add tests about sending notifications. #7

1 files changed, 64 insertions, 0 deletions

diff --git a/budget/tests.py b/budget/tests.py
new file mode 100644
index 0000000..6e365bd
--- /dev/null
+++ b/budget/tests.py
@@ -0,0 +1,64 @@
+import os
+import tempfile
+import unittest
+
+from flask import g
+
+import web
+import models
+
+class BudgetTestCase(unittest.TestCase):
+
+    def test_notifications(self):
+        """Test that the notifications are sent, and that email adresses
+        are checked properly.
+        """
+        # create a project
+        self.create_project("raclette")
+
+        with web.app.test_client() as c:
+            self.login("raclette", test_client=c)
+            result = c.post("/raclette/invite", 
+                    data={"emails": 'test@test.com'})
+            # check here that the mails are sent.
+
+
+    def login(self, project, password=None, test_client=None):
+        password = password or project
+        test_client = test_client or self.app
+
+        return test_client.post('/authenticate', data=dict(
+            id=project, password=password), follow_redirects=True)
+
+    def create_project(self, name):
+        """Create a fake project"""
+        # create the project
+        project = models.Project(id=name, name=unicode(name), password=name, 
+                contact_email="%s@notmyidea.org" % name)
+        models.db.session.add(project)
+        models.db.session.commit()
+
+        return project
+
+    def setUp(self):
+        web.app.config['TESTING'] = True
+
+        self.fd, self.fp = tempfile.mkstemp()
+        web.app.config['SQLALCHEMY_DATABASE_URI'] = "sqlite:///%s" % self.fp
+        web.app.config['CSRF_ENABLED'] = False # simplify the tests
+        self.app = web.app.test_client()
+
+        models.db.init_app(web.app)
+        web.mail.init_app(web.app)
+
+        models.db.app = web.app
+        models.db.create_all()
+
+    def tearDown(self):
+        # clean after testing
+        os.close(self.fd)
+        os.unlink(self.fp)
+
+
+if __name__ == "__main__":
+    unittest.main()