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 patchwork.models import Project, Person
23 from django.contrib.auth.models import User
24 from django.forms.fields import EmailField
26 from email import message_from_file
28 from email.mime.text import MIMEText
29 from email.mime.multipart import MIMEMultipart
31 # Python 2.4 compatibility
32 from email.MIMEText import MIMEText
33 from email.MIMEMultipart import MIMEMultipart
35 # helper functions for tests
36 _test_mail_dir = os.path.join(os.path.dirname(__file__), 'mail')
37 _test_patch_dir = os.path.join(os.path.dirname(__file__), 'patches')
39 class defaults(object):
40 project = Project(linkname = 'test-project', name = 'Test Project',
41 listid = 'test.example.com')
43 patch_author = 'Patch Author <patch-author@example.com>'
44 patch_author_person = Person(name = 'Patch Author',
45 email = 'patch-author@example.com')
47 comment_author = 'Comment Author <comment-author@example.com>'
49 sender = 'Test Author <test-author@example.com>'
51 subject = 'Test Subject'
53 patch_name = 'Test Patch'
55 patch = """--- /dev/null 2011-01-01 00:00:00.000000000 +0800
56 +++ a 2011-01-01 00:00:00.000000000 +0800
62 'email': 'Enter a valid email address.',
68 userid = 'test%d' % _user_idx
69 email = '%s@example.com' % userid
72 user = User.objects.create_user(userid, email, userid)
75 person = Person(email = email, name = userid, user = user)
80 def create_maintainer(project):
82 profile = user.profile
83 profile.maintainer_projects.add(project)
87 def find_in_context(context, key):
88 if isinstance(context, list):
90 v = find_in_context(c, key)
98 def read_patch(filename, encoding = None):
99 file_path = os.path.join(_test_patch_dir, filename)
100 if encoding is not None:
101 f = codecs.open(file_path, encoding = encoding)
107 def read_mail(filename, project = None):
108 file_path = os.path.join(_test_mail_dir, filename)
109 mail = message_from_file(open(file_path))
110 if project is not None:
111 mail['List-Id'] = project.listid
114 def create_email(content, subject = None, sender = None, multipart = False,
115 project = None, content_encoding = None):
117 subject = defaults.subject
119 sender = defaults.sender
121 project = defaults.project
122 if content_encoding is None:
123 content_encoding = 'us-ascii'
126 msg = MIMEMultipart()
127 body = MIMEText(content, _subtype = 'plain',
128 _charset = content_encoding)
131 msg = MIMEText(content, _charset = content_encoding)
133 msg['Subject'] = subject
135 msg['List-Id'] = project.listid