Coverage for ion/services/cei/epu_controller_list_client : 84.00%

#!/usr/bin/env python 

""" 

@file ion/services/cei/epu_controller_list_client.py 

@brief Provides client interface to controller_list service that provides a 

       list of active EPU Controller service names in the system 

""" 

import ion.util.ionlog 

log = ion.util.ionlog.getLogger(__name__) 

from twisted.internet import defer 

from ion.core.process.service_process import ServiceClient 

class EPUControllerListClient(ServiceClient): 

    """Client for querying EPUControllerListService 

    """ 

    def __init__(self, proc=None, **kwargs): 

        if not 'targetname' in kwargs: 

            kwargs['targetname'] = "epu_controller_list" 

        self.force_service_exists = bool(kwargs.get('force_service_exists', False)) 

        self.service_name = kwargs['targetname'] 

        ServiceClient.__init__(self, proc, **kwargs) 

    @defer.inlineCallbacks 

    def list(self): 

        """Query the EPUControllerListService 

        """ 

        log.debug("ServiceName = " + self.service_name) 

        yield self._check_init() 

        service_exists = True 

        if not self.force_service_exists: 

            service_exists = yield self.does_service_exist(self.service_name) 

        if not service_exists: 

            log.debug("controller_list_client.list: Returning static list for AIS unit testing") 

            defer.returnValue(['dataservices_epu_controller', 

                               'agentservices_epu_controller', 

                               'associationservices_epu_controller']) 

        log.debug("controller_list_client.list: Sending EPU controller list query") 

        (content, headers, msg) = yield self.rpc_send('list', {}) 

        log.debug("controller_list_client.list: list returned "+str(content)) 

        defer.returnValue(content)