From c4436fd42f2936e5fb0f95434d06e45aa9959ca0 Mon Sep 17 00:00:00 2001 From: Simon Busch Date: Wed, 9 Apr 2014 13:18:33 +0200 Subject: [PATCH] Add functionality to add own objects to internal object list In some case the tests might want to create dynamically dbus objects which extended functionality from own class definitions within templates. In such cases we need to register those objects with the internal object manager of dbusmock. Signed-off-by: Simon Busch --- dbusmock/__init__.py | 4 ++-- dbusmock/mockobject.py | 8 ++++++++ 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/dbusmock/__init__.py b/dbusmock/__init__.py index 8a482ab..3d5d71a 100644 --- a/dbusmock/__init__.py +++ b/dbusmock/__init__.py @@ -14,8 +14,8 @@ __license__ = 'LGPL 3+' __version__ = '0.16.7' from dbusmock.mockobject import (DBusMockObject, MOCK_IFACE, - OBJECT_MANAGER_IFACE, get_object, get_objects) + OBJECT_MANAGER_IFACE, get_object, get_objects, add_object) from dbusmock.testcase import DBusTestCase __all__ = ['DBusMockObject', 'MOCK_IFACE', 'OBJECT_MANAGER_IFACE', - 'DBusTestCase', 'get_object', 'get_objects'] + 'DBusTestCase', 'get_object', 'get_objects', 'add_object'] diff --git a/dbusmock/mockobject.py b/dbusmock/mockobject.py index 586dbad..e4f130f 100644 --- a/dbusmock/mockobject.py +++ b/dbusmock/mockobject.py @@ -688,6 +688,14 @@ dbus.service._method_lookup = _dbusmock_method_lookup # Helper API for templates # +def add_object(path, obj): + if path in objects: + raise dbus.exceptions.DBusException( + 'org.freedesktop.DBus.Mock.NameError', + 'object %s already exists' % path) + + objects[path] = obj + def get_objects(): '''Return all existing object paths'''