Jan Dittberner
3d18392b67
- drop Python 2 __future__ imports - fix tests to handle new Django and Python 3 module names - reformat changed files with black
84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
"""
|
|
Tests for :py:mod:`contact_form.forms`.
|
|
|
|
"""
|
|
from unittest.mock import MagicMock, Mock, patch
|
|
|
|
from django.contrib.sites.models import Site
|
|
from django.test import TestCase
|
|
from django.urls import reverse
|
|
|
|
from contact_form.forms import ContactForm
|
|
|
|
TEST_DATA = {"name": "Test User", "email": "test@example.org", "body": "Test message"}
|
|
|
|
|
|
class ContactFormTest(TestCase):
|
|
def test_constructor_needs_request(self):
|
|
with self.assertRaises(KeyError):
|
|
ContactForm()
|
|
|
|
def test_constructor(self):
|
|
request = MagicMock()
|
|
form = ContactForm(request=request)
|
|
self.assertTrue(hasattr(form, "request"))
|
|
self.assertEqual(form.request, request)
|
|
self.assertTrue(hasattr(form, "helper"))
|
|
self.assertEqual(form.helper.form_action, reverse("contact_form"))
|
|
self.assertEqual(len(form.helper.inputs), 1)
|
|
self.assertEqual(form.helper.inputs[0].name, "submit")
|
|
|
|
def test_constructor_fields(self):
|
|
request = MagicMock()
|
|
form = ContactForm(request=request)
|
|
self.assertEqual(len(form.fields), 3)
|
|
self.assertIn("email", form.fields)
|
|
self.assertIn("name", form.fields)
|
|
self.assertIn("body", form.fields)
|
|
self.assertEqual(len(form.data), 0)
|
|
|
|
def test_get_context_invalid(self):
|
|
request = MagicMock()
|
|
form = ContactForm(request=request)
|
|
with self.assertRaisesMessage(
|
|
ValueError, "Cannot generate context from invalid contact form"
|
|
):
|
|
form.get_context()
|
|
|
|
def test_get_context_valid_site_installed(self):
|
|
request = MagicMock()
|
|
form = ContactForm(request=request, data=TEST_DATA)
|
|
context = form.get_context()
|
|
self.assertIn("site", context)
|
|
self.assertIn("name", context)
|
|
self.assertIn("email", context)
|
|
self.assertIn("body", context)
|
|
|
|
def test_get_context_valid_site_not_installed(self):
|
|
request = MagicMock()
|
|
form = ContactForm(request=request, data=TEST_DATA)
|
|
with patch("contact_form.forms.Site") as sitemock:
|
|
sitemock._meta.installed = False
|
|
context = form.get_context()
|
|
self.assertIn("site", context)
|
|
self.assertIn("name", context)
|
|
self.assertIn("email", context)
|
|
self.assertIn("body", context)
|
|
|
|
def test_message(self):
|
|
request = Mock()
|
|
request.META = {"REMOTE_ADDR": "127.0.0.1"}
|
|
form = ContactForm(request=request, data=TEST_DATA)
|
|
message = form.message()
|
|
self.assertIn(TEST_DATA["name"], message)
|
|
self.assertIn(TEST_DATA["email"], message)
|
|
self.assertIn(TEST_DATA["body"], message)
|
|
self.assertIn("127.0.0.1", message)
|
|
|
|
def test_subject(self):
|
|
request = Mock()
|
|
form = ContactForm(request=request, data=TEST_DATA)
|
|
subject = form.subject()
|
|
self.assertIn(Site.objects.get_current().name, subject)
|
|
self.assertIn(TEST_DATA["name"], subject)
|