~allanlesage/autopilot/unity-panel-service-mixin

# -*- Mode: Python; coding: utf-8; indent-tabs-mode: nil; tab-width: 4 -*-

# Copyright 2013 Canonical

# Author: Allan LeSage

#

# This program is free software: you can redistribute it and/or modify it

# under the terms of the GNU General Public License version 3, as published

# by the Free Software Foundation.

 

import psutil

import dbus

from autopilot.introspection import ApplicationIntrospectionTestMixin, get_autopilot_proxy_object_for_process

from autopilot.emulators.dbus_handler import get_session_bus

 

 

 

def find_process(process_name):

    """Returns a psutil process similar to a Popen process."""

 

    def process_name_equals(process):

        if process.name == process_name:

            return True

        else:

            return False

    processes = filter(process_name_equals, psutil.get_process_list())

    try:

        return processes[0]

    except IndexError:

        return None

 

 

class UnityPanelServiceTestMixin(ApplicationIntrospectionTestMixin):

 

    def _update_activation_environment_with_autopilot(self):

        """Add 'autopilot' to GTK_MODULES via Unity dbus-activation.

 

        When unity-panel-service automatically re-launches, it'll come back with

        'autopilot' in its GTK_MODULES.

        """

 

        session_bus = get_session_bus()

        proxy = session_bus.get_object('org.freedesktop.DBus',

                                       "/org/freedesktop/DBus")

        proxy.UpdateActivationEnvironment({'GTK_MODULES':

                                               "autopilot"})

 

 

    def _kill_unity_panel_service(self):

        """Kill the process named 'unity-panel-service'."""

 

        unity_panel_service = find_process('unity-panel-service')

        if unity_panel_service==None:

            return

        unity_panel_service.kill()

 

 

    def prepare_environment(self):

        """Set up 'unity-panel-service' as our app."""

 

        # add our GTK_MODULES to Unity's DBus launching env

        self._update_activation_environment_with_autopilot()

        # kill the indicators; they'll come back with 'autopilot' appended to GTK_MODULES

        self._kill_unity_panel_service()

        # wait until the unity-panel-service comes back

        unity_panel_service_process = None

        while unity_panel_service_process == None:

            unity_panel_service_process = find_process('unity-panel-service')

        self.app = get_autopilot_proxy_object_for_process(

            unity_panel_service_process)