1 # Patchwork - automated patch tracking system
2 # Copyright (C) 2008 Jeremy Kerr <jk@ozlabs.org>
4 # This file is part of the Patchwork package.
6 # Patchwork is free software; you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 2 of the License, or
9 # (at your option) any later version.
11 # Patchwork is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 # GNU General Public License for more details.
16 # You should have received a copy of the GNU General Public License
17 # along with Patchwork; if not, write to the Free Software
18 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 from email import message_from_string
23 from patchwork.models import Project, Person, Patch, Comment
24 from patchwork.tests.utils import read_patch, create_email, defaults
27 from email.mime.text import MIMEText
29 # Python 2.4 compatibility
30 from email.MIMEText import MIMEText
32 class PatchTest(unittest.TestCase):
33 default_sender = defaults.sender
34 default_subject = defaults.subject
35 project = defaults.project
37 from patchwork.bin.parsemail import find_content, find_author, parse_mail
39 class InlinePatchTest(PatchTest):
40 patch_filename = '0001-add-line.patch'
41 test_comment = 'Test for attached patch'
44 self.orig_patch = read_patch(self.patch_filename)
45 email = create_email(self.test_comment + '\n' + self.orig_patch)
46 (self.patch, self.comment) = find_content(self.project, email)
48 def testPatchPresence(self):
49 self.assertTrue(self.patch is not None)
51 def testPatchContent(self):
52 self.assertEquals(self.patch.content, self.orig_patch)
54 def testCommentPresence(self):
55 self.assertTrue(self.comment is not None)
57 def testCommentContent(self):
58 self.assertEquals(self.comment.content, self.test_comment)
61 class AttachmentPatchTest(InlinePatchTest):
62 patch_filename = '0001-add-line.patch'
63 test_comment = 'Test for attached patch'
64 content_subtype = 'x-patch'
67 self.orig_patch = read_patch(self.patch_filename)
68 email = create_email(self.test_comment, multipart = True)
69 attachment = MIMEText(self.orig_patch, _subtype = self.content_subtype)
70 email.attach(attachment)
71 (self.patch, self.comment) = find_content(self.project, email)
73 class AttachmentXDiffPatchTest(AttachmentPatchTest):
74 content_subtype = 'x-diff'
76 class UTF8InlinePatchTest(InlinePatchTest):
77 patch_filename = '0002-utf-8.patch'
78 patch_encoding = 'utf-8'
81 self.orig_patch = read_patch(self.patch_filename, self.patch_encoding)
82 email = create_email(self.test_comment + '\n' + self.orig_patch,
83 content_encoding = self.patch_encoding)
84 (self.patch, self.comment) = find_content(self.project, email)
86 class NoCharsetInlinePatchTest(InlinePatchTest):
87 """ Test mails with no content-type or content-encoding header """
88 patch_filename = '0001-add-line.patch'
91 self.orig_patch = read_patch(self.patch_filename)
92 email = create_email(self.test_comment + '\n' + self.orig_patch)
93 del email['Content-Type']
94 del email['Content-Transfer-Encoding']
95 (self.patch, self.comment) = find_content(self.project, email)
97 class SignatureCommentTest(InlinePatchTest):
98 patch_filename = '0001-add-line.patch'
99 test_comment = 'Test comment\nmore comment'
102 self.orig_patch = read_patch(self.patch_filename)
103 email = create_email( \
104 self.test_comment + '\n' + \
105 '-- \nsig\n' + self.orig_patch)
106 (self.patch, self.comment) = find_content(self.project, email)
109 class ListFooterTest(InlinePatchTest):
110 patch_filename = '0001-add-line.patch'
111 test_comment = 'Test comment\nmore comment'
114 self.orig_patch = read_patch(self.patch_filename)
115 email = create_email( \
116 self.test_comment + '\n' + \
117 '_______________________________________________\n' + \
118 'Linuxppc-dev mailing list\n' + \
120 (self.patch, self.comment) = find_content(self.project, email)
123 class UpdateCommentTest(InlinePatchTest):
124 """ Test for '---\nUpdate: v2' style comments to patches. """
125 patch_filename = '0001-add-line.patch'
126 test_comment = 'Test comment\nmore comment\n---\nUpdate: test update'
128 class UpdateSigCommentTest(SignatureCommentTest):
129 """ Test for '---\nUpdate: v2' style comments to patches, with a sig """
130 patch_filename = '0001-add-line.patch'
131 test_comment = 'Test comment\nmore comment\n---\nUpdate: test update'
133 class SenderEncodingTest(unittest.TestCase):
134 sender_name = u'example user'
135 sender_email = 'user@example.com'
136 from_header = 'example user <user@example.com>'
139 mail = 'From: %s\n' % self.from_header + \
140 'Subject: test\n\n' + \
142 self.email = message_from_string(mail)
143 (self.person, new) = find_author(self.email)
150 self.assertEquals(self.person.name, self.sender_name)
153 self.assertEquals(self.person.email, self.sender_email)
155 def testDBQueryName(self):
156 db_person = Person.objects.get(name = self.sender_name)
157 self.assertEquals(self.person, db_person)
159 def testDBQueryEmail(self):
160 db_person = Person.objects.get(email = self.sender_email)
161 self.assertEquals(self.person, db_person)
164 class SenderUTF8QPEncodingTest(SenderEncodingTest):
165 sender_name = u'\xe9xample user'
166 from_header = '=?utf-8?q?=C3=A9xample=20user?= <user@example.com>'
168 class SenderUTF8QPSplitEncodingTest(SenderEncodingTest):
169 sender_name = u'\xe9xample user'
170 from_header = '=?utf-8?q?=C3=A9xample?= user <user@example.com>'
172 class SenderUTF8B64EncodingTest(SenderUTF8QPEncodingTest):
173 from_header = '=?utf-8?B?w6l4YW1wbGUgdXNlcg==?= <user@example.com>'
176 class SenderCorrelationTest(unittest.TestCase):
177 existing_sender = 'Existing Sender <existing@example.com>'
178 non_existing_sender = 'Non-existing Sender <nonexisting@example.com>'
180 def mail(self, sender):
181 return message_from_string('From: %s\nSubject: Test\n\ntest\n' % sender)
184 self.existing_sender_mail = self.mail(self.existing_sender)
185 self.non_existing_sender_mail = self.mail(self.non_existing_sender)
186 (self.person, new) = find_author(self.existing_sender_mail)
189 def testExisingSender(self):
190 (person, new) = find_author(self.existing_sender_mail)
191 self.assertEqual(new, False)
192 self.assertEqual(person.id, self.person.id)
194 def testNonExisingSender(self):
195 (person, new) = find_author(self.non_existing_sender_mail)
196 self.assertEqual(new, True)
197 self.assertEqual(person.id, None)
199 def testExistingDifferentFormat(self):
200 mail = self.mail('existing@example.com')
201 (person, new) = find_author(mail)
202 self.assertEqual(new, False)
203 self.assertEqual(person.id, self.person.id)
205 def testExistingDifferentCase(self):
206 mail = self.mail(self.existing_sender.upper())
207 (person, new) = find_author(mail)
208 self.assertEqual(new, False)
209 self.assertEqual(person.id, self.person.id)
214 class MultipleProjectPatchTest(unittest.TestCase):
215 """ Test that patches sent to multiple patchwork projects are
216 handled correctly """
218 test_comment = 'Test Comment'
219 patch_filename = '0001-add-line.patch'
220 msgid = '<1@example.com>'
223 self.p1 = Project(linkname = 'test-project-1', name = 'Project 1',
224 listid = '1.example.com', listemail='1@example.com')
225 self.p2 = Project(linkname = 'test-project-2', name = 'Project 2',
226 listid = '2.example.com', listemail='2@example.com')
231 patch = read_patch(self.patch_filename)
232 email = create_email(self.test_comment + '\n' + patch)
233 email['Message-Id'] = self.msgid
236 email['List-ID'] = '<' + self.p1.listid + '>'
240 email['List-ID'] = '<' + self.p2.listid + '>'
243 def testParsedProjects(self):
244 self.assertEquals(Patch.objects.filter(project = self.p1).count(), 1)
245 self.assertEquals(Patch.objects.filter(project = self.p2).count(), 1)
252 class MultipleProjectPatchCommentTest(MultipleProjectPatchTest):
253 """ Test that followups to multiple-project patches end up on the
256 comment_msgid = '<2@example.com>'
257 comment_content = 'test comment'
260 super(MultipleProjectPatchCommentTest, self).setUp()
262 for project in [self.p1, self.p2]:
263 email = MIMEText(self.comment_content)
264 email['From'] = defaults.sender
265 email['Subject'] = defaults.subject
266 email['Message-Id'] = self.comment_msgid
267 email['List-ID'] = '<' + project.listid + '>'
268 email['In-Reply-To'] = self.msgid
271 def testParsedComment(self):
272 for project in [self.p1, self.p2]:
273 patch = Patch.objects.filter(project = project)[0]
274 # we should see two comments now - the original mail with the patch,
275 # and the one we parsed in setUp()
276 self.assertEquals(Comment.objects.filter(patch = patch).count(), 2)