获取固定装置中的主键（id）（Python，SQLAlchemy）

Question

我正在使用 fixture 来测试 Pylons 应用程序，但我偶然发现了一个问题。

假设我有这样的数据集：

class CompanyData(DataSet):

    class test_company:
        company_full_name = u'Firma Tęst'
        company_short_name = u'TęstCo'

class UserData(DataSet):

    class test_user:
        user_login = 'testuser'
        user_password = 'test'
        company = CompanyData.test_company

现在的问题是，当我在功能测试中使用这些数据时（如 http://farmdev.com/projects/fixture/using-fixture-with-pylons.html)，我无法获取 id (主键)公司。

在我的应用程序中，用户登录后应重定向到公司资料页面，这就是我需要公司 ID 的原因。测试看起来或多或少像这样：

self.app.post(url(controller='main', action='login'), params={
    'login': UserData.test_user.user_login,
    'password': UserData.test_user.user_password
})

response = self.app.get(url(
    controller='events', action='index',
    company_id=UserData.test_user.company.company_id, # This doesn't work
    view='active'))
assert ... in response

第一个请求登录用户，第二个请求检查登录后她是否可以访问公司资料页面。

这样我得到：

AttributeError：类 test_company 有 没有属性“company_id”

我也尝试过：

UserData.test_user.company.ref('company_id')

但结果是：

这对我来说似乎很奇怪......为什么它没有加载？

有什么办法可以找出主键吗？

Answer 1

UserData.test_user.company.ref('id')
or
UserData.test_user.ref('company_id')