1

我在 Ubuntu 13.04 上使用 Django 1.4 和 Python 2.7。我正在处理的 Django 项目有一个使用 Nose 和 Coverage 的完整测试套件。在运行任何测试之前，我需要在数据库中创建一个用户。我会在哪里做这个？

system_user = User(username=accounts_constants.SYSTEM_USER)
system_user.save()

这样做的动机源于处理大量数据库事务的“系统机器人”的存在。我在我们的项目中创建了一个完整的跟踪套件。如果用户更改任何敏感信息，我们会对其进行跟踪。如果系统进行了这些更改，我们希望系统用户被列在跟踪信息中。

代码中有几个地方需要修改以支持此更改。显然测试数据库没有这个用户。我想在任何测试运行之前添加它（但在创建数据库之后）。

我一直在挖掘，但无法确切地看到为测试目的而创建数据库的位置。我的第一次尝试是在我们的测试套件运行器的初始化中。

class CustomTestSuiteRunner(NoseTestSuiteRunner):
    """
    Runs testing suite and adds code coverage report.
    """
    def __init__(self, *args, **kwargs):
        super(CustomTestSuiteRunner, self).__init__(*args, **kwargs)
        self.coverage = coverage.coverage()
        self.coverage.use_cache(0) # don't use caching with coverage.py
        system_user = User(username=accounts_constants.SYSTEM_USER)
        system_user.save()

这失败了，因为它仍在使用开发数据库（系统用户已存在于该数据库中）。

我尝试在测试序列中进一步添加用户，但似乎无法找到合适的位置。

任何方向将不胜感激。

编辑1：

来自 girasquid 建议的解决方案：

我不想将每个修改setUp为超级到CustomTestCase. 相反，我试图在__init__.

class CustomTestCase(TestCase):
    def __init__(self, *args, **kwargs):
        super(CustomTestCase, self).__init__(*args, **kwargs)
        try:
            system_user = User.objects.create(
                username=accounts_constants.SYSTEM_USER)
            system_user.save()
        except psycopg2.IntegrityError:
            pass

但我得到了一个IntegrityErrorfrom nose。任何想法为什么会这样？

lib/python2.7/site-packages/nose/loader.py", line 518, in makeTest
    return self._makeTest(obj, parent)
IntegrityError: duplicate key value violates unique constraint "auth_user_username_key"

编辑2：

我们的团队说服我实施逐案设置更改以包括用户而不是系统范围的更改。

4

1 回答 1

3

我会 为此在我的测试用例中定义setUp ：

class MyTestCase(unittest.TestCase):
    def setUp(self):
        self.system_user = User.objects.create(username=accounts_constants.SYSTEM_USER)

如果您有很多需要执行此操作的测试用例，则将其放在基础上TestCase然后从该基础上继承所有测试可能会很有用。

于 2013-05-29T19:03:50.143 回答