Skip to main content
 首页 » 编程设计

python之Django之本地主机上的单元测试在主数据库中创建对象,但稍后在方法中未在测试数据库中找到它们

2025年05月04日26youxin

在 Django 单元测试中遇到一些相当奇怪的行为。我有一个继承自 django.test.testcase 的类,它验证注册表单创建了一个人对象和一个用户对象,它确实如此,但我随后可以在测试完成后在 Admin 界面中查找和找到这些对象。这是测试:

import unittest 
import time 
from django.test import TestCase 
from selenium import webdriver 
from selenium.webdriver.common.keys import Keys 
from django.contrib.auth.models import User 
from apps.persons.models import Person 
 
class NewVisitorTest(TestCase): 
 
def setUp(self): 
    self.browser = webdriver.Firefox() 
    self.browser.implicitly_wait(3) 
 
def tearDown(self): 
    self.browser.quit() 
 
def test_can_sign_up_and_login(self): 
 
    # User goes to register url 
    self.browser.get('http://localhost:8000/register/') 
 
    # User can see that she has come to the right page on the website 
    self.assertIn('Register', self.browser.title) 
 
    # User can see that there is a sign up form 
    signup_form = self.browser.find_element_by_id('id_registration_form') 
 
    # User can enter text into the first name field 
    first_name_input = self.browser.find_element_by_id('id_first_name') 
    first_name_input.send_keys('Jim') 
 
    # User can enter text into the last name field 
    last_name_input = self.browser.find_element_by_id('id_last_name') 
    last_name_input.send_keys('bob') 
 
    # User can enter text into the username field 
    username_input = self.browser.find_element_by_id('id_username') 
    username_input.send_keys('jim_bob') 
 
    # User can enter text into the email field 
    email_input = self.browser.find_element_by_id('id_email') 
    email_input.send_keys('jim_bob@jimbob.com') 
 
    # User can enter text into the first password field 
    password_input = self.browser.find_element_by_id('id_password') 
    password_input.send_keys('kittensarecute') 
 
    # User can enter text into the second password field 
    password_1_input = self.browser.find_element_by_id('id_password1') 
    password_1_input.send_keys('kittensarecute') 
 
    # Submit form 
    signup_form.submit() 
 
    time.sleep(20) 
 
 
    persons = Person.objects.all() 
    print persons 
    self.assertEqual(len(persons), 1) 
 
    users = User.objects.all() 
    print users 
    self.assertEqual(len(users), 1) 
 
    # Deliberately fail 
    self.fail('Finish the test!') 
 
if __name__ == '__main__': 
    unittest.main() 

代码在 assertEqual(len(persons), 1) 上失败。我可以在浏览器中重现相同的行为并且没有任何问题。每次我运行测试时,我都必须删除它在主本地数据库中创建的用户,所以我假设它将对象保存在设置 ['数据库'] 中定义的数据库中,然后在测试数据库中查找它们稍后在方法中。我在终端中看到测试数据库是在每次测试开始时创建的,所以它肯定在那里,这似乎令人困惑。任何帮助将不胜感激。

编辑:一切都在 sqlite 上

 File ".../tests/tests.py", line 60, in test_can_sign_up_and_login 
    self.assertEqual(len(persons), 1) 
AssertionError: 0 != 1 

编辑 2:

DATABASES = { 
    'default': { 
        'ENGINE': 'django.db.backends.sqlite3', # Add 'postgresql_psycopg2', 'mysql', 'sqlite3' or 'oracle'. 
        'NAME': '/Users/me/github/stage/project/data/db.db', # Or path to database file if using sqlite3. 
        # The following settings are not used with sqlite3: 
        'USER': '', 
        'PASSWORD': '', 
        'HOST': '',                      # Empty for localhost through domain sockets or '127.0.0.1' for localhost through TCP. 
        'PORT': '',                      # Set to empty string for default. 
    } 
} 

我有一个包含应用程序的文件夹,因此要运行这些测试并运行:

python manage.py test tests 

请您参考如下方法:

你是对的,django 在你的测试中创建了一个测试数据库。来自文档:

Tests that require a database (namely, model tests) will not use your “real” (production) database. Separate, blank databases are created for the tests. full documentation

这是正确的做法。单元测试应该彼此隔离,以避免脏环境(之前的测试忘记删除它创建的记录/文件/对象,这会影响其他测试的结果)。


你正在做的叫做Functional Testing .您正在测试您的 Web 应用程序表单是否在一次测试中提交、验证、保存等。由于执行此操作意味着要运行许多代码片段,因此它不被视为单元 测试。当您执行这些类型的测试时,您的断言也应该类似:

  • 断言显示感谢/欢迎消息
  • 转到用户管理页面并断言那里有新人
  • 检查他们是否可以记录
  • 等等

这会测试您的应用程序是否正常运行。

这两种测试之间的区别在于,功能测试测试您的应用程序在做什么,而单元测试测试如何它在做什么。

这很重要,因为您可以通过多种方式保存数据:

  • 将 JSON 作为文件
  • 作为 XML
  • 在本地数据库中
  • 在云端某处

您的功能测试并不真正关心如何这是如何完成的,它只关心当您转到特定页面时,它会获取数据。但单元测试会根据您使用的方式发生巨大变化。


在您的情况下,您正在使用两种测试方式。 Selenium(打开页面、填写页面、提交页面)正在测试应用程序的功能,但您的断言(数据是否已保存到数据库)正在测试如何 你存储的数据(单位)。

因此我建议您更改您的断言以适合您的其余测试。并将当前的移动到您的单元测试中:

def test_user_is_save(self): 
   addUser(...) 
   users = User.objects.all() 
   self.assertEqual(len(users), 1) 

这将测试您的 addUser 方法是否正常工作(保存到数据库),这就是 unit 测试的目的。您当前的测试将测试当您提交表单时,用户会看到正确的内容。