Модульное тестирование и издевательство над отправителем электронной почты в Python с помощью Google AppEngine - PullRequest
Новая
       12

Модульное тестирование и издевательство над отправителем электронной почты в Python с помощью Google AppEngine

5 голосов
/ 09 января 2009

Я новичок в Python и движке приложений.

У меня есть этот код, который отправляет электронное письмо на основе параметров запроса после некоторой логики авторизации. в моих модульных тестах (я использую GAEUnit ), как мне подтвердить, что письмо с определенным содержимым было отправлено? - как я могу издеваться над электронным почтовым ящиком, чтобы убедиться, что отправка была вызвана? 

class EmailHandler(webapp.RequestHandler):
 def bad_input(self):
  self.response.set_status(400)
  self.response.headers['Content-Type'] = 'text/plain'
  self.response.out.write("<html><body>bad input </body></html>")

 def get(self):
  to_addr = self.request.get("to")
  subj = self.request.get("subject")
  msg = self.request.get("body")
  if not mail.is_email_valid(to_addr):
    # Return an error message...
    #   self.bad_input()
    pass

  # authenticate here

  message = mail.EmailMessage()
  message.sender = "my.company@gmail.com"
  message.to = to_addr
  message.subject = subj
  message.body = msg
  message.send()
  self.response.headers['Content-Type'] = 'text/plain'
  self.response.out.write("<html><body>success!</body></html>")

И юнит-тесты 

import unittest
from webtest import TestApp
from google.appengine.ext import webapp
from email import EmailHandler

class SendingEmails(unittest.TestCase):

  def setUp(self):
    self.application = webapp.WSGIApplication([('/', EmailHandler)], debug=True)

  def test_success(self):
    app = TestApp(self.application)
    response = app.get('http://localhost:8080/send?to=vijay.santhanam@gmail.com&body=blah_blah_blah&subject=mySubject')
    self.assertEqual('200 OK', response.status)
    self.assertTrue('success' in response)
    # somehow, assert email was sent

Ответы [ 5 ]

3 голосов
/ 11 сентября 2009

Вы также можете переопределить метод _GenerateLog в mail_stub внутри AppEngine.

Вот родительский класс TestCase, который я использую в качестве миксина при проверке отправки электронных писем: 

from google.appengine.api import apiproxy_stub_map, mail_stub

__all__ = ['MailTestCase']

class MailTestCase(object):
    def setUp(self):
        super(MailTestCase, self).setUp()
        self.set_mail_stub()
        self.clear_sent_messages()

    def set_mail_stub(self):
        test_case = self
        class MailStub(mail_stub.MailServiceStub):
            def _GenerateLog(self, method, message, log, *args, **kwargs):
                test_case._sent_messages.append(message)
                return super(MailStub, self)._GenerateLog(method, message, log, *args, **kwargs)

        if 'mail' in apiproxy_stub_map.apiproxy._APIProxyStubMap__stub_map:
            del apiproxy_stub_map.apiproxy._APIProxyStubMap__stub_map['mail']

        apiproxy_stub_map.apiproxy.RegisterStub('mail', MailStub())

    def clear_sent_messages(self):
        self._sent_messages = []

    def get_sent_messages(self):
        return self._sent_messages

    def assertEmailSent(self, to=None, sender=None, subject=None, body=None):
        for message in self.get_sent_messages():
            if to and to not in message.to_list(): continue
            if sender and sender != message.sender(): continue
            if subject and subject != message.subject(): continue
            if body and body not in message.textbody(): continue
            return

        failure_message = "Expected e-mail message sent."

        args = []
        if to: args.append('To: %s' % to)
        if sender: args.append('From: %s' % sender)
        if subject: args.append('Subject: %s' % subject)
        if body: args.append('Body (contains): %s' % body)

        if args:
            failure_message += ' Arguments expected: ' + ', '.join(args)

        self.fail(failure_message)

После этого пример теста может выглядеть следующим образом: 

import unittest, MailTestCase

class MyTestCase(unittest.TestCase, MailTestCase):
    def test_email_sent(self):
        send_email_to('test@example.org') # Some method that would send an e-mail.
        self.assertEmailSent(to='test@example.org')
        self.assertEqual(len(self.get_sent_messages()), 1)
2 голосов
/ 11 сентября 2015

У Google есть очень простой способ сделать это в их документации .

Вы создаете тестовую площадку и почтовую заглушку в вашей настройке следующим образом: 

self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_mail_stub()
self.mail_stub = self.testbed.get_stub(testbed.MAIL_SERVICE_NAME)

Запустите ваш метод и проверьте его следующим образом: 

 messages = self.mail_stub.get_sent_messages(to='alice@example.com')
 self.assertEqual(1, len(messages))
 self.assertEqual('alice@example.com', messages[0].to)
2 голосов
/ 28 сентября 2011

Просто используйте следующее, чтобы получить все сообщения, отправленные после активации почтовой заглушки. 

from google.appengine.api import apiproxy_stub_map
sent_messages = apiproxy_stub_map.apiproxy._APIProxyStubMap__stub_map['mail'].get_sent_messages()
2 голосов
/ 09 января 2009

Очень краткое введение обеспечивает PyPI: MiniMock 1.0 . Это очень маленькая библиотека для установки макетов.

  1. Вставьте макет в модуль, который должен быть смоделирован
  2. Определите, что вернет ваш макет
  3. вызов метода
  4. Ваша издевка скажет, какой метод был вызван.

Удачи!

1 голос
/ 04 апреля 2010

Я использовал тестовый стенд GAE от jgeewax с GAEUnit. Использовать GAEUnit вместо NoseGAE для меня было проще, так как у меня уже был GAEUnit. Вот шаги:

Добавить GAEUnit в свое приложение

  1. Загрузите сжатый архив GAEUnit со его страницы хостинга проекта Google Code .
  2. Распакуйте архив.
  3. Из папки, извлеченной из архива, скопируйте gaeunit.py в корневую папку вашего приложения.
  4. Добавьте следующие 2 строки к вашему app.yaml, прямо под строкой, которая говорит handlers::

Код: 

- url: /test.*
  script: gaeunit.py

(Необязательно) Из папки, извлеченной из архива, находится папка с именем sample_app, и внутри нее находится модифицированная версия модуля webtest. Скопируйте модуль webtest (вся папка, содержащая debugapp.py и __init__.py) в корень вашего приложения.

Добавить испытательный стенд GAE в GAEUnit:

  1. Загрузите сжатый архив GAE Testbed tar со с его страницы размещения проекта Google Code .
  2. Распакуйте архив.
  3. Внутри распакованного архива находится модуль gaetestbed (это папка с именем «gaetestbed»). Скопируйте модуль в корень вашего приложения.
  4. Создайте файл в тестовой папке корня вашего приложения. Ради этого примера назовем его test_mailer.py.
  5. Используя пример со страницы хостинга проекта Google Code для тестового стенда GAE, добавьте следующие строки в test_mailer.py:

Код: 

import unittest
from gaetestbed import MailTestCase

class MyTestCase(MailTestCase, unittest.TestCase):
    def test_email_sent(self):
        send_email_to('test@example.org') # Some method that sends e-mail...
        self.assertEmailSent(to='test@example.org')
        self.assertEqual(len(self.get_sent_messages()), 1)

Запустите свой сервер и перейдите на http://localhost:8080/test. Вы должны заметить, что (дополнительный) тест 1/1 был запущен с http://localhost:8080/test.

Источник: Использование GAE Testbed с GAEUnit: проверка того, что электронное письмо было отправлено

...