如何正确关联 FactoryBoy 子工厂与子对象

Question

当我使用常规数据库对象运行它们时，我有一些测试正在运行，但现在我使用的是 FactoryBoy 工厂。我想我理解它们为什么坏了，但我正在努力寻找正确的设置方法。

这是我的工厂：

@register
class UserFactory(BaseFactory):
    """User factory."""

    username = Sequence(lambda n: 'user{0}'.format(n))
    email = Sequence(lambda n: 'user{0}@example.com'.format(n))
    password = PostGenerationMethodCall('set_password', 'example')
    active = True

    class Meta:
        """Factory configuration."""

        model = User

@register
class ExperimentFactory(BaseFactory):
    """Experiment Factory."""

    date = fake.date_this_decade(before_today=True, after_today=False)
    scanner = Iterator(['GE', 'Sie', 'Phi'])

    class Meta:
        """Factory configuration."""
        model = Experiment

    user = factory.SubFactory(UserFactory)

根据this answer 和其他示例，FactoryBoy 应该在幕后处理外键分配。

但是当我尝试在我的夹具中初始化我的 ExperimentFactory 对象时，我遇到了问题。

@pytest.fixture(scope='function')
@pytest.mark.usefixtures('db')
def mocked_scan_service(db, mocker, request):
    user = UserFactory(password='myprecious')
    db.session.add(user)
    num_exp, num_scans, exp_id, scan_id, exp_uri, scan_uri = request.param

    for i in range(num_exp):
        experiment = ExperimentFactory(user_id = user.id)
        db.session.add(experiment)

    db.session.commit()

    ss = ScanService(user.id, experiment.id)
    for i in range(num_scans):
        ss._add_scan_to_database()

    ss.xc.upload_scan = mocker.MagicMock()
    ss.xc.upload_scan.return_value = ('/data/archive/subjects/000001', exp_uri, scan_uri)
    mocker.spy(ss, '_generate_xnat_identifiers')
    ss.param = request.param
    return ss

如果我没有向 ExperimentFactory 传递用户 ID，我会收到以下错误：

TypeError: __init__() missing 1 required positional argument: 'user_id'

这是模型；我觉得工厂需要一个参数 user_id 来初始化：

class Experiment(SurrogatePK, Model):
    """A user's experiment, during which they are scanned."""

    __tablename__ = 'experiment'
    date = Column(db.Date(), nullable=False)
    scanner = Column(db.String(80), nullable=True)
    num_scans = Column(db.Integer(), nullable=True, default=0)
    xnat_experiment_id = Column(db.String(80), nullable=True)
    xnat_uri = Column(db.String(80), nullable=True)
    user_id = reference_col('user', nullable=False)
    scans = relationship('Scan', backref='experiment')

    def __init__(self, date, scanner, user_id, **kwargs):
        """Create instance."""
        db.Model.__init__(self, date=date, scanner=scanner, user_id=user_id, **kwargs)

    def __repr__(self):
        """Represent instance as a unique string."""
        return '<Experiment({date})>'.format(date=self.date)

但是，如果像我写的那样，我显式地创建了一个用户，然后传递了用户 ID，看起来 ExperimentFactory 最终会用它生成的 SubFactory 覆盖外键。所以稍后当我初始化一个名为ScanService 的对象时，必须使用 user_id 和experiment_id 进行初始化，我的测试由于两个原因之一失败。要么我用我显式创建的用户的 user_id 对其进行初始化，但我的测试失败是因为他们没有找到实验 ID 所属的实验的任何同级实验，或者我用experiment.user.id 对其进行初始化，我的测试失败是因为他们期望数据库中有一个用户，实际上有两个。通过重写我的测试，后一个问题很容易解决，但这似乎很笨拙且不清楚。当 Experiment 模型需要 user_id 进行初始化时，我应该如何初始化 ExperimentFactory？

Answer 1

如果有人有更好的解决方案，请随时发表评论，但这是我意识到的：我为 user_id 传递的内容并不重要；我只需要传递一些东西，这样模型初始化就不会失败。同时传入user=user 会产生我想要的情况：所有实验都属于同一个用户。现在我所有的测试都通过了。这是修改后的夹具代码；其他一切都保持不变：

@pytest.fixture(scope='function')
@pytest.mark.usefixtures('db')
def mocked_scan_service(db, mocker, request):
    num_exp, num_scans, exp_id, scan_id, exp_uri, scan_uri = request.param

    user = UserFactory(password='myprecious')

    for i in range(num_exp):
        experiment = ExperimentFactory(user_id=user.id, user=user)
        db.session.add(experiment)

    db.session.commit()

    ss = ScanService(experiment.user.id, experiment.id)
    for i in range(num_scans):
        ss._add_scan_to_database()

    ss.xc.upload_scan = mocker.MagicMock()
    ss.xc.upload_scan.return_value = ('/data/archive/subjects/000001', exp_uri, scan_uri)
    mocker.spy(ss, '_generate_xnat_identifiers')
    ss.param = request.param
    return ss