mirror of
https://github.com/coddingtonbear/django-mailbox.git
synced 2026-07-09 22:38:19 +02:00
65 lines
3.1 KiB
Python
65 lines
3.1 KiB
Python
from distutils.version import LooseVersion
|
|
from unittest import mock
|
|
|
|
from django.core.management import call_command, CommandError
|
|
from django.test import TestCase
|
|
import django
|
|
|
|
class CommandsTestCase(TestCase):
|
|
def test_processincomingmessage_no_args(self):
|
|
"""Check that processincomingmessage works with no args"""
|
|
|
|
mailbox_name = None
|
|
# Mock handle so that the test doesn't hang waiting for input. Note that we are only testing
|
|
# the argument parsing here -- functionality should be tested elsewhere
|
|
with mock.patch('django_mailbox.management.commands.processincomingmessage.Command.handle') as handle:
|
|
# Don't care about the return value
|
|
handle.return_value = None
|
|
|
|
call_command('processincomingmessage')
|
|
args, kwargs = handle.call_args
|
|
|
|
# Make sure that we called with the right arguments
|
|
try:
|
|
self.assertEqual(kwargs['mailbox_name'], mailbox_name)
|
|
except KeyError:
|
|
# Handle Django 1.7
|
|
# It uses optparse instead of argparse, so instead of being
|
|
# set to None, mailbox_name is simply left out altogether
|
|
# Thus we expect an empty tuple here
|
|
self.assertEqual(args, tuple())
|
|
|
|
|
|
def test_processincomingmessage_with_arg(self):
|
|
"""Check that processincomingmessage works with mailbox_name given"""
|
|
|
|
mailbox_name = 'foo_mailbox'
|
|
|
|
with mock.patch('django_mailbox.management.commands.processincomingmessage.Command.handle') as handle:
|
|
handle.return_value = None
|
|
|
|
call_command('processincomingmessage', mailbox_name)
|
|
args, kwargs = handle.call_args
|
|
try:
|
|
self.assertEqual(kwargs['mailbox_name'], mailbox_name)
|
|
except (AssertionError, KeyError):
|
|
# Handle Django 1.7
|
|
# It uses optparse instead of argparse, so instead of being
|
|
# in kwargs, mailbox_name is in args
|
|
self.assertEqual(args[0], mailbox_name)
|
|
|
|
def test_processincomingmessage_too_many_args(self):
|
|
"""Check that processincomingmessage raises an error if too many args"""
|
|
# Only perform this test for Django versions greater than 1.7.*. This
|
|
# is because, with optparse, too many arguments doesn't result in an
|
|
# error, which means this test is worthless anyway
|
|
# For the "compatibility" versions, unexpected arguments aren't handled
|
|
# very well, and result in a TypeError
|
|
if (LooseVersion(django.get_version()) >= LooseVersion('1.8') and
|
|
LooseVersion(django.get_version()) < LooseVersion('1.10')):
|
|
with self.assertRaises(TypeError):
|
|
call_command('processincomingmessage', 'foo_mailbox', 'invalid_arg')
|
|
# In 1.10 and later a proper CommandError should be raised
|
|
elif LooseVersion(django.get_version()) >= LooseVersion('1.10'):
|
|
with self.assertRaises(CommandError):
|
|
call_command('processincomingmessage', 'foo_mailbox', 'invalid_arg')
|