Question

我有一个具有一些非托管模型的应用程序，这些模型是从另一个DB中的表中提取的。

我已经为我的所有模型创建了工厂，并且在大多数情况下，factory_boy工作得很好！

我遇到的问题与特定的非托管模型有关，该模型在其中一个字段上使用管理器。

class DailyRoll(StaffModel):
    id            = models.IntegerField(db_column='ID', primary_key=True)
    location_id   = models.IntegerField(db_column='locationID')
    grade         = models.CharField(db_column='grade', max_length=10)
    end_year      = models.IntegerField(db_column='endYear')
    run_date      = models.DateField(db_column='runDate')
    person_count = models.IntegerField(db_column='personCount')
    contained_count = models.IntegerField(db_column='containedCount')
    double_count = models.IntegerField(db_column='doubleCount')
    aide_count = models.IntegerField(db_column='aideCount')
    one_count = models.IntegerField(db_column='oneCount')
    bi_count = models.IntegerField(db_column='biCount')
    last_run_date = managers.DailyRollManager()

    class Meta(object):
        managed  = False
        db_table = '[DATA].[position_dailyRoll]'

经理如下：

class DailyRollManager(models.Manager):
    def get_queryset(self):
        last_run_date = super().get_queryset().using('staff').last().run_date
        return super().get_queryset().using('staff').filter(
            run_date=last_run_date
        )

我的factory_boy设置看起来像这样（目前非常基本，因为我只是想使其正常工作）：

class DailyRollFactory(factory.Factory):
    class Meta:
        model = staff.DailyRoll

    id = 1
    location_id = 1
    grade = 1
    end_year = 2019
    run_date = factory.fuzzy.FuzzyDateTime(timezone.now())
    person_count = 210
    contained_count = 1
    double_count = 2
    aide_count = 3
    one_count = 4
    bi_count = 5
    last_run_date = managers.DailyRollManager()

我确定last_run_date的设置不正确-但我不知道如何最好地使用factory_boy中的管理器。

当我运行测试时-它创建一个名为'defaultdb'的默认sqlite3数据库来运行测试（我们在生产环境中使用SQL Server-但由于安全性原因，我们无法让Django控制master来管理测试）。为此，我必须创建一种解决方法，但是它可以工作。

我的设置如下：

if 'test' in sys.argv or 'test_coverage' in sys.argv:
    DATABASES = {
        'default': {
            'ENGINE': 'django.db.backends.sqlite3',
            'NAME': os.path.join(BASE_DIR, 'defaultdb.sqlite3'),
        },
        'staff': {
            'ENGINE': 'django.db.backends.sqlite3',
            'NAME': os.path.join(BASE_DIR, 'staff.sqlite3'),
        },
    }

我的测试很简单：

def test_index_returns_200_and_correct_template(self):
    """
    index page returns a 200
    """
    # Load test data which index needs
    # if this is not included the test fails with
    # 'no such table: [DATA].[position_dailyRoll]'
    f.DailyRollFactory()
    response = self.client.get(reverse('position:index'))
    self.assertEqual(response.status_code, 200)
    self.assertTemplateUsed(response, 'position/index.html')

如果我未在测试中包括f.DailyRollFactory()，它将失败并显示以下错误：

no such table: [DATA].[position_dailyRoll]

当我包含f.DailyRollFactory()时，会出现此错误：

TypeError: 'last_run_date' is an invalid keyword argument for this function

我认为，最终该经理正在尝试查看“ staff”数据库-我已经创建了该数据库，但是该数据库为空并且不包含任何数据。

我正在尝试找出：

A）我该如何用我工厂男孩模型中的数据预先填充staff数据库？我认为可能会解决我遇到的问题。会吗我不确定。

B）在factory_boy工厂处理经理是正确的方法吗？我不确定该如何处理。

Answer 1

我最终使用Pytest-Django，因为该测试框架实际上可以执行我想要的事情。

使用--nomigrations标志，它可以获取仅由django在测试中管理的模型，并在测试数据库中为其创建适当的表名（使用其db_table属性）。然后，我可以使用factory_boy创建模拟数据并进行测试！

使用factory_boy数据填充特定数据库以进行测试

1 个答案: