summaryrefslogtreecommitdiffstats
path: root/apps/patchwork/tests/patchparser.py
blob: 3b10ef5166808ae7006224fbfb0330e59d06c201 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
# Patchwork - automated patch tracking system
# Copyright (C) 2008 Jeremy Kerr <jk@ozlabs.org>
#
# This file is part of the Patchwork package.
#
# Patchwork is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# Patchwork is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Patchwork; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

import unittest
import os
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from patchwork.models import Project

test_mail_dir  = 'patchwork/tests/mail'
test_patch_dir = 'patchwork/tests/patches'

class PatchTest(unittest.TestCase):
    default_sender = 'Test Author <test@exmaple.com>'
    default_subject = 'Test Subject'
    project = Project(linkname = 'test-project')

    def create_email(self, content, subject = None, sender = None,
            multipart = False):
        if subject is None:
            subject = self.default_subject
        if sender is None:
            sender = self.default_sender

        if multipart:
            msg = MIMEMultipart()
            body = MIMEText(content, _subtype = 'plain')
            msg.attach(body)
        else:
            msg = MIMEText(content)

        msg['Subject'] = subject
        msg['From'] = sender
        msg['List-Id'] = self.project.linkname

        return msg

    def read_patch(self, filename):
        return file(os.path.join(test_patch_dir, filename)).read()


from patchwork.bin.parsemail import find_content

class InlinePatchTest(PatchTest):
    patch_filename = '0001-add-line.patch'
    test_comment = 'Test for attached patch'

    def setUp(self):
        self.orig_patch = self.read_patch(self.patch_filename)
        email = self.create_email(self.test_comment + '\n' + self.orig_patch)
        (self.patch, self.comment) = find_content(self.project, email)

    def testPatchPresence(self):
        self.assertTrue(self.patch is not None)

    def testPatchContent(self):
        self.assertEquals(self.patch.content, self.orig_patch)

    def testCommentPresence(self):
        self.assertTrue(self.comment is not None)

    def testCommentContent(self):
        self.assertEquals(self.comment.content, self.test_comment)


class AttachmentPatchTest(InlinePatchTest):
    patch_filename = '0001-add-line.patch'
    test_comment = 'Test for attached patch'

    def setUp(self):
        self.orig_patch = self.read_patch(self.patch_filename)
        email = self.create_email(self.test_comment, multipart = True)
        attachment = MIMEText(self.orig_patch, _subtype = 'x-patch')
        email.attach(attachment)
        (self.patch, self.comment) = find_content(self.project, email)


class SignatureCommentTest(InlinePatchTest):
    patch_filename = '0001-add-line.patch'
    test_comment = 'Test comment\nmore comment'

    def setUp(self):
        self.orig_patch = self.read_patch(self.patch_filename)
        email = self.create_email( \
                self.test_comment + '\n' + \
                '-- \nsig\n' + self.orig_patch)
        (self.patch, self.comment) = find_content(self.project, email)