Question

我第一次使用夹具模块，试图获得更好的夹具数据，这样我就可以使我们的功能测试更加完整。

我发现夹具模块有点笨重，我希望有更好的方法来做我正在做的事情。这是Python 2.7中的Flask / SQLAlchemy应用程序，我们使用nose作为测试运行器。

所以我有一组员工。员工有角色。有几个页面具有相当复杂的权限，我想确保这些页面经过测试。

我创建了一个具有每种角色类型的DataSet（我们的应用程序中大约有15个角色）：

class EmployeeData(DataSet):

  class Meta:
    storable = Employee

  class engineer:
    username = "engineer"
    role = ROLE_ENGINEER

  class manager:
    username = "manager"
    role = ROLE_MANAGER

  class admin:
    username = "admin"
    role = ROLE_ADMIN

我想做的是编写一个功能测试，只检查合适的人可以访问页面。（实际的权限更复杂，我只是想给你一个玩具示例。）

这样的事情：

def test_only_admin_can_see_this_page():

  for employee in Employee.query.all():
    login(employee)

    with self.app.test_request_context('/'):
    response = self.test_client.get(ADMIN_PAGE)
    if employee.role == ROLE_ADMIN
      eq_(200, response.status_code)
    else:
      eq_(401, response.status_code)

    logout(employee)

有没有办法生成灯具数据，所以我的开发人员不必记得每次添加角色时都会在灯具上添加一行？我们将所有角色的规范列表作为应用程序中其他位置的配置，所以我有。

我没有与任何这个或夹具模块结合，所以我很高兴听到建议！

Answer 1

选项是使用factory_boy创建测试数据。

假设您保留并相应更新了一个角色列表（稍后将使用），如下所示：
```
roles = [ROLE_ENGINEER, ROLE_ADMIN, ROLE_MANAGER, ...]
```

让我们为Employee表创建一个工厂：

import factory
from somewhere.in.the.app import roles

class EmployeeFactory(factory.alchemy.SQLAlchemyModelFactory):
    class Meta:
        model = Employee
        sqlalchemy_session = session

    username = factory.Sequence(lambda n: u'User %d' % n)
    # Other attributes
    ...
    # Now the role choice
    role = factory.fuzzy.FuzzyChoice(roles)

FuzzyChoice方法采用一系列选项，并从此列表中随机选择现在，这将能够按需创建任意数量的Employee个对象。

使用工厂：

from factory.location import EmployeeFactory

def test_only_admin_can_see_this_page():
    EmployeeFactory.create_batch(size=100)

    for employee in session.query(Employee).all():
        login(employee)

        with self.app.test_request_context('/'):
        response = self.test_client.get(ADMIN_PAGE)
        if employee.role == ROLE_ADMIN
            eq_(200, response.status_code)
        else:
            eq_(401, response.status_code)

        logout(employee)

<强>故障：

EmployeeFactory.create_batch(size=100)在测试会话中创建100个Employee个对象。
我们可以从工厂session访问这些对象。

有关在SQLAlchemy中使用factory_boy的更多信息：https://factoryboy.readthedocs.io/en/latest/orms.html?highlight=sqlalchemy#sqlalchemy 特别注意会话管理：https://factoryboy.readthedocs.io/en/latest/orms.html?highlight=sqlalchemy#managing-sessions

使用Python的fixture模块生成夹具数据

1 个答案: