如何在 django 中使用验证码字段对表单进行单元测试？

Question

我想通过提交表单来对 django 视图进行单元测试。问题是这个表单有一个验证码字段（基于 django-simple-captcha）。

from django import forms
from captcha.fields import CaptchaField

class ContactForm(forms.forms.Form):
    """
    The information needed for being able to download
    """
    lastname = forms.CharField(max_length=30, label='Last name')
    firstname = forms.CharField(max_length=30, label='First name')
    ...
    captcha = CaptchaField()

测试代码：

class ContactFormTest(TestCase):

    def test_submitform(self):
        """Test that the contact page"""
        url = reverse('contact_form')

        form_data = {}
        form_data['firstname'] = 'Paul'
        form_data['lastname'] = 'Macca'
        form_data['captcha'] = '28if'

        response = self.client.post(url, form_data, follow=True)

是否有任何方法可以对该代码进行单元测试并在测试时摆脱验证码？

提前致谢

Answer 1

我知道这是一篇旧文章，但 django-simple-captcha 现在有一个设置 CAPTCHA_TEST_MODE，如果您提供值“PASSED”，则验证码会成功。您只需确保为两个验证码输入字段发送一些内容：

post_data['captcha_0'] = 'dummy-value'
post_data['captcha_1'] = 'PASSED'
self.client.post(url, data=post_data)

CAPTCHA_TEST_MODE 设置仅应在测试期间使用。我的设置.py：

if 'test' in sys.argv:
    CAPTCHA_TEST_MODE = True 

Answer 2

这是我解决这个问题的方法。导入实际保存验证码信息的模型：

from captcha.models import CaptchaStore

首先，我检查测试验证码表是否为空：

captcha_count = CaptchaStore.objects.count()
self.failUnlessEqual(captcha_count, 0)

加载页面（在本例中，它是注册页面）后，检查是否有一个新的验证码对象实例：

captcha_count = CaptchaStore.objects.count()
self.failUnlessEqual(captcha_count, 1)

然后，我检索验证码实例数据并将其与表单一起发布。就我而言，POST 期望“captcha_0”包含哈希键，“captcha_1”包含响应。

captcha = CaptchaStore.objects.all()[0]
registration_data = { # other registration data here
                     'captcha_0': captcha.hashkey,
                     'captcha_1': captcha.response }

如果您在运行此测试之前从 CaptchaStore 实例开始，您可能需要对此进行一些调整。希望有帮助。

Answer 3

我们是这样做的。

@patch("captcha.fields.ReCaptchaField.validate")
def test_contact_view(self, validate_method):

    response = self.client.get(reverse("contact"))
    self.assertEqual(response.status_code, 200)

    data = {
        "name": "Bob Johnson",
        "email": "[email protected]",
        "phone": "800-212-2001",
        "subject": "I want Axis!",
        "message": "This is a giant\nThree liner..\nLove ya\n",
        "captcha": "XXX",
    }
    validate_method.return_value = True
    response = self.client.post(reverse("contact"), data=data)

    self.assertEqual(response.status_code, 302)

Answer 4

我通过模拟 ReCaptchaField 对它进行了单元测试。首先，我在构造函数中添加了 recaptcha 字段。它不能作为常规字段添加，因为您将无法模拟它（一旦在应用模拟之前评估代码）：

class MyForm(forms.ModelForm):

    ...

    def __init__(self, *args, **kwargs):
        # Add captcha in the constructor to allow mock it
        self.fields["captcha"] = ReCaptchaField()

然后，我只是将 ReCaptchaField 替换为不需要的 CharField。这样，我相信 django-recaptcha 会起作用。我只能测试我自己的东西：

@mock.patch("trials.forms.ReCaptchaField", lambda: CharField(required=False))
def test_my_stuff(self):
    response = self.client.post(self.url, data_without_captcha)
    self.assert_my_response_fit_the_needs(response)

Answer 5

一种解决方案是设置“测试”，该值要么为真，要么为假。然后就很

if not testing:
   # do captcha stuff here

简单了，而且切换也很容易。

Answer 6

另一个解决方案类似于 Jim McGaw 的答案，但不需要空表 CaptchaStore 表。

captcha = CaptchaStore.objects.get(hashkey=CaptchaStore.generate_key())

registration_data = { # other registration data here
                 'captcha_0': captcha.hashkey,
                 'captcha_1': captcha.response }

这将为该测试生成新的验证码。

Answer 7

这是唯一对我有用的方法，

在测试设置方法中设置 CAPTCHA_TEST_MODE=True 。

class ApplicationTestCase(TestCase):
    def setUp(self):
        self.client = Client()
        self.url = reverse('application')
        from captcha.conf import settings as captcha_settings
        captcha_settings.CAPTCHA_TEST_MODE = True
    
    def test_post_valid_form(self):
        data = {
            'name': 'John Doe',
            "captcha_0": "8e10ebf60c5f23fd6e6a9959853730cd69062a15",
            "captcha_1": "PASSED",
        }

        response = self.client.post(self.url, data)
        self.assertEqual(response.status_code, 200)

Answer 8

采用与 Jim McGaw 类似的方法，但使用 BeautifulSoup：

from captcha.models import CaptchaStore
from BeautifulSoup import BeautifulSoup

data = {...} #The data to post
soup = BeautifulSoup(self.client.get(url).content)
for field_name in ('captcha_0', ...): #get fields from the form
    data[field_name] = soup.find('input',{'name':field_name})['value']
captcha = CaptchaStore.objects.get(hashkey=data['captcha_0'])
data['captcha_1'] = captcha.challenge
response = self.client.post(url, data=data)

# check the results
...