python - 您如何模拟 App Engine 中的用户服务？

Question

我正在使用 Google App Engine testbed 框架来编写带有模拟对象的测试用例。这已记录在 here .我的数据存储测试使用模拟数据库 (Testbed.init_datastore_v3_stub) 运行良好，这让我的测试用例可以在一个快速、新鲜的数据库上运行，该数据库为每个测试用例重新初始化。现在我想测试依赖于当前用户的功能。

还有另一个名为 Testbed.init_user_stub 的测试平台服务，我可以激活它以获得“假”用户服务。不幸的是，这似乎没有任何文档。我正在像这样激活和使用它:

import unittest
from google.appengine.ext import testbed
from google.appengine.api import users

class MyTest(unittest.TestCase):
    def setUp(self):
        self.testbed = testbed.Testbed()
        self.testbed.activate()
        self.testbed.init_user_stub()

    def testUser(self):
        u = users.get_current_user()
        self.assertNotEqual(u, None)

问题是我还没有找到任何方法来告诉“假”用户服务对“假”用户进行身份验证。所以运行那个测试，我(可以预见)得到

AssertionError: None == None

意思是假用户服务告诉我的应用当前用户没有登录。我怎样才能告诉假用户服务假装用户已登录？理想情况下，我希望能够指定假用户的昵称、电子邮件、user_id 以及他们是否是管理员。这似乎是一件很常见的事情(因为您需要测试应用程序在以下情况下的行为:a)没有人登录，b)用户登录，c)管理员登录)，但是谷歌搜索"init_user_stub"几乎不返回任何内容。

注意:如果你想测试上面的程序，你需要在顶部添加这个:

import sys
sys.path.append('/PATH/TO/APPENGINE/SDK')
import dev_appserver
dev_appserver.fix_sys_path()

这个到底部:

if __name__ == '__main__':
    unittest.main()

Answer 1

好吧，我认为没有官方的方法可以做到这一点，但是我一直在阅读源代码，并且发现了一种到目前为止运行良好的“hack”方法。 (通常我会担心使用未记录的行为，但它是一个测试套件，所以它只有在开发服务器上工作时才重要。)

开发服务器根据三个环境变量找出当前登录的用户:

• USER_EMAIL:用户的电子邮件地址，和用户的昵称。
• USER_ID:用户的唯一 Google ID(字符串)。
• USER_IS_ADMIN:如果用户是非管理员，则为“0”，如果用户是管理员，则为“1”。

您可以像设置任何其他环境变量一样使用 os.environ 来设置它们，它们会立即生效(显然这在生产服务器上不起作用 )。但是您可以将它们与 testbed 的 user_stub 一起使用，并且当您停用 testbed 时它们将被重置(您应该在 tearDown 上执行此操作，以便为每个测试用例获得一个全新的环境)。

由于设置环境变量有点笨拙，所以我写了一些包装函数来打包它们:

import os

def setCurrentUser(email, user_id, is_admin=False):
    os.environ['USER_EMAIL'] = email or ''
    os.environ['USER_ID'] = user_id or ''
    os.environ['USER_IS_ADMIN'] = '1' if is_admin else '0'

def logoutCurrentUser():
    setCurrentUser(None, None)