您如何在App Engine中模拟用户服务？

Question

我正在使用Google App
Enginetestbed框架编写带有模拟对象的测试用例。这在此处记录。我已经使用模拟数据库（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像设置任何其他环境变量一样使用它们来设置它们，它们会立即生效（ 显然，这在生产服务器上将不起作用
）。但是您可以将它们与测试床的user_stub一起使用，并且当您停用测试床时，它们将被重置（您应该在上执行此操作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)