git.ozlabs.org Git - patchwork/blob - apps/patchwork/tests/utils.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 os
  21 import codecs
  22 from patchwork.models import Project, Person
  23 from django.contrib.auth.models import User
  24 from django.forms.fields import EmailField
  25
  26 from email import message_from_file
  27 try:
  28     from email.mime.text import MIMEText
  29     from email.mime.multipart import MIMEMultipart
  30 except ImportError:
  31     # Python 2.4 compatibility
  32     from email.MIMEText import MIMEText
  33     from email.MIMEMultipart import MIMEMultipart
  34
  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')
  38
  39 class defaults(object):
  40     project = Project(linkname = 'test-project', name = 'Test Project')
  41
  42     patch_author = 'Patch Author <patch-author@example.com>'
  43     patch_author_person = Person(name = 'Patch Author',
  44         email = 'patch-author@example.com')
  45
  46     comment_author = 'Comment Author <comment-author@example.com>'
  47
  48     sender = 'Test Author <test-author@example.com>'
  49
  50     subject = 'Test Subject'
  51
  52     patch_name = 'Test Patch'
  53
  54     patch = """--- /dev/null    2011-01-01 00:00:00.000000000 +0800
  55 +++ a   2011-01-01 00:00:00.000000000 +0800
  56 @@ -0,0 +1 @@
  57 +a
  58 """
  59
  60 error_strings = {
  61     'email': 'Enter a valid email address.',
  62 }
  63
  64 _user_idx = 1
  65 def create_user():
  66     global _user_idx
  67     userid = 'test%d' % _user_idx
  68     email = '%s@example.com' % userid
  69     _user_idx += 1
  70
  71     user = User.objects.create_user(userid, email, userid)
  72     user.save()
  73
  74     person = Person(email = email, name = userid, user = user)
  75     person.save()
  76
  77     return user
  78
  79 def create_maintainer(project):
  80     user = create_user()
  81     profile = user.get_profile()
  82     profile.maintainer_projects.add(project)
  83     profile.save()
  84     return user
  85
  86 def find_in_context(context, key):
  87     if isinstance(context, list):
  88         for c in context:
  89             v = find_in_context(c, key)
  90             if v is not None:
  91                 return v
  92     else:
  93         if key in context:
  94             return context[key]
  95     return None
  96
  97 def read_patch(filename, encoding = None):
  98     file_path = os.path.join(_test_patch_dir, filename)
  99     if encoding is not None:
 100         f = codecs.open(file_path, encoding = encoding)
 101     else:
 102         f = file(file_path)
 103
 104     return f.read()
 105
 106 def read_mail(filename, project = None):
 107     file_path = os.path.join(_test_mail_dir, filename)
 108     mail = message_from_file(open(file_path))
 109     if project is not None:
 110         mail['List-Id'] = project.listid
 111     return mail
 112
 113 def create_email(content, subject = None, sender = None, multipart = False,
 114         project = None, content_encoding = None):
 115     if subject is None:
 116         subject = defaults.subject
 117     if sender is None:
 118         sender = defaults.sender
 119     if project is None:
 120         project = defaults.project
 121     if content_encoding is None:
 122         content_encoding = 'us-ascii'
 123
 124     if multipart:
 125         msg = MIMEMultipart()
 126         body = MIMEText(content, _subtype = 'plain',
 127                         _charset = content_encoding)
 128         msg.attach(body)
 129     else:
 130         msg = MIMEText(content, _charset = content_encoding)
 131
 132     msg['Subject'] = subject
 133     msg['From'] = sender
 134     msg['List-Id'] = project.linkname
 135
 136
 137     return msg