kallithea Changeset - f65661179895

Changeset - f65661179895

Parent rev.

Child rev.

[Not reviewed]

default

0 2 1

Mads Kiilerich - 9 years ago 2016-06-29 16:52:07
madski@unity3d.com

tests: introduce tests and reference dump for notification mails

The mails are dumped to a tracked html file:
* changes shows up as diffs and are easy to spot and review
* all mails can easily can be investigated in a browser and checked for content
and consistency

The tests are mocking canonical_url because it has deep dependencies to
pylons.url which requires (thread local?) environment setup that the tests
doesn't have.

3 files changed with 718 insertions and 7 deletions:

kallithea/model/db.py

kallithea/tests/models/test_dump_html_mails.ref.html

629

kallithea/tests/models/test_notifications.py

0 comments (0 inline, 0 general)

kallithea/model/db.py

➞

Show inline comments

@@ @@ -2353,97 +2353,97 @@ class PullRequest(Base, BaseModel): @@
     def __json__(self):
         return dict(
             revisions=self.revisions
+        )
     def url(self, **kwargs):
         canonical = kwargs.pop('canonical', None)
         import kallithea.lib.helpers as h
         b = self.org_ref_parts[1]
         if b != self.other_ref_parts[1]:
             s = '/_/' + b
         else:
             s = '/_/' + self.title
         kwargs['extra'] = urlreadable(s)
         if canonical:
             return h.canonical_url('pullrequest_show', repo_name=self.other_repo.repo_name,
                                    pull_request_id=self.pull_request_id, **kwargs)
         return h.url('pullrequest_show', repo_name=self.other_repo.repo_name,
                      pull_request_id=self.pull_request_id, **kwargs)
 class PullRequestReviewers(Base, BaseModel):
     __tablename__ = 'pull_request_reviewers'
     __table_args__ = (
         Index('pull_request_reviewers_user_id_idx', 'user_id'),
         _table_args_default_dict,
+    )
     def __init__(self, user=None, pull_request=None):
         self.user = user
         self.pull_request = pull_request
     pull_requests_reviewers_id = Column(Integer(), unique=True, primary_key=True)
     pull_request_id = Column(Integer(), ForeignKey('pull_requests.pull_request_id'), nullable=False)
     user_id = Column(Integer(), ForeignKey('users.user_id'), nullable=False)
     user = relationship('User')
     pull_request = relationship('PullRequest')
 class Notification(Base, BaseModel):
     __tablename__ = 'notifications'
     __table_args__ = (
         Index('notification_type_idx', 'type'),
         _table_args_default_dict,
+    )
     TYPE_CHANGESET_COMMENT = u'cs_comment'
     TYPE_MESSAGE = u'message'
     TYPE_MENTION = u'mention'
