git.ozlabs.org Git - patchwork/blob - apps/patchwork/tests/patchparser.py

   1 # Patchwork - automated patch tracking system
   2 # Copyright (C) 2008 Jeremy Kerr <jk@ozlabs.org>
   3 #
   4 # This file is part of the Patchwork package.
   5 #
   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.
  10 #
  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.
  15 #
  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
  19
  20 import unittest
  21 import os
  22 from email import message_from_string
  23 from patchwork.models import Project, Person
  24 from patchwork.tests.utils import read_patch, create_email, defaults
  25
  26 try:
  27     from email.mime.text import MIMEText
  28 except ImportError:
  29     # Python 2.4 compatibility
  30     from email.MIMEText import MIMEText
  31
  32 class PatchTest(unittest.TestCase):
  33     default_sender = defaults.sender
  34     default_subject = defaults.subject
  35     project = defaults.project
  36
  37 from patchwork.bin.parsemail import find_content, find_author
  38
  39 class InlinePatchTest(PatchTest):
  40     patch_filename = '0001-add-line.patch'
  41     test_comment = 'Test for attached patch'
  42
  43     def setUp(self):
  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)
  47
  48     def testPatchPresence(self):
  49         self.assertTrue(self.patch is not None)
  50
  51     def testPatchContent(self):
  52         self.assertEquals(self.patch.content, self.orig_patch)
  53
  54     def testCommentPresence(self):
  55         self.assertTrue(self.comment is not None)
  56
  57     def testCommentContent(self):
  58         self.assertEquals(self.comment.content, self.test_comment)
  59
  60
  61 class AttachmentPatchTest(InlinePatchTest):
  62     patch_filename = '0001-add-line.patch'
  63     test_comment = 'Test for attached patch'
  64     content_subtype = 'x-patch'
  65
  66     def setUp(self):
  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)
  72
  73 class AttachmentXDiffPatchTest(AttachmentPatchTest):
  74     content_subtype = 'x-diff'
  75
  76 class UTF8InlinePatchTest(InlinePatchTest):
  77     patch_filename = '0002-utf-8.patch'
  78     patch_encoding = 'utf-8'
  79
  80     def setUp(self):
  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)
  85
  86 class SignatureCommentTest(InlinePatchTest):
  87     patch_filename = '0001-add-line.patch'
  88     test_comment = 'Test comment\nmore comment'
  89
  90     def setUp(self):
  91         self.orig_patch = read_patch(self.patch_filename)
  92         email = create_email( \
  93                 self.test_comment + '\n' + \
  94                 '-- \nsig\n' + self.orig_patch)
  95         (self.patch, self.comment) = find_content(self.project, email)
  96
  97
  98 class ListFooterTest(InlinePatchTest):
  99     patch_filename = '0001-add-line.patch'
 100     test_comment = 'Test comment\nmore comment'
 101
 102     def setUp(self):
 103         self.orig_patch = read_patch(self.patch_filename)
 104         email = create_email( \
 105                 self.test_comment + '\n' + \
 106                 '_______________________________________________\n' + \
 107                 'Linuxppc-dev mailing list\n' + \
 108                 self.orig_patch)
 109         (self.patch, self.comment) = find_content(self.project, email)
 110
 111
 112 class UpdateCommentTest(InlinePatchTest):
 113     """ Test for '---\nUpdate: v2' style comments to patches. """
 114     patch_filename = '0001-add-line.patch'
 115     test_comment = 'Test comment\nmore comment\n---\nUpdate: test update'
 116
 117 class UpdateSigCommentTest(SignatureCommentTest):
 118     """ Test for '---\nUpdate: v2' style comments to patches, with a sig """
 119     patch_filename = '0001-add-line.patch'
 120     test_comment = 'Test comment\nmore comment\n---\nUpdate: test update'
 121
 122 class SenderEncodingTest(unittest.TestCase):
 123     sender_name = u'example user'
 124     sender_email = 'user@example.com'
 125     from_header = 'example user <user@example.com>'
 126
 127     def setUp(self):
 128         mail = 'From: %s\n' % self.from_header + \
 129                'Subject: test\n\n' + \
 130                'test'
 131         self.email = message_from_string(mail)
 132         (self.person, new) = find_author(self.email)
 133         self.person.save()
 134
 135     def tearDown(self):
 136         self.person.delete()
 137
 138     def testName(self):
 139         self.assertEquals(self.person.name, self.sender_name)
 140
 141     def testEmail(self):
 142         self.assertEquals(self.person.email, self.sender_email)
 143
 144     def testDBQueryName(self):
 145         db_person = Person.objects.get(name = self.sender_name)
 146         self.assertEquals(self.person, db_person)
 147
 148     def testDBQueryEmail(self):
 149         db_person = Person.objects.get(email = self.sender_email)
 150         self.assertEquals(self.person, db_person)
 151
 152
 153 class SenderUTF8QPEncodingTest(SenderEncodingTest):
 154     sender_name = u'\xe9xample user'
 155     from_header = '=?utf-8?q?=C3=A9xample=20user?= <user@example.com>'
 156
 157 class SenderUTF8QPSplitEncodingTest(SenderEncodingTest):
 158     sender_name = u'\xe9xample user'
 159     from_header = '=?utf-8?q?=C3=A9xample?= user <user@example.com>'
 160
 161 class SenderUTF8B64EncodingTest(SenderUTF8QPEncodingTest):
 162     from_header = '=?utf-8?B?w6l4YW1wbGUgdXNlcg==?= <user@example.com>'
 163
 164
 165 class SenderCorrelationTest(unittest.TestCase):
 166     existing_sender = 'Existing Sender <existing@example.com>'
 167     non_existing_sender = 'Non-existing Sender <nonexisting@example.com>'
 168
 169     def mail(self, sender):
 170         return message_from_string('From: %s\nSubject: Test\n\ntest\n' % sender)
 171
 172     def setUp(self):
 173         self.existing_sender_mail = self.mail(self.existing_sender)
 174         self.non_existing_sender_mail = self.mail(self.non_existing_sender)
 175         (self.person, new) = find_author(self.existing_sender_mail)
 176         self.person.save()
 177
 178     def testExisingSender(self):
 179         (person, new) = find_author(self.existing_sender_mail)
 180         self.assertEqual(new, False)
 181         self.assertEqual(person.id, self.person.id)
 182
 183     def testNonExisingSender(self):
 184         (person, new) = find_author(self.non_existing_sender_mail)
 185         self.assertEqual(new, True)
 186         self.assertEqual(person.id, None)
 187
 188     def testExistingDifferentFormat(self):
 189         mail = self.mail('existing@example.com')
 190         (person, new) = find_author(mail)
 191         self.assertEqual(new, False)
 192         self.assertEqual(person.id, self.person.id)
 193
 194     def testExistingDifferentCase(self):
 195         mail = self.mail(self.existing_sender.upper())
 196         (person, new) = find_author(mail)
 197         self.assertEqual(new, False)
 198         self.assertEqual(person.id, self.person.id)
 199
 200     def tearDown(self):
 201         self.person.delete()