如何为用Python编写的DBUS服务编写功能测试?

时间:2009-02-04 10:29:14

标签: python unit-testing dbus

(标题是:“如何为用Python编写的DBUS服务编写单元测试?”)

我已经开始使用dbus-python编写DBUS服务,但是我在编写测试用例时遇到了麻烦。

以下是我尝试创建的测试示例。请注意,我在setUp()中放置了一个GLib事件循环,这就是问题所在:

import unittest

import gobject
import dbus
import dbus.service
import dbus.glib

class MyDBUSService(dbus.service.Object):
    def __init__(self):
        bus_name = dbus.service.BusName('test.helloservice', bus = dbus.SessionBus())
        dbus.service.Object.__init__(self, bus_name, '/test/helloservice')

    @dbus.service.method('test.helloservice')
    def hello(self):
        return "Hello World!"


class BaseTestCase(unittest.TestCase):

    def setUp(self):
        myservice = MyDBUSService()
        loop = gobject.MainLoop()
        loop.run()
        # === Test blocks here ===

    def testHelloService(self):
        bus = dbus.SessionBus()
        helloservice = bus.get_object('test.helloservice', '/test/helloservice')
        hello = helloservice.get_dbus_method('hello', 'test.helloservice')
        assert hello() == "Hello World!"

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

我的问题是DBUS实现要求您启动一个事件循环,以便它可以开始调度事件。常见的方法是使用GLib的gobject.MainLoop()。start()(虽然我没有嫁给这种方法,如果有人有更好的建议)。如果您没有启动事件循环,该服务仍会阻止,您也无法查询它。

如果我在测试中启动我的服务,事件循环会阻止测试完成。我知道该服务正在运行,因为我可以使用qdbus工具从外部查询服务,但我无法在启动它的测试中自动执行此操作。

我正在考虑在测试中使用某种程序来处理这个问题,但是我希望有人可能有一个更整洁的解决方案,或者至少是一个很好的起点,我将如何编写这样的测试。

6 个答案:

答案 0 :(得分:6)

在Ali A的帖子的帮助下,我设法解决了我的问题。阻塞事件循环需要启动到一个单独的进程中,以便它可以监听事件而不会阻止测试。

请注意我的问题标题包含一些不正确的术语,我试图编写功能测试,而不是单元测试。我知道这种区别,但直到后来才意识到我的错误。

我在我的问题中调整了这个例子。它松散地类似于“test_pidavim.py”示例,但使用导入“dbus.glib”来处理glib循环依赖,而不是在所有DBusGMainLoop中编码:

import unittest

import os
import sys
import subprocess
import time

import dbus
import dbus.service
import dbus.glib
import gobject

class MyDBUSService(dbus.service.Object):

    def __init__(self):
        bus_name = dbus.service.BusName('test.helloservice', bus = dbus.SessionBus())
        dbus.service.Object.__init__(self, bus_name, '/test/helloservice')

    def listen(self):
        loop = gobject.MainLoop()
        loop.run()

    @dbus.service.method('test.helloservice')
    def hello(self):
        return "Hello World!"


class BaseTestCase(unittest.TestCase):

    def setUp(self):
        env = os.environ.copy()
        self.p = subprocess.Popen(['python', './dbus_practice.py', 'server'], env=env)
        # Wait for the service to become available
        time.sleep(1)
        assert self.p.stdout == None
        assert self.p.stderr == None

    def testHelloService(self):
        bus = dbus.SessionBus()
        helloservice = bus.get_object('test.helloservice', '/test/helloservice')
        hello = helloservice.get_dbus_method('hello', 'test.helloservice')
        assert hello() == "Hello World!"

    def tearDown(self):
        # terminate() not supported in Python 2.5
        #self.p.terminate()
        os.kill(self.p.pid, 15)

if __name__ == '__main__':

    arg = ""
    if len(sys.argv) > 1:
        arg = sys.argv[1]

    if arg == "server":
        myservice = MyDBUSService()
        myservice.listen()

    else:
        unittest.main()

答案 1 :(得分:3)

简单的解决方案:不要通过dbus进行单元测试。

而是编写单元测试以直接调用您的方法。这更符合单元测试的本质。

您可能还需要一些自动集成测试,它们检查通过dbus运行,但它们不需要如此完整,也不需要单独运行。您可以在单独的过程中进行设置以启动服务器的实际实例。

答案 2 :(得分:2)

我可能有点离开我的联盟,因为我不知道python并且只知道这个神奇的“dbus”是什么,但如果我理解正确,它需要你创建一个相当不寻常的测试环境runloops,扩展设置/拆卸等。

问题的答案是使用mocking。创建一个定义接口的抽象类,然后构建一个用于实际代码的对象。出于测试的目的,您构建一个模拟对象通过相同的接口进行通信,但具有为测试目的定义的行为。您可以使用此方法“模拟”通过事件循环运行的dbus对象,执行某些工作等,然后只关注测试您的类应该如何对该对象完成的“工作”结果做出反应。

答案 3 :(得分:2)

您只需要确保正确处理主循环。

def refresh_ui():
    while gtk.events_pending():
       gtk.main_iteration_do(False)

这将运行gtk主循环,直到它完成所有处理,而不是仅运行它并阻止。

有关它在实践中的完整示例,请对dbus接口进行单元测试,请访问:http://pida.co.uk/trac/browser/pida/editors/vim/test_pidavim.py

答案 4 :(得分:2)

你也可以在setUp方法中非常简单地在一个单独的线程中启动mainloop。

这样的事情:

import threading
class BaseTestCase(unittest.TestCase):
    def setUp(self):
        myservice = MyDBUSService()
        self.loop = gobject.MainLoop()
        threading.Thread(name='glib mainloop', target=self.loop.run)
    def tearDown(self):
        self.loop.quit()

答案 5 :(得分:0)

查看python-dbusmock库。

它隐藏了丑陋的子进程逻辑,所以你不必在测试中担心它。