autopilot, launcher: get an icon by controlled xid

(bzr r2349.1.3)

1 files changed, 8 insertions, 0 deletions

diff --git a/tests/autopilot/autopilot/emulators/unity/launcher.py b/tests/autopilot/autopilot/emulators/unity/launcher.py
index a6b028972..742dcf77c 100644
--- a/tests/autopilot/autopilot/emulators/unity/launcher.py
+++ b/tests/autopilot/autopilot/emulators/unity/launcher.py
@@ -338,6 +338,14 @@ class LauncherModel(UnityIntrospectionObject):
 
 return None
 
+ def get_icon_by_window_xid(self, xid):
+ """Gets a launcher icon that controls the specified window xid."""
+ icons = [i for i in self.get_children_by_type(SimpleLauncherIcon) if i.xids.contains(xid)]
+ if (len(icons)):
+ return icons[0]
+
+ return None
+
 def get_icons_by_filter(self, **kwargs):
 """Get a list of icons that satisfy the given filters.