如何使用Twisted Trial对Autobahn应用程序进行单元测试？

Question

假设您只使用Autobahn连接(不是原始WebSocket).

我们如何测试没有网络,我们的RPC方法和事件？
由于它是Twisted,我认为最合适的工具是Twisted Trial.

但我无法弄清楚如何编写这些测试,而不需要编写大量的样板代码并重新使用Autobahn的内部实现(甚至我不确定我是否能够这样做).

你会怎么做？

Answer 1

这是试图回答我自己的问题。

问题

然后，为了对 RPC 方法和事件进行单元测试，我们需要假设 Autobahn 已经经过充分测试，我们不必测试它，解决方案变得简单：

解决方案

嘲笑一切。

语境

在我的应用程序中，我有两种类型的组件（请阅读 ApplicationSession）：StandardComponent 和 DatabaseComponent（继承自 StandardComponent）。

单元测试期间最大的问题是我们有很多依赖项，例如数据库连接、Redis 连接等......

例子

我在测试中所做的是通过子类化来修补所有这些对象unittest.TestCase：

class APITestCase(unittest.TestCase):

    def _patchObject(self, module_name, **kwargs):
        patcher = patch(module_name, **kwargs)
        mock = patcher.start()

        self.patches.append(patcher)
        return mock

    def setUp(self):
        logging.disable(logging.CRITICAL)
        self.patches = []
        self.session = self._patchObject('components.ApplicationAPI')
        self.database = self._patchObject('txpostgres.txpostgres.Connection')

    def tearDown(self):
        for patcher in self.patches:
            patcher.stop()

我在测试用例中注入了一个模拟会话和一个模拟数据库。

然后，测试就变得非常非常简单。

每当我调用需要调用数据库或从数据库获取结果的 RPC 方法时，我都会对其进行修补： self.mocked_auth_user.return_value = (1, "abc", "something", "admin")

在我的测试方法中：

def test_authenticate_success(self):
    self.mocked_auth_user.return_value = (1, "abc", "paris", "admin")

    def _doTest(auth_result):
        attempted_auth_result = {
            "secret": "abc",
            "role": "admin",
            "authid": "1",
            "salt": "paris",
            "iterations": 1000,
            "keylen": 32
        }

        self.assertEqual(auth_result, attempted_auth_result)
        self.mocked_auth_user.assert_called_with(self.api.database, "raito")

    return self.api.authenticate("test", "raito", {}).addCallback(_doTest)

您可以做一些更高级和有趣的测试来看看您的方法是否是防失败的：

def test_authenticate_authid_not_found(self):
    def _raiseException(db, user):
        return defer.fail(Exception("User {} not found!".format(user)))

    self.mocked_auth_user.side_effect = _raiseException
    return self.failUnlessFailure(self.api.authenticate("test", "raito", {}), AuthenticationError)

事件也是如此，您只需调用它们并测试它们是否发布事件 ( self.session.publish.assert_called_with(...))

它变得神奇！

无论如何，它解决了单元测试问题，但集成还没有。我正在研究这个问题，但是使用一些虚拟化技术（Docker）或类似的技术可能会解决这个问题。