+    TYPE_MENTION = u'mention' # not used
     TYPE_REGISTRATION = u'registration'
     TYPE_PULL_REQUEST = u'pull_request'
     TYPE_PULL_REQUEST_COMMENT = u'pull_request_comment'
     notification_id = Column(Integer(), unique=True, primary_key=True)
     subject = Column(Unicode(512), nullable=False)
     body = Column(UnicodeText(50000), nullable=False)
     created_by = Column(Integer(), ForeignKey('users.user_id'), nullable=False)
     created_on = Column(DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
     type_ = Column('type', Unicode(255), nullable=False)
     created_by_user = relationship('User')
     notifications_to_users = relationship('UserNotification', cascade="all, delete-orphan")
     @property
     def recipients(self):
         return [x.user for x in UserNotification.query()
                 .filter(UserNotification.notification == self)
                 .order_by(UserNotification.user_id.asc()).all()]
     @classmethod
     def create(cls, created_by, subject, body, recipients, type_=None):
         if type_ is None:
             type_ = Notification.TYPE_MESSAGE
         notification = cls()
         notification.created_by_user = created_by
         notification.subject = subject
         notification.body = body
         notification.type_ = type_
         notification.created_on = datetime.datetime.now()
         for recipient in recipients:
             un = UserNotification()
             un.notification = notification
             un.user_id = recipient.user_id
             # Mark notifications to self "pre-read" - should perhaps just be skipped
             if recipient == created_by:
                 un.read = True
             Session().add(un)
         Session().add(notification)
         Session().flush() # assign notification.notification_id
         return notification
     @property
     def description(self):
         from kallithea.model.notification import NotificationModel

kallithea/tests/models/test_dump_html_mails.ref.html

➞

Show inline comments

@@ new file 100644 @@
 <html><body>
 <h1>cs_comment, is_mention=False, status_change=None</h1>
 <pre>
 From: u1
 To: u2@example.com
 Subject: [Comment] repo/name changeset cafe on brunch
 --------------------
 Comment from Commenter Name on repo_target changeset c0ffeecafe:
 This is the new comment.
  - and here it ends indented.
 URL: http://comment.org
 Changeset: c0ffeecafe
 Description:
 This changeset did something clever which is hard to explain
 --
 This is an automatic notification. Don't reply to this mail.
 --------------------</pre>
 <p>Comment from Commenter Name on repo_target changeset c0ffeecafe:</p>
 <p><div class="formatted-fixed">This is the new comment.
  - and here it ends indented.</div></p>
 <p>URL: <a href="http://comment.org">http://comment.org</a></p>
 <p>Changeset: c0ffeecafe</p>
 <p>Description:<br/>
 This changeset did something clever which is hard to explain
 </p>
 <br/>
 <br/>
 -- <br/>
 This is an automatic notification. Don&#39;t reply to this mail.
 <pre>--------------------</pre>
 <h1>cs_comment, is_mention=True, status_change=None</h1>
 <pre>
 From: u1
 To: u2@example.com
 Subject: [Comment] repo/name changeset cafe on brunch
 --------------------
 Comment from Commenter Name on repo_target changeset c0ffeecafe mentioned you:
 This is the new comment.
  - and here it ends indented.
 URL: http://comment.org
 Changeset: c0ffeecafe
 Description:
 This changeset did something clever which is hard to explain
 --
 This is an automatic notification. Don't reply to this mail.
 --------------------</pre>
 <p>Comment from Commenter Name on repo_target changeset c0ffeecafe mentioned you:</p>
 <p><div class="formatted-fixed">This is the new comment.
  - and here it ends indented.</div></p>
 <p>URL: <a href="http://comment.org">http://comment.org</a></p>
 <p>Changeset: c0ffeecafe</p>
 <p>Description:<br/>
 This changeset did something clever which is hard to explain
 </p>
 <br/>
 <br/>
 -- <br/>
 This is an automatic notification. Don&#39;t reply to this mail.
 <pre>--------------------</pre>
 <h1>cs_comment, is_mention=False, status_change='Approved'</h1>
 <pre>
 From: u1
 To: u2@example.com
 Subject: [Approved: Comment] repo/name changeset cafe on brunch
 --------------------
 Comment from Commenter Name on repo_target changeset c0ffeecafe:
 This is the new comment.
  - and here it ends indented.
 The changeset status was changed to: Approved
 URL: http://comment.org
 Changeset: c0ffeecafe
 Description:
 This changeset did something clever which is hard to explain
 --
 This is an automatic notification. Don't reply to this mail.
 --------------------</pre>
 <p>Comment from Commenter Name on repo_target changeset c0ffeecafe:</p>
 <p><div class="formatted-fixed">This is the new comment.
  - and here it ends indented.</div></p>
     <p>The changeset status was changed to: <b>Approved</b></p>
 <p>URL: <a href="http://comment.org">http://comment.org</a></p>
 <p>Changeset: c0ffeecafe</p>
 <p>Description:<br/>
 This changeset did something clever which is hard to explain
 </p>
 <br/>
 <br/>
 -- <br/>
 This is an automatic notification. Don&#39;t reply to this mail.
 <pre>--------------------</pre>
 <h1>cs_comment, is_mention=True, status_change='Approved'</h1>
 <pre>
 From: u1
 To: u2@example.com
 Subject: [Approved: Comment] repo/name changeset cafe on brunch
 --------------------
 Comment from Commenter Name on repo_target changeset c0ffeecafe mentioned you:
 This is the new comment.
  - and here it ends indented.
 The changeset status was changed to: Approved
 URL: http://comment.org
 Changeset: c0ffeecafe
 Description:
 This changeset did something clever which is hard to explain
 --
 This is an automatic notification. Don't reply to this mail.
 --------------------</pre>
 <p>Comment from Commenter Name on repo_target changeset c0ffeecafe mentioned you:</p>
 <p><div class="formatted-fixed">This is the new comment.
  - and here it ends indented.</div></p>
     <p>The changeset status was changed to: <b>Approved</b></p>
 <p>URL: <a href="http://comment.org">http://comment.org</a></p>
 <p>Changeset: c0ffeecafe</p>
 <p>Description:<br/>
 This changeset did something clever which is hard to explain
 </p>
 <br/>
 <br/>
 -- <br/>
 This is an automatic notification. Don&#39;t reply to this mail.
 <pre>--------------------</pre>
 <h1>message</h1>
 <pre>
 From: u1
 To: u2@example.com
 Subject: Test Message
 --------------------
 This is the body of the test message
  - nothing interesting here except indentation.
 --
 This is an automatic notification. Don't reply to this mail.
 --------------------</pre>
 <div class="formatted-fixed">This is the body of the test message
  - nothing interesting here except indentation.</div>
 <br/>
 <br/>
 -- <br/>
 This is an automatic notification. Don&#39;t reply to this mail.
 <pre>--------------------</pre>
 <h1>registration</h1>
 <pre>
 From: u1
 To: u2@example.com
 Subject: New user newbie registered
 --------------------
 Registration body
 View this user here: http://newbie.org
 --
 This is an automatic notification. Don't reply to this mail.
 --------------------</pre>
 <div class="formatted-fixed">Registration body</div>
 View this user here: <a href="http://newbie.org">http://newbie.org</a>
 <br/>
 <br/>
 -- <br/>
 This is an automatic notification. Don&#39;t reply to this mail.
 <pre>--------------------</pre>
 <h1>pull_request, is_mention=False</h1>
 <pre>
 From: u1
 To: u2@example.com
 Subject: [Added] repo/name pull request 7 from ref
 --------------------
 Requester Name requested your review of repo/name pull request "The Title"
 URL: http://pr.org
 Description:
 This PR is awesome because it does stuff
  - please approve indented!
 Changesets:
 abc: http://changeset_home/?repo_name=repo_org&amp;revision=123abc
 Introduce one and two
 fed: http://changeset_home/?repo_name=repo_org&amp;revision=567fed
 Make one plus two equal tree
 --
 This is an automatic notification. Don't reply to this mail.
 --------------------</pre>
 <p>Requester Name requested your review of repo/name pull request &#34;The Title&#34;</p>
 <p>URL: <a href="http://pr.org">http://pr.org</a></p>
 <p>Description:</p>
 <p style="white-space: pre-wrap; font-family: monospace"><div class="formatted-fixed">This PR is awesome because it does stuff
  - please approve indented!</div></p>
 <p>Changesets:</p>
 <p style="white-space: pre-wrap">
 <i><a href="http://changeset_home/?repo_name=repo_org&amp;revision=123abc">123abc</a></i>:
 Introduce one and two
 <i><a href="http://changeset_home/?repo_name=repo_org&amp;revision=567fed">567fed</a></i>:
 Make one plus two equal tree
 </p>
 <br/>
 <br/>
 -- <br/>
 This is an automatic notification. Don&#39;t reply to this mail.
 <pre>--------------------</pre>
 <h1>pull_request, is_mention=True</h1>
 <pre>
 From: u1
 To: u2@example.com
 Subject: [Added] repo/name pull request 7 from ref
 --------------------
 Requester Name mentioned you on repo/name pull request "The Title"
 URL: http://pr.org
 Description:
 This PR is awesome because it does stuff
  - please approve indented!
 Changesets:
 abc: http://changeset_home/?repo_name=repo_org&amp;revision=123abc
 Introduce one and two
 fed: http://changeset_home/?repo_name=repo_org&amp;revision=567fed
 Make one plus two equal tree
 --
 This is an automatic notification. Don't reply to this mail.
 --------------------</pre>
 <p>Requester Name mentioned you on repo/name pull request &#34;The Title&#34;</p>
 <p>URL: <a href="http://pr.org">http://pr.org</a></p>
 <p>Description:</p>
 <p style="white-space: pre-wrap; font-family: monospace"><div class="formatted-fixed">This PR is awesome because it does stuff
  - please approve indented!</div></p>
 <p>Changesets:</p>
 <p style="white-space: pre-wrap">
 <i><a href="http://changeset_home/?repo_name=repo_org&amp;revision=123abc">123abc</a></i>:
 Introduce one and two
 <i><a href="http://changeset_home/?repo_name=repo_org&amp;revision=567fed">567fed</a></i>:
 Make one plus two equal tree
 </p>
 <br/>
 <br/>
 -- <br/>
 This is an automatic notification. Don&#39;t reply to this mail.
 <pre>--------------------</pre>
 <h1>pull_request_comment, status_change=None, closing_pr=False</h1>
 <pre>
 From: u1
 To: u2@example.com
 Subject: [Comment] repo/name pull request 7 from ref
 --------------------
 Comment from Commenter Name on repo/name pull request "The Title":
 Me too!
  - and indented on second line
 URL: http://pr.org/comment
 --
 This is an automatic notification. Don't reply to this mail.
 --------------------</pre>
 <p>Comment from Commenter Name on repo/name pull request &#34;The Title&#34;:</p>
 <p><div class="formatted-fixed">Me too!
  - and indented on second line</div></p>
 <p>URL: <a href="http://pr.org/comment">http://pr.org/comment</a></p>
 <br/>
 <br/>
 -- <br/>
 This is an automatic notification. Don&#39;t reply to this mail.
 <pre>--------------------</pre>
 <h1>pull_request_comment, status_change='Under Review', closing_pr=False</h1>
 <pre>
 From: u1
 To: u2@example.com
 Subject: [Under Review: Comment] repo/name pull request 7 from ref
 --------------------
 Comment from Commenter Name on repo/name pull request "The Title":
 Me too!
  - and indented on second line
 The comment was made with status: Under Review
 URL: http://pr.org/comment
 --
 This is an automatic notification. Don't reply to this mail.
 --------------------</pre>
 <p>Comment from Commenter Name on repo/name pull request &#34;The Title&#34;:</p>
 <p><div class="formatted-fixed">Me too!
  - and indented on second line</div></p>
        <p>The comment was made with status: <b>Under Review</b></p>
 <p>URL: <a href="http://pr.org/comment">http://pr.org/comment</a></p>
 <br/>
 <br/>
 -- <br/>
 This is an automatic notification. Don&#39;t reply to this mail.
 <pre>--------------------</pre>
 <h1>pull_request_comment, status_change=None, closing_pr=True</h1>
 <pre>
 From: u1
 To: u2@example.com
 Subject: [Closing: Comment] repo/name pull request 7 from ref
 --------------------
 Comment from Commenter Name on repo/name pull request "The Title":
 Me too!
  - and indented on second line
 URL: http://pr.org/comment
 --
 This is an automatic notification. Don't reply to this mail.
 --------------------</pre>
 <p>Comment from Commenter Name on repo/name pull request &#34;The Title&#34;:</p>
 <p><div class="formatted-fixed">Me too!
  - and indented on second line</div></p>
 <p>URL: <a href="http://pr.org/comment">http://pr.org/comment</a></p>
 <br/>
 <br/>
 -- <br/>
 This is an automatic notification. Don&#39;t reply to this mail.
 <pre>--------------------</pre>
 <h1>pull_request_comment, status_change='Under Review', closing_pr=True</h1>
 <pre>
 From: u1
 To: u2@example.com
 Subject: [Under Review, Closing: Comment] repo/name pull request 7 from ref
 --------------------
 Comment from Commenter Name on repo/name pull request "The Title":
 Me too!
  - and indented on second line
 The comment closed the pull request with status: Under Review
 URL: http://pr.org/comment
 --
 This is an automatic notification. Don't reply to this mail.
 --------------------</pre>
 <p>Comment from Commenter Name on repo/name pull request &#34;The Title&#34;:</p>
 <p><div class="formatted-fixed">Me too!
  - and indented on second line</div></p>
        <p>The comment closed the pull request with status: <b>Under Review</b></p>
 <p>URL: <a href="http://pr.org/comment">http://pr.org/comment</a></p>
 <br/>
 <br/>
 -- <br/>
 This is an automatic notification. Don&#39;t reply to this mail.
 <pre>--------------------</pre>
 <h1>TYPE_PASSWORD_RESET</h1>
 <pre>
 From: u1
 To: john@doe.com
 Subject: Password reset link
 --------------------
 Hello John Doe
 We have received a request to reset the password for your account.
 To set a new password, click the following link:
 http://reset.com/decbf64715098db5b0bd23eab44bd792670ab746
 Should you not be able to use the link above, please type the following code into the password reset form: decbf64715098db5b0bd23eab44bd792670ab746
 If it weren't you who requested the password reset, just disregard this message.
 --
 This is an automatic notification. Don't reply to this mail.
 --------------------</pre>
 <h4>Hello John Doe</h4>
 <p>We have received a request to reset the password for your account.</p>
 <p>To set a new password, click the following link:</p>
 <p><a href="http://reset.com/decbf64715098db5b0bd23eab44bd792670ab746">http://reset.com/decbf64715098db5b0bd23eab44bd792670ab746</a></p>
 <p>Should you not be able to use the link above, please type the following code into the password reset form: <code>decbf64715098db5b0bd23eab44bd792670ab746</code></p>
 <p>If it weren&#39;t you who requested the password reset, just disregard this message.</p>
 <br/>
 <br/>
 -- <br/>
 This is an automatic notification. Don&#39;t reply to this mail.
 <pre>--------------------</pre>
 </body></html>

kallithea/tests/models/test_notifications.py

➞

Show inline comments

 import os
 import mock
 import routes.util
 from kallithea.tests import *
+from kallithea.lib import helpers as h
 from kallithea.model.db import User, Notification, UserNotification
 from kallithea.model.user import UserModel
 from kallithea.model.meta import Session
 from kallithea.model.notification import NotificationModel, EmailNotificationModel
 from kallithea.model.meta import Session
 from kallithea.model.notification import NotificationModel
 import kallithea.lib.celerylib
 import kallithea.lib.celerylib.tasks
 class TestNotifications(TestController):
     def setup_method(self, method):
         Session.remove()
         u1 = UserModel().create_or_update(username=u'u1',
                                         password=u'qweqwe',
                                         email=u'u1@example.com',
                                         firstname=u'u1', lastname=u'u1')
         Session().commit()
         self.u1 = u1.user_id
         u2 = UserModel().create_or_update(username=u'u2',
                                         password=u'qweqwe',
                                         email=u'u2@example.com',
                                         firstname=u'u2', lastname=u'u3')
         Session().commit()
         self.u2 = u2.user_id
         u3 = UserModel().create_or_update(username=u'u3',
                                         password=u'qweqwe',
                                         email=u'u3@example.com',
                                         firstname=u'u3', lastname=u'u3')
         Session().commit()
         self.u3 = u3.user_id
         self.remove_all_notifications()
         assert [] == Notification.query().all()
         assert [] == UserNotification.query().all()
     def test_create_notification(self):
         usrs = [self.u1, self.u2]
         notification = NotificationModel().create(created_by=self.u1,
                                            subject=u'subj', body=u'hi there',
                                            recipients=usrs)
         def send_email(recipients, subject, body='', html_body='', headers=None, author=None):
             assert recipients == ['u2@example.com']
             assert subject == 'Test Message'
             assert body == "\n\nhi there\n\n\n-- \nThis is an automatic notification. Don't reply to this mail.\n"
             assert '>hi there<' in html_body
             assert author.username == 'u1'
         with mock.patch.object(kallithea.lib.celerylib.tasks, 'send_email', send_email):
             notification = NotificationModel().create(created_by=self.u1,
                                                subject=u'subj', body=u'hi there',
                                                recipients=usrs)
         Session().commit()
         u1 = User.get(self.u1)
         u2 = User.get(self.u2)
         u3 = User.get(self.u3)
         notifications = Notification.query().all()
         assert len(notifications) == 1
         assert notifications[0].recipients == [u1, u2]
         assert notification.notification_id == notifications[0].notification_id
         unotification = UserNotification.query() \
             .filter(UserNotification.notification == notification).all()
         assert len(unotification) == len(usrs)
         assert set([x.user.user_id for x in unotification]) == set(usrs)
     def test_user_notifications(self):
         notification1 = NotificationModel().create(created_by=self.u1,
                                             subject=u'subj', body=u'hi there1',
                                             recipients=[self.u3])
         Session().commit()
         notification2 = NotificationModel().create(created_by=self.u1,
                                             subject=u'subj', body=u'hi there2',
                                             recipients=[self.u3])
         Session().commit()
         u3 = Session().query(User).get(self.u3)
         assert sorted([x.notification for x in u3.notifications]) == sorted([notification2, notification1])
     def test_delete_notifications(self):
         notification = NotificationModel().create(created_by=self.u1,
                                            subject=u'title', body=u'hi there3',
                                     recipients=[self.u3, self.u1, self.u2])
         Session().commit()
         notifications = Notification.query().all()
         assert notification in notifications
         Notification.delete(notification.notification_id)
         Session().commit()
         notifications = Notification.query().all()
         assert not notification in notifications
         un = UserNotification.query().filter(UserNotification.notification
                                              == notification).all()
         assert un == []
     def test_delete_association(self):
@@ @@ -104,48 +118,116 @@ class TestNotifications(TestController): @@
         NotificationModel().delete(self.u3,
                                    notification.notification_id)
         Session().commit()
         u3notification = UserNotification.query() \
                             .filter(UserNotification.notification ==
                                     notification) \
                             .filter(UserNotification.user_id == self.u3) \
                             .scalar()
         assert u3notification == None
         # notification object is still there
         assert Notification.query().all() == [notification]
         #u1 and u2 still have assignments
         u1notification = UserNotification.query() \
                             .filter(UserNotification.notification ==
                                     notification) \
                             .filter(UserNotification.user_id == self.u1) \
                             .scalar()
         assert u1notification != None
         u2notification = UserNotification.query() \
                             .filter(UserNotification.notification ==
                                     notification) \
                             .filter(UserNotification.user_id == self.u2) \
                             .scalar()
         assert u2notification != None
     def test_notification_counter(self):
         NotificationModel().create(created_by=self.u1,
                             subject=u'title', body=u'hi there_delete',
                             recipients=[self.u3, self.u1])
         Session().commit()
         assert NotificationModel().get_unread_cnt_for_user(self.u1) == 0
         assert NotificationModel().get_unread_cnt_for_user(self.u2) == 0
         assert NotificationModel().get_unread_cnt_for_user(self.u3) == 1
         notification = NotificationModel().create(created_by=self.u1,
                                            subject=u'title', body=u'hi there3',
                                     recipients=[self.u3, self.u1, self.u2])
         Session().commit()
         assert NotificationModel().get_unread_cnt_for_user(self.u1) == 0
         assert NotificationModel().get_unread_cnt_for_user(self.u2) == 1
         assert NotificationModel().get_unread_cnt_for_user(self.u3) == 2
     @mock.patch.object(h, 'canonical_url', (lambda arg, **kwargs: 'http://%s/?%s' % (arg, '&'.join('%s=%s' % (k, v) for (k, v) in sorted(kwargs.items())))))
     def test_dump_html_mails(self):
         # Exercise all notification types and dump them to one big html file
         l = []
         def send_email(recipients, subject, body='', html_body='', headers=None, author=None):
             l.append('\n\n<h1>%s</h1>\n' % desc) # desc is from outer scope
             l.append('<pre>\n\n')
             l.append('From: %s\n' % author.username)
             l.append('To: %s\n' % ' '.join(recipients))
             l.append('Subject: %s\n' % subject)
             l.append('\n--------------------\n%s\n--------------------' % body)
             l.append('</pre>\n')
             l.append('\n%s\n' % html_body)
             l.append('<pre>--------------------</pre>\n')
         l.append('<html><body>\n')
         with mock.patch.object(kallithea.lib.celerylib.tasks, 'send_email', send_email):
             pr_kwargs = dict(pr_nice_id='7', ref='ref', org_repo_name='repo_org', pr_title='The Title', pr_url='http://pr.org')
             for type_, body, kwargs in [
                 (Notification.TYPE_CHANGESET_COMMENT, 'This is the new comment.\n\n - and here it ends indented.', dict(short_id='cafe', raw_id='c0ffeecafe', branch='brunch', cs_comment_user='Commenter Name', cs_comment_url='http://comment.org', is_mention=[False, True], message='This changeset did something clever which is hard to explain', status_change=[None, 'Approved'], cs_target_repo='repo_target', cs_url='http://changeset.com')),
                 (Notification.TYPE_MESSAGE, 'This is the body of the test message\n - nothing interesting here except indentation.', dict()),
                 #(Notification.TYPE_MENTION, '$body', None), # not used
                 (Notification.TYPE_REGISTRATION, 'Registration body', dict(new_username='newbie', registered_user_url='http://newbie.org', new_email='new@email.com', new_full_name='New Full Name')),
                 (Notification.TYPE_PULL_REQUEST, 'This PR is awesome because it does stuff\n - please approve indented!', dict(pr_user_created='Requester Name', is_mention=[False, True], pr_revisions=[('123abc', 'Introduce one and two'), ('567fed', 'Make one plus two equal tree')], **pr_kwargs)),
                 (Notification.TYPE_PULL_REQUEST_COMMENT, 'Me too!\n\n - and indented on second line', dict(closing_pr=[False, True], pr_comment_user='Commenter Name', pr_comment_url='http://pr.org/comment', status_change=[None, 'Under Review'], pr_target_repo='http://target.com/repo', **pr_kwargs)),
                 ]:
                 kwargs['repo_name'] = 'repo/name'
                 params = [(type_, type_, body, kwargs)]
                 for param_name in ['is_mention', 'status_change', 'closing_pr']: # TODO: inline/general
                     if not isinstance(kwargs.get(param_name), list):
                         continue
                     new_params = []
                     for v in kwargs[param_name]:
                         for desc, type_, body, kwargs in params:
                             kwargs = dict(kwargs)
                             kwargs[param_name] = v
                             new_params.append(('%s, %s=%r' % (desc, param_name, v), type_, body, kwargs))
                     params = new_params
                 for desc, type_, body, kwargs in params:
                     # desc is used as "global" variable
                     notification = NotificationModel().create(created_by=self.u1,
                                                        subject='unused', body=body, email_kwargs=kwargs,
                                                        recipients=[self.u2], type_=type_)
             # Email type TYPE_PASSWORD_RESET has no corresponding notification type - test it directly:
             desc = 'TYPE_PASSWORD_RESET'
             kwargs = dict(user='John Doe', reset_token='decbf64715098db5b0bd23eab44bd792670ab746', reset_url='http://reset.com/decbf64715098db5b0bd23eab44bd792670ab746')
             kallithea.lib.celerylib.run_task(kallithea.lib.celerylib.tasks.send_email, ['john@doe.com'],
                 "Password reset link",
                 EmailNotificationModel().get_email_tmpl(EmailNotificationModel.TYPE_PASSWORD_RESET, 'txt', **kwargs),
                 EmailNotificationModel().get_email_tmpl(EmailNotificationModel.TYPE_PASSWORD_RESET, 'html', **kwargs),
                 author=User.get(self.u1))
         l.append('\n</body></html>\n')
         out = ''.join(l)
         outfn = os.path.join(os.path.dirname(__file__), 'test_dump_html_mails.out.html')
         reffn = os.path.join(os.path.dirname(__file__), 'test_dump_html_mails.ref.html')
         with file(outfn, 'w') as f:
             f.write(out)
         with file(reffn) as f:
             ref = f.read()
         assert ref == out # copy test_dump_html_mails.out.html to test_dump_html_mails.ref.html to update expectations
         os.unlink(outfn)

0 comments (0 inline, 0 general)