From cdc6f2e1c456c9ba748db773c2f52428736c2255 Mon Sep 17 00:00:00 2001
From: Alexis Metaireau <alexis@notmyidea.org>
Date: Tue, 9 Aug 2011 23:22:57 +0200
Subject: [PATCH] Add tests to make test that invitations are sent.

Fixes #7
---
 budget/tests.py | 89 ++++++++++++++++++++++++++++++++-----------------
 1 file changed, 58 insertions(+), 31 deletions(-)

diff --git a/budget/tests.py b/budget/tests.py
index 6e365bd..b79f16f 100644
--- a/budget/tests.py
+++ b/budget/tests.py
@@ -9,37 +9,6 @@ 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
 
@@ -59,6 +28,64 @@ class BudgetTestCase(unittest.TestCase):
         os.close(self.fd)
         os.unlink(self.fp)
 
+    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 test_notifications(self):
+        """Test that the notifications are sent, and that email adresses
+        are checked properly.
+        """
+        # create a project
+        self.create_project("raclette")
+
+        self.login("raclette")
+
+        # sending a message to one person
+        with web.mail.record_messages() as outbox:
+            self.app.post("/raclette/invite", data=
+                    {"emails": 'alexis@notmyidea.org'})
+
+            self.assertEqual(len(outbox), 1)
+            self.assertEqual(outbox[0].recipients, ["alexis@notmyidea.org"])
+
+        # sending a message to multiple persons
+        with web.mail.record_messages() as outbox:
+            self.app.post("/raclette/invite", data=
+                    {"emails": 'alexis@notmyidea.org, toto@notmyidea.org'})
+
+            # only one message is sent to multiple persons
+            self.assertEqual(len(outbox), 1)
+            self.assertEqual(outbox[0].recipients, 
+                    ["alexis@notmyidea.org", "toto@notmyidea.org"])
+
+        # mail address checking
+        with web.mail.record_messages() as outbox:
+            response = self.app.post("/raclette/invite", data={"emails": "toto"})
+            self.assertEqual(len(outbox), 0) # no message sent
+            self.assertIn("The email toto is not valid", response.data)
+
+        # mixing good and wrong adresses shouldn't send any messages
+        with web.mail.record_messages() as outbox:
+            self.app.post("/raclette/invite", data=
+                    {"emails": 'alexis@notmyidea.org, alexis'}) # not valid
+
+            # only one message is sent to multiple persons
+            self.assertEqual(len(outbox), 0)
+
 
 if __name__ == "__main__":
     unittest.main